class Robot(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    robot_name = db.Column(db.String(100), unique=True, nullable=False)


class Command(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    command_name = db.Column(db.String(100), unique=True, nullable=False)
    command_words = db.Column(db.String(1000), unique=True, nullable=False)
    response_id = db.Column(db.Integer, db.ForeignKey('response.id'), nullable=False)
    result_id = db.Column(db.Integer, db.ForeignKey('result.id'))
    status_id = db.Column(db.Integer, db.ForeignKey('status.id'))
    info_id = db.Column(db.Integer, db.ForeignKey('info.id'))


class Result(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    commands = db.relationship('Command', backref='result', lazy='dynamic')
    result = db.Column(db.String(100))
    finish_date = db.Column(db.DateTime())


class Status(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    commands = db.relationship('Command', backref='status', lazy='dynamic')
    status = db.Column(db.String(100))
    last_updated = db.Column(db.DateTime())


class Info(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    commands = db.relationship('Command', backref='info', lazy='dynamic')
    title = db.column(db.String(100))
    description = db.column(db.String(500))


class Response(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    response_text = db.Column(db.String(100))
    commands = db.relationship('Command', backref='response', lazy='dynamic')


class User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)

    username = db.Column(db.String(100), unique=True, nullable=False)
    email = db.Column(db.String(255), unique=True)
    password = db.Column(db.String(100), nullable=False)

    def set_id(self, _id):
        self.username = _id

    def get_id(self):
        return self.username