mirror of
https://github.com/Xevion/runnerspace.git
synced 2025-12-06 05:16:23 -06:00
987e4f3256
- As well as related methods - In preparation for PostLikes and CommentLikes model implementations
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
import datetime
|
|
import json
|
|
from typing import List
|
|
|
|
import humanize
|
|
from flask_login import UserMixin
|
|
from sqlalchemy import func
|
|
|
|
from database import db
|
|
|
|
MAXIMUM_ONLINE_DELTA = datetime.timedelta(minutes=1)
|
|
|
|
|
|
class User(UserMixin, db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
username = db.Column(db.String(100), unique=True)
|
|
password = db.Column(db.String(100))
|
|
name = db.Column(db.String(1000))
|
|
about_me = db.Column(db.String(5000), nullable=True)
|
|
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_ip = db.Column(db.String(64), nullable=True)
|
|
is_admin = db.Column(db.Boolean, default=False)
|
|
posts = db.relationship("Post", backref='author')
|
|
comments = db.relationship("Comment", backref='author')
|
|
|
|
def get_last_seen(self) -> str:
|
|
delta: datetime.timedelta = datetime.datetime.utcnow() - self.last_seen
|
|
if delta > MAXIMUM_ONLINE_DELTA:
|
|
return f'Last seen {humanize.naturaldelta(delta)} ago'
|
|
return 'Online now!'
|
|
|
|
def get_registration_delta(self) -> str:
|
|
delta: datetime.timedelta = datetime.datetime.utcnow() - self.time_registered
|
|
return humanize.naturaldelta(delta)
|
|
|
|
def get_post_count(self) -> int:
|
|
return len(self.posts)
|
|
|
|
def display_about(self) -> str:
|
|
return self.about_me or "This user hasn't written a bio yet."
|
|
|
|
|
|
class Post(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
author_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
|
|
text = db.Column(db.Text)
|
|
date_posted = db.Column(db.DateTime, server_default=func.now())
|
|
date_updated = db.Column(db.DateTime, nullable=True)
|
|
comments = db.relationship("Comment", backref='post')
|
|
|
|
def get_time_ago(self) -> str:
|
|
delta: datetime.timedelta = datetime.datetime.utcnow() - self.date_posted
|
|
return humanize.naturaldelta(delta)
|
|
|
|
|
|
class Comment(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
text = db.Column(db.Text, nullable=False)
|
|
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
|
|
post_id = db.Column(db.Integer, db.ForeignKey('post.id'), nullable=False)
|
|
date_posted = db.Column(db.DateTime, server_default=func.now())
|