mirror of
https://github.com/Derpy-Leggies/OnlyLegs.git
synced 2024-12-28 18:36:21 +00:00
109 lines
3 KiB
Python
109 lines
3 KiB
Python
"""
|
|
OnlyLegs - Database models and ions for SQLAlchemy
|
|
"""
|
|
from uuid import uuid4
|
|
from flask_login import UserMixin
|
|
from onlylegs.extensions import db
|
|
|
|
|
|
class GroupJunction(db.Model): # pylint: disable=too-few-public-methods, C0103
|
|
"""
|
|
Junction table for posts and groups
|
|
Joins with posts and groups
|
|
"""
|
|
|
|
__tablename__ = "group_junction"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
|
|
group_id = db.Column(db.Integer, db.ForeignKey("group.id"))
|
|
post_id = db.Column(db.Integer, db.ForeignKey("post.id"))
|
|
|
|
date_added = db.Column(
|
|
db.DateTime,
|
|
nullable=False,
|
|
server_default=db.func.now(), # pylint: disable=E1102
|
|
)
|
|
|
|
|
|
class Post(db.Model): # pylint: disable=too-few-public-methods, C0103
|
|
"""
|
|
Post table
|
|
"""
|
|
|
|
__tablename__ = "post"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
|
|
author_id = db.Column(db.Integer, db.ForeignKey("user.id"))
|
|
|
|
filename = db.Column(db.String, unique=True, nullable=False)
|
|
mimetype = db.Column(db.String, nullable=False)
|
|
exif = db.Column(db.PickleType, nullable=False)
|
|
colours = db.Column(db.PickleType, nullable=False)
|
|
|
|
description = db.Column(db.String, nullable=False)
|
|
alt = db.Column(db.String, nullable=False)
|
|
|
|
created_at = db.Column(
|
|
db.DateTime,
|
|
nullable=False,
|
|
server_default=db.func.now(), # pylint: disable=E1102
|
|
)
|
|
|
|
junction = db.relationship("GroupJunction", backref="posts")
|
|
|
|
|
|
class Group(db.Model): # pylint: disable=too-few-public-methods, C0103
|
|
"""
|
|
Group table
|
|
"""
|
|
|
|
__tablename__ = "group"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
|
|
name = db.Column(db.String, nullable=False)
|
|
description = db.Column(db.String, nullable=False)
|
|
|
|
author_id = db.Column(db.Integer, db.ForeignKey("user.id"))
|
|
created_at = db.Column(
|
|
db.DateTime,
|
|
nullable=False,
|
|
server_default=db.func.now(), # pylint: disable=E1102
|
|
)
|
|
|
|
junction = db.relationship("GroupJunction", backref="groups")
|
|
|
|
|
|
class User(db.Model, UserMixin): # pylint: disable=too-few-public-methods, C0103
|
|
"""
|
|
User table
|
|
"""
|
|
|
|
__tablename__ = "user"
|
|
|
|
# Gallery used information
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
alt_id = db.Column(db.String, unique=True, nullable=False, default=str(uuid4()))
|
|
|
|
picture = db.Column(db.String, nullable=True, default=None)
|
|
picture_colour = db.Column(db.String, nullable=True, default=None)
|
|
banner = db.Column(db.String, nullable=True, default=None)
|
|
bannner_colour = db.Column(db.String, nullable=True, default=None)
|
|
|
|
username = db.Column(db.String, unique=True, nullable=False)
|
|
email = db.Column(db.String, unique=True, nullable=False)
|
|
password = db.Column(db.String, nullable=False)
|
|
joined_at = db.Column(
|
|
db.DateTime,
|
|
nullable=False,
|
|
server_default=db.func.now(), # pylint: disable=E1102
|
|
)
|
|
|
|
posts = db.relationship("Post", backref="author")
|
|
groups = db.relationship("Group", backref="author")
|
|
|
|
def get_id(self):
|
|
return str(self.alt_id)
|