from sqlalchemy.dialects.mysql import INTEGER

from app import db


class Folder(db.Model):
    __tablename__ = 'folders'

    id = db.Column(INTEGER(unsigned=True), autoincrement=True, primary_key=True)
    user_id = db.Column(INTEGER(unsigned=True), db.ForeignKey('users.id', ondelete='CASCADE'), nullable=False)
    name = db.Column(db.String(128), nullable=False)

    vault_items = db.relationship('VaultItem', backref='folder', lazy='dynamic')

    def to_dict(self):
        return {'id': self.id, 'name': self.name}

    def __repr__(self):
        return f'<Folder {self.name}>'