from flask_marshmallow.sqla import SQLAlchemyAutoSchema
from marshmallow import EXCLUDE
from sqlalchemy import func
import marshmallow
from marshmallow import INCLUDE, fields

from crc import db, ma

class DataStoreModel(db.Model):
    __tablename__ = 'data_store'
    id = db.Column(db.Integer, primary_key=True)
    last_updated = db.Column(db.DateTime(timezone=True), default=func.now())
    key = db.Column(db.String, nullable=False)
    workflow_id = db.Column(db.Integer)
    study_id = db.Column(db.Integer, nullable=True)
    task_id = db.Column(db.String)
    spec_id = db.Column(db.String)
    user_id = db.Column(db.String, nullable=True)
    value = db.Column(db.String)


class DataStoreSchema(ma.Schema):
    id = fields.Integer(required=False)
    key = fields.String(required=True)
    last_updated = fields.DateTime(server_default=func.now(), onupdate=func.now())
    workflow_id = fields.Integer()
    study_id = fields.Integer(allow_none=True)
    task_id = fields.String()
    spec_id = fields.String()
    user_id = fields.String(allow_none=True)
    value = fields.String()