xevion/runnerspace
1
0
Fork 0
mirror of https://github.com/Xevion/runnerspace.git synced 2025-12-18 12:13:11 -06:00
Code Issues Packages Projects Releases Wiki Activity

Add Comment model & add relevant relationships between all models

Browse Source
This commit is contained in:
Xevion
2022-03-27 09:40:23 -05:00
parent 23fd613d8f
commit 5ebbbd1b03
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
models.py
View File

@@ -7,6 +7,7 @@ from flask_login import UserMixin
from sqlalchemy import func from sqlalchemy import func
from .create_app import db from .create_app import db
MAXIMUM_ONLINE_DELTA = datetime.timedelta(minutes=5) MAXIMUM_ONLINE_DELTA = datetime.timedelta(minutes=5)
@@ -19,6 +20,8 @@ class User(UserMixin, db.Model):
time_registered = db.Column(db.DateTime, nullable=False, server_default=func.now()) time_registered = db.Column(db.DateTime, nullable=False, server_default=func.now())
last_seen = db.Column(db.DateTime, nullable=False, server_default=func.now()) last_seen = db.Column(db.DateTime, nullable=False, server_default=func.now())
last_ip = db.Column(db.String(64), nullable=True) last_ip = db.Column(db.String(64), nullable=True)
posts = db.relationship("Post", backref='author')
comments = db.relationship("Comment", backref='author')
def get_last_seen(self) -> str: def get_last_seen(self) -> str:
delta: datetime.timedelta = datetime.datetime.utcnow() - self.last_seen delta: datetime.timedelta = datetime.datetime.utcnow() - self.last_seen
@@ -33,11 +36,12 @@ class User(UserMixin, db.Model):
class Post(db.Model): class Post(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
author = db.Column(db.Integer) author = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
text = db.Column(db.Text) text = db.Column(db.Text)
date_posted = db.Column(db.DateTime, server_default=func.now()) date_posted = db.Column(db.DateTime, server_default=func.now())
date_updated = db.Column(db.DateTime, nullable=True) date_updated = db.Column(db.DateTime, nullable=True)
likes = db.Column(db.Text, default='[]') likes = db.Column(db.Text, default='[]')
comments = db.relationship("Comment", backref='post')
def get_likes(self) -> List[int]: def get_likes(self) -> List[int]:
"""Return the IDs of the Users who have liked this post.""" """Return the IDs of the Users who have liked this post."""
@@ -53,3 +57,10 @@ class Post(db.Model):
if user_id not in likes: if user_id not in likes:
likes.append(user_id) likes.append(user_id)
self.set_likes(likes) self.set_likes(likes)
class Comment(db.Model):
id = db.Column(db.Integer, primary_key=True)
text = db.Column(db.Text, nullable=False)
author = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
post = db.Column(db.Integer, db.ForeignKey('post.id'), nulllable=False)