support more kwargs to __init__

This commit is contained in:
j 2014-09-05 13:04:18 +02:00
parent a0666acf89
commit 954312e0d6

View file

@ -1,10 +1,10 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4 # vi:si:et:sw=4:sts=4:ts=4
from django.db.models.query import QuerySet
from django.db.models.sql import Query from django.db.models.sql import Query
from django.db.models.sql.compiler import SQLCompiler from django.db.models.sql.compiler import SQLCompiler
from django.db import connections from django.db import connections
import django.db.models.query
''' '''
models.py: models.py:
@ -60,10 +60,10 @@ class Query(Query):
return SQLCompiler(self, connection, using) return SQLCompiler(self, connection, using)
class QuerySet(QuerySet): class QuerySet(django.db.models.query.QuerySet):
def __init__(self, model=None, query=None, using=None): def __init__(self, model=None, query=None, using=None, **kwargs):
super(QuerySet, self).__init__(model=model, query=query, using=None) super(QuerySet, self).__init__(model=model, query=query, using=None, **kwargs)
self.query = query or Query(self.model) self.query = query or Query(self.model)
def order_by(self, *args, **kwargs): def order_by(self, *args, **kwargs):