class Fields(EmbeddedDocument):
    _id = False
    label = StringField(required=True)
    name = StringField(required=True)
    regex = StringField(default=None)
    type = StringField(required=True)
    value = StringField(default=None)


class Bank(Document):
    _id = IntField(required=True, unique=True, primary_key=True)
    charged = BooleanField()
    code = StringField()
    fields = EmbeddedDocumentListField(Fields)
    hidden = BooleanField()
    id_weboob = StringField()
    name = StringField()


class User(Document):
    _id = IntField(0, unique=True, primary_key=True)
    email = EmailField()
    password = StringField()
    bi_id_user = IntField()


class Connection(Document):
    _id = IntField(0, required=True, unique=True)
    id_user = ReferenceField(User)
    active = BooleanField()
    bank = ReferenceField(Bank)
    id_bank = IntField()
    bi_id_user = IntField()
    error = StringField()
    expire = DateTimeField()
    last_update = DateTimeField()
    accounts = ListField(ReferenceField('Account'))
    # def clean(self):
    #     print self.id_user


class Currency(EmbeddedDocument):
    _id = StringField(required=True, unique=True)
    prefix = BooleanField()
    symbol = StringField()

class SecurityType(Document):
    color = StringField()
    _id = IntField(primary_key=True)
    idParent = StringField()
    name = StringField()

class Investment(Document):
    _id = IntField(required=True, unique=True, primary_key=True)
    id_account = ReferenceField('Account')
    code = StringField()
    description = StringField()
    diff = IntField()
    label = StringField()
    portfolio_share = DecimalField()
    prev_diff = DecimalField()
    quantity = DecimalField()
    id_security = IntField()
    id_type = IntField()
    type = ReferenceField(SecurityType)
    unitprice = DecimalField()
    unitvalue = DecimalField()
    userid = IntField()
    vdate = DateTimeField()
    valuation = DecimalField()

class Transaction(Document):
    _id = IntField(required=True, unique=True)
    id_user = ReferenceField(User)
    id_account = ReferenceField('Account')
    active = BooleanField()
    application_date = DateTimeField()
    id_category = IntField()
    id_cluster = IntField()
    coming = BooleanField()
    comment = StringField()
    commission = DecimalField()
    country = StringField()
    date = DateTimeField()
    date_scraped = DateTimeField()
    deleted = BooleanField()
    documents_count = IntField()
    formatted_value = StringField()
    last_update = DateTimeField()
    new = BooleanField()
    original_currency = StringField()
    original_value = DecimalField()
    original_wording = StringField()
    rdate = DateTimeField()
    simplified_wording = StringField()
    state = StringField()
    stemmed_wording = StringField()
    type = StringField()
    value = DecimalField()
    webid = StringField()
    wording = StringField()

class Account(Document):
    _id = IntField(required=True, unique=True)

    balance = DecimalField()
    currency = EmbeddedDocumentField(Currency)
    deleted = BooleanField()
    display = BooleanField()
    formatted_balance = StringField()
    iban = StringField()
    id_connection = ReferenceField(Connection)
    investments = ListField(ReferenceField(Investment))
    transactions = ListField(ReferenceField(Transaction))
    last_update = DateTimeField()
    name = StringField()
    number = StringField()
    type = StringField()
    id_user = ReferenceField(User)
    bi_id_user = IntField()
    prev_diff_percent =DecimalField()
    prev_diff = DecimalField()
    diff = DecimalField()
    diff_percent = DecimalField()