domain.py

import os
from mongoengine import *

connect(host=os.getenv('MONGODB_URI') + '/' + os.getenv('MONGODB_DATABASE'))

__all__ = (
    'NotificationFlag', 'Notification', 'ExchangeNotification',
    'MaterialInformation', 'MaterialInformationSubscription'
)


class NotificationFlag(EmbeddedDocument):
    rate = DecimalField()
    notified = BooleanField()


class Notification(EmbeddedDocument):
    currency = StringField(required=True, primary_key=True)
    sell = EmbeddedDocumentField(NotificationFlag)
    buy = EmbeddedDocumentField(NotificationFlag)


class ExchangeNotification(Document):
    source = StringField(required=True, primary_key=True)
    notify_rates = EmbeddedDocumentListField(Notification)


class MaterialInformation(Document):
    id = StringField(required=True, primary_key=True)
    company_code = IntField(required=True)
    spoken_at = DateTimeField(required=True)
    speaker = StringField()
    speaker_title = StringField()
    speaker_phone = StringField()
    terms = StringField()
    fact_date = DateField(required=True)
    title = StringField(required=True)
    content = StringField(required=True)

    @property
    def spoken_roc_date(self):
        spoken_date_str = self.spoken_at.strftime('%Y/%m/%d')
        return spoken_date_str.replace(spoken_date_str[:4],
                                       str(int(spoken_date_str[:4]) - 1911))

    @property
    def fact_roc_date(self):
        fact_date_str = self.fact_date.strftime('%Y/%m/%d')
        return fact_date_str.replace(fact_date_str[:4],
                                     str(int(fact_date_str[:4]) - 1911))


class MaterialInformationSubscription(Document):
    company_code = StringField(required=True, primary_key=True)
    sources = ListField(StringField())
    last_pushed_id = StringField()