import datetime

from mongoengine import *


class Organisation(Document):

    name = StringField(required=True)
    name_short = StringField(required=True)

    date = DateTimeField(default=datetime.datetime.now,
                         wtf_options={"render_kw": {
                             "step": "60"
                         }})

    labtest_citt = StringField(required=False)
    labtest_shear_extension = StringField(required=False)

    meta = {
        'allow_inheritance': True,
        'index_opts': {},
        'index_background': True,
        'index_cls': False,
        'auto_create_index': True,
        'collection': 'organisation',
        'db_alias': 'dblabtests',
    }


class User(Document):

    active = BooleanField(required=True, default=True)

    org_id = LazyReferenceField(Organisation,
                                required=True,
                                reverse_delete_rule=CASCADE)

    date_added = DateTimeField(default=datetime.datetime.now,
                               wtf_options={"render_kw": {
                                   "step": "60"
                               }})

    name = StringField(required=False)
    
    email = EmailField(required=True)
    
    password = StringField(required=True)

    meta = {
        'allow_inheritance': True,
        'index_opts': {},
        'index_background': True,
        'index_cls': False,
        'auto_create_index': True,
        'collection': 'user',
        'db_alias': 'dblabtests',
    }