class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
password = db.Column(db.String(60), nullable=False)
image_file = db.Column(db.String(20),
nullable=False,
default='profile.png')
posts = db.relationship('Post', backref='author', lazy=True)
menu = db.relationship('Menu', backref='author', lazy=True)
complaints = db.relationship('Complaint', backref='author', lazy=True)
newcomplaints = db.relationship('NewComplaint',
backref='author',
lazy=True)
def __repr__(self):
return "User( '{}', '{}', '{}')".format(self.username, self.email,
self.password)
class Complaint(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), nullable=False)
date_posted = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
content = db.Column(db.Text, nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
def __repr__(self):
return "Post('{}', '{}')".format(self.title, self.date_posted)