mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-06 20:24:31 +08:00
19 lines
418 B
Python
19 lines
418 B
Python
from typing import Set
|
|
|
|
import dataclasses
|
|
|
|
|
|
def get_default_base_models():
|
|
return {'django.db.models.base.Model'}
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class DjangoModelsRegistry(object):
|
|
base_models: Set[str] = dataclasses.field(default_factory=get_default_base_models)
|
|
|
|
def __contains__(self, item: str) -> bool:
|
|
return item in self.base_models
|
|
|
|
def __iter__(self):
|
|
return iter(self.base_models)
|