29 lines
689 B
Python
29 lines
689 B
Python
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
|