mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-08 21:14:49 +08:00
23 lines
552 B
Python
23 lines
552 B
Python
import json
|
|
from typing import Dict
|
|
|
|
import dataclasses
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class ModelInfo(object):
|
|
# class_name: str
|
|
related_managers: Dict[str, 'ModelInfo'] = dataclasses.field(default_factory=dict)
|
|
|
|
|
|
def get_default_base_models():
|
|
return {'django.db.models.base.Model': ModelInfo()}
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class DjangoModelsRegistry(object):
|
|
base_models: Dict[str, ModelInfo] = dataclasses.field(default_factory=get_default_base_models)
|
|
|
|
def __contains__(self, item: str) -> bool:
|
|
return item in self.base_models
|