20 lines
588 B
Python
20 lines
588 B
Python
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}>'
|