import sqlalchemy as sa
from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()


class User(db.Model):
    id = sa.Column(sa.Integer, primary_key=True)
    name = sa.Column(sa.String)
    surname = sa.Column(sa.String)
    email = sa.Column(sa.String)
    mtan = sa.Column(sa.String, unique=True)


class TimeStamp(db.Model):
    id = sa.Column(sa.Integer, primary_key=True)
    user_id = sa.ForeignKey(User.id)
    create = sa.Column(sa.TIMESTAMP)  # create time
    timestamp = sa.Column(sa.TIMESTAMP)  # inserted by user
    on_work = sa.Column(sa.Boolean)  # working time?


class RFIDNumber(db.Model):
    id = sa.Column(sa.Integer, primary_key=True)
    user_id = sa.ForeignKey(User.id)
    number = sa.Column(sa.String)