1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
from typing import ClassVar, Optional
from beanie.odm.settings.document import DocumentSettings
class SettersInterface:
_document_settings: ClassVar[Optional[DocumentSettings]]
@classmethod
def set_collection(cls, collection):
"""
Collection setter
"""
cls._document_settings.pymongo_collection = collection
@classmethod
def set_database(cls, database):
"""
Database setter
"""
cls._document_settings.pymongo_db = database
@classmethod
def set_collection_name(cls, name: str):
"""
Collection name setter
"""
cls._document_settings.name = name # type: ignore
|
