Как логично совместить документы mongodb с объектами проекта?

Question

[Max Payne]

У меня в проекте есть некоторые сущности, объекты, например класс 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)

Только кажется красиво, но с большими структурами документа, при таком раскладе, можно просто помереть.

Comments

[Nurbol Suleimenov]

Max Payne, у Вас невнятное постановка вопроса.

1. Я правильно понимаю, что у Вас ORM для доступа к документам в Mongo? Если да, то какой?
   
2. Пример Вы приводите с классом Chat, а использование с User. Опечатка?
   
3. Идея программы запутана, т.е. не понятно что Вы хотели изначально сделать. Попытайтесь четко описать цель программы, т.к. это важнее Вашей реализации. Возможно выбор реализации идеи ошибочно или неоптимально и есть лучшие решения
   

Answer 1

[Max Payne]

Нашёл для себя такое решение.

class Chat:
    def __init__(self, bot, i=0):
        self._bot = bot

        data = modules.Chat.objects(bot=bot.id, i=i)

        if not data:
            self._data = modules.Chat(bot=bot.id, i=i).save()
        else:
            self._data = data[0]

    def __getattr__(self, item):
        self._data.reload()
        return getattr(self._data, item)

    def __setattr__(self, key, value):
        if key not in ('_data', '_bot'):
            data = self._data.__setattr__(key, value)
            self._data.save()
            return data
        else:
            return super().__setattr__(key, value)

Но осталась проблемка: когда поле является EmbeddedDocumentListField - в него нельзя сделать append / extend / pop и тд.

Answer 2

[alternativshik]

user.balance += 1000
наверное будет неатомарным и все может пойти по *изде (хотя, неизвестно, как ваша обертка для моно работает на этот счет.)

Comments

[Max Payne]

Здесь, на сколько я понимаю, тоже будет работать update. Но у меня большая вложенная структура, для которой update не подойдёт.