class Saves(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
posts_id = db.Column(db.Integer, db.ForeignKey('posts.id'), nullable=False)
def __repr__(self):
return f"/post/{self.posts_id}"
class Book(db.Model):
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
title = db.Column(db.String(80), nullable=False)
description = db.Column(db.Text())
isbn = db.Column(db.String(13))
def __repr__(self):
return f"Book('{self.title}', '{self.isbn}')"
class Comments(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
comment_text = db.Column(db.Text, nullable=False)
comment_time = db.Column(db.DateTime, nullable=False, default=datetime.now)
posts_id = db.Column(db.Integer, db.ForeignKey('posts.id'), nullable=False)
def __repr__(self):
return f"'{self.comment_text},')"
class Notifications(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
post_id = db.Column(db.Integer, db.ForeignKey('posts.id'), nullable=False)
comment_id = db.Column(db.Integer,
db.ForeignKey('comments.id'),
nullable=False)
seen = db.Column(db.Boolean, default=False)
def __repr__(self):
return f"{self.post_id}"
class AuthorModel(db.Model):
__tablename__ = 'authors'
author_id = db.Column(db.Integer, primary_key=True)
first_name = db.Column(db.String(25), nullable=False)
last_name = db.Column(db.String(25), nullable=False)
books = db.relationship('BookModel', backref='author', lazy=True)
@classmethod
def find_by_id(cls, author_id: int) -> 'AuthorModel':
return cls.query.filter_by(author_id=author_id).first()
def save_to_db(self) -> None:
db.session.add(self)
db.session.commit()
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)
image_file = db.Column(db.String(20),
nullable=False,
default='default.jpg')
password = db.Column(db.String(60), nullable=False)
payment_profile = db.Column(db.String(60))
major = db.Column(db.String(20))
posts = db.relationship('Posts', backref='author', lazy=True)
#notifications = db.Column(db.Integer, db.ForeignKey('posts.id'))
def get_reset_token(self, expires_sec=1800):
s = Serializer(app.config['SECRET_KEY'], expires_sec)
return s.dumps({'user_id': self.id}).decode('utf-8')
@staticmethod
def verify_reset_token(token):
s = Serializer(app.config['SECRET_KEY'])
try:
user_id = s.loads(token)['user_id']
except:
return None
return User.query.get(user_id)
def __repr__(self):
return f"User('{self.username}', '{self.email}', '{self.image_file}')"
class PublisherModel(db.Model):
__tablename__ = 'publishers'
publisher_id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(250), nullable=False)
books = db.relationship('BookModel', backref='publisher', lazy=True)
@classmethod
def find_by_id(cls, id: int) -> "PublisherModel":
return cls.query.filter_by(publisher_id=id).first()
def save_to_db(self) -> None:
db.session.add(self)
db.session.commit()
def delete_from_db(self) -> None:
db.session.delete(self)
db.session.commit()
class BookModel(db.Model):
__tablename__ = 'books'
book_id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), nullable=False)
pages = db.Column(db.Integer, nullable=False)
author_id = db.Column(db.Integer,
db.ForeignKey('authors.author_id'),
nullable=False)
publisher_id = db.Column(db.Integer,
db.ForeignKey('publishers.publisher_id'),
nullable=False)
year_published = db.Column(db.Integer, nullable=False)
@classmethod
def find_by_author_id(cls, author_id: int) -> "BookModel":
return cls.query.filter_by(author_id=author_id).all()
@classmethod
def find_by_id(cls, id: int) -> "BookModel":
return cls.query.filter_by(book_id=id).first()
def save_to_db(self) -> None:
db.session.add(self)
db.session.commit()
def delete_from_db(self) -> None:
db.session.delete(self)
db.session.commit()
class Posts(db.Model):
id = db.Column(db.Integer, primary_key=True)
isbn = db.Column(db.String(120), nullable=False)
date_posted = db.Column(db.DateTime, nullable=False, default=datetime.now)
title = db.Column(db.Text, nullable=False)
publisher = db.Column(db.Text, nullable=True)
writers = db.Column(db.Text)
image_ref = db.Column(db.String(120),
default="http://127.0.0.1:5000/static/book.jpg")
condition = db.Column(db.Text, nullable=False)
price = db.Column(db.String(60), nullable=False)
major = db.Column(db.String(20))
edition = db.Column(db.String(20))
binding = db.Column(db.String(20))
#comments = db.Column(db.Integer, default=0)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
def __repr__(self):
return f"Post('{self.title},')"
