Coverage for mongodb_odm / utils / _internal_models.py: 100%

1from typing import Any, Optional, Union 6 ctx1abcdef

2 

3from mongodb_odm.types import DICT_TYPE 6 ctx1abcdef

4from pydantic import BaseModel 6 ctx1abcdef

5from pydantic._internal._repr import Representation as PydanticRepresentation 6 ctx1abcdef

6from pymongo import AsyncMongoClient, MongoClient 6 ctx1abcdef

7 

8 

9class Connection: 6 ctx1abcdef

10 # Use this model to maintain database connection object structure. 

11 url: Optional[str] = None 6 ctx1abcdef

12 client: Optional[Union[MongoClient[Any], AsyncMongoClient[Any]]] = None 6 ctx1abcdef

13 databases: Optional[set[str]] = None 6 ctx1abcdef

14 connection_kwargs: Optional[DICT_TYPE] = None 6 ctx1abcdef

15 async_is_enabled: bool = False 6 ctx1abcdef

16 

17 

18class RelationalFieldInfo(PydanticRepresentation): 6 ctx1abcdef

19 def __init__( 6 ctx1abcdef

20 self, *, model: Any, local_field: str, related_field: Optional[str] = None 

21 ) -> None: 

22 self.model = model 6 ctx1abcdef

23 self.local_field = local_field 6 ctx1abcdef

24 self.related_field = related_field 6 ctx1abcdef

25 

26 

27class CollectionConfig(BaseModel): 6 ctx1abcdef

28 # Connection configuration model for each model. 

29 collection_name: str 6 ctx1abcdef

30 child_collection_name: Optional[str] = None 6 ctx1abcdef

31 database_name: Optional[str] = None 6 ctx1abcdef

32 has_children: bool = False 6 ctx1abcdef