projects / wpp.git / blame
| Commit | Line | Data |
|---|---|---|
| 047d4d29 MR |
1 | from app import db |
| 2 | from config import ADMINS | |
| 3 | import enum | |
| 4 | ||
| 5 | class Format(enum.Enum): | |
| 6 | jpg = 1 | |
| 7 | raw = 2 | |
| 8 | tiff = 3 | |
| 9 | ||
| 10 | class Role(enum.Enum): | |
| 11 | read = 1 | |
| 12 | write = 2 | |
| 13 | admin = 3 | |
| 14 | ||
| 15 | class User(db.Model): | |
| 16 | __tablename__ = 'user' | |
| 17 | id = db.Column(db.Integer, primary_key=True, autoincrement=True) | |
| 18 | name = db.Column(db.String(255)) | |
| 19 | email = db.Column(db.String(255), index=True, unique=True) | |
| 20 | username = db.Column(db.String(255), index=True, unique=True) | |
| 21 | password = db.Column(db.String(255)) | |
| 22 | portfolios = db.relationship('Portfolio', backref='owner', lazy=True, cascade='save-update, merge, delete, delete-orphan') | |
| 23 | acl = db.relationship('AccessRight', backref='user', lazy=True) | |
| 24 | ||
| 25 | @property | |
| 26 | def is_authenticated(self): | |
| 27 | return True | |
| 28 | ||
| 29 | @property | |
| 30 | def is_active(self): | |
| 31 | return True | |
| 32 | ||
| 33 | @property | |
| 34 | def is_anonymous(self): | |
| 35 | return False | |
| 36 | ||
| 37 | @property | |
| 38 | def is_admin(self): | |
| 39 | return self.email in ADMINS | |
| 40 | ||
| 41 | def get_id(self): | |
| 42 | return str(self.id) # python 3 | |
| 43 | ||
| 44 | def __repr__(self): | |
| 45 | return '<User Name: %r Email: %r username: %r>' % (self.name, self.email, self.username) | |
| 46 | ||
| 47 | class Portfolio(db.Model): | |
| 48 | __tablename__ = 'portfolio' | |
| 49 | id = db.Column(db.Integer, primary_key=True, autoincrement=True) | |
| 50 | name = db.Column(db.String(255)) | |
| 51 | user_id = db.Column(db.ForeignKey('user.id', onupdate='CASCADE', ondelete='CASCADE'), nullable=False) | |
| 52 | albums = db.relationship('Album', backref='portfolio', lazy=True, cascade='save-update, merge, delete, delete-orphan') | |
| 53 | ||
| 54 | def __repr__(self): | |
| 55 | return '<Portfolio Name: %r Owner: %r>' % (self.name, self.owner) | |
| 56 | ||
| 57 | class Album(db.Model): | |
| 58 | __tablename__ = 'album' | |
| 59 | id = db.Column(db.Integer, primary_key=True, autoincrement=True) | |
| 60 | name = db.Column(db.String(255)) | |
| 61 | photos = db.relationship('Photo', backref='album', lazy=True, cascade='save-update, merge, delete, delete-orphan') | |
| 62 | portfolio_id = db.Column(db.ForeignKey('portfolio.id', onupdate='CASCADE', ondelete='CASCADE'), nullable=False) | |
| 63 | ||
| 64 | def __repr__(self): | |
| 65 | return '<Album Name: %r Portfolio: %r>' % (self.name, self.portfolio) | |
| 66 | ||
| 67 | class Photo(db.Model): | |
| 68 | __tablename__ = 'photo' | |
| 69 | id = db.Column(db.Integer, primary_key=True, autoincrement=True) | |
| 70 | file = db.Column(db.String(255)) | |
| 71 | name = db.Column(db.String(255)) | |
| 72 | exif = db.Column(db.Text) | |
| 73 | format = db.Column(db.Enum(Format)) | |
| 74 | sibling = db.Column(db.ForeignKey('photo.id')) | |
| 75 | album_id = db.Column(db.ForeignKey('album.id', onupdate='CASCADE', ondelete='CASCADE'), nullable=False) | |
| 76 | ||
| 77 | def __repr__(self): | |
| 78 | return '<Photo Name: %r File: %r Format: %r Album: %r>' % (self.name, self.file, self.format, self.album) | |
| 79 | ||
| 80 | class AccessRight(db.Model): | |
| 81 | __tablename__ = 'accessright' | |
| 82 | id = db.Column(db.Integer, primary_key = True, autoincrement=True) | |
| 83 | right = db.Column(db.Enum(Role)) | |
| 84 | user_id = db.Column(db.ForeignKey('user.id', onupdate='CASCADE', ondelete='CASCADE'), nullable=False) | |
| 85 | object = db.Column(db.Integer) | |
| 86 | ||
| 87 | def __repr__(self): | |
| 88 | return '<AccessRight Right: %r User: %r Object: %r>' % (self.right, self.user, self.object) |