У меня в проекте есть некоторые сущности, объекты, например класс User, класс Chat и есть документы к которым они привязаны. Могу ли я, каким-либо образом, расширить объект документа? То есть сейчас чтобы обходится только классом Chat мне приходится использовать такую лапшу, а ведь мне всего навсего нужно только создавать чат в базе данных, если его нет и добавить один проперти-метод.
class Chat(Document):
bot = ObjectIdField(required=True)
i = LongField(required=True)
greeting = BooleanField(default=True, required=True)
rules = StringField(min_length=2)
silence_ts = LongField(required=True)
games = EmbeddedDocumentListField(Game)
meta = {
'indexes': [
{"fields": ["bot", "i"], "unique": True}
]
}
class Chat:
def __init__(self, bot, i=0):
self.bot = bot
self.i = i
def __getattr__(self, item):
data = modules.Chat.objects(bot=self.bot.id, i=self.i)
if not data:
data = modules.Chat(bot=self.bot.id, i=self.i).save()
else:
data = data[0]
return getattr(data, item)
@property
def is_silence(self):
return self.silence_ts >= tools.ts()
И за это я плачу невозможностью обновлять многие значения "на лету". Приходится вместо
user = modules.User.objects.first()
# ...
user.balance += 1000
Делать
user = User(bot, 1)
# ...
user.update(inc__balance=1000)
Только кажется красиво, но с большими структурами документа, при таком раскладе, можно просто помереть.
