59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
from app import db
|
|
from datetime import datetime
|
|
|
|
|
|
class Post(db.Model):
|
|
"""Contents created and published by Users."""
|
|
|
|
__tablename__ = 'post'
|
|
|
|
# Unique Post ID for the whole site
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
# Post type (polymorphic discriminator)
|
|
type = db.Column(db.String(20))
|
|
|
|
# Creation and edition date
|
|
date_created = db.Column(db.DateTime)
|
|
date_modified = db.Column(db.DateTime)
|
|
|
|
# Post author
|
|
author_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
|
|
author = db.relationship('User', backref="posts", foreign_keys=author_id)
|
|
|
|
# TODO: Post attachments?
|
|
|
|
__mapper_args__ = {
|
|
'polymorphic_identity': __tablename__,
|
|
'polymorphic_on': type
|
|
}
|
|
|
|
def __init__(self, author):
|
|
"""
|
|
Create a new Post.
|
|
|
|
Arguments:
|
|
author -- post author (User)
|
|
"""
|
|
|
|
self.author = author
|
|
self.date_created = datetime.now()
|
|
self.date_modified = self.date_created
|
|
|
|
def touch(self):
|
|
"""Touch a Post when it is edited."""
|
|
|
|
self.date_modified = datetime.now()
|
|
|
|
def change_ownership(self, new_author):
|
|
"""
|
|
Change ownership of a Post. This is a privileged operation!
|
|
|
|
Arguments:
|
|
new_author -- new post author (User)
|
|
"""
|
|
|
|
self.author = new_author
|
|
|
|
def __repr__(self):
|
|
return f'<Post: #{self.id}>'
|