Skip to content
This repository was archived by the owner on Aug 19, 2025. It is now read-only.

Define the queryset_class in model #150

Open
ponytailer wants to merge 5 commits into
base: master
Choose a base branch
from
Open

Define the queryset_class in model #150

Changes from 1 commit
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
d2e2996
define the queryset_class in model
Feb 22, 2022
3a92a63
fix lint
Feb 22, 2022
b615d5c
fixc
Feb 22, 2022
a12bf50
fix queryset class in metaclass
Feb 22, 2022
2c4315c
Merge branch 'encode:master' into master
ponytailer Mar 2, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fixc
  • Loading branch information
huangsong committed Feb 22, 2022
commit b615d5cf9dbe89b03ca47ce21a86b976b6765d29
8 changes: 2 additions & 6 deletions orm/models.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -83,11 +83,7 @@ def __new__(cls, name, bases, attrs):
if "tablename" not in attrs:
setattr(model_class, "tablename", name.lower())

if "queryset_class" in attrs:
if not isinstance(attrs["queryset_class"], QuerySet):
raise ValueError("queryset must extend QuerySet class.")
else:
attrs["queryset_class"] = None
attrs.setdefault("queryset_class", QuerySet)

for name, field in attrs.get("fields", {}).items():
setattr(field, "registry", attrs.get("registry"))
Expand Down Expand Up @@ -504,7 +500,7 @@ def __init__(self, **kwargs):

@property
def objects(self) -> QuerySet:
return cls.queryset_class() if cls.queryset_class else QuerySet()
return cls.queryset_class()

@property
def pk(self):
Expand Down