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)
songs = db.relationship("Songs", backref="author", lazy=True)
liked = db.relationship("SongLike",
foreign_keys="SongLike.user_id",
backref="user",
lazy="dynamic")
comments = db.relationship("Comments",
foreign_keys="Comments.user_name",
backref="user",
lazy="dynamic")
def like(self, songs):
if not self.is_liking(songs):
like = SongLike(user_id=self.id, song_id=songs.id)
db.session.add(like)
def unlike(self, songs):
if self.is_liking(songs):
SongLike.query.filter_by(user_id=self.id,
song_id=songs.id).delete()
def is_liking(self, songs):
return (SongLike.query.filter(SongLike.user_id == self.id,
SongLike.song_id == songs.id).count() >
0)
def __repr__(self):
return f"User('{self.username}', '{self.email}', '{self.image_file}')"
class Comments(db.Model):
__tablename__ = 'comments'
id = db.Column(db.Integer, primary_key=True)
user_name = db.Column(db.String(50), db.ForeignKey('user.username'))
song_id = db.Column(db.Integer, db.ForeignKey('songs.id'))
comment = db.Column(db.Text, nullable=False)
timestamp = db.Column(db.Float)
date_posted = db.Column(db.DateTime, nullable=False,
default=datetime.utcnow)
class Songs(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)
image_file = db.Column(db.String(100), nullable=False,
default='default.jpg')
filepath = db.Column(db.String(100), nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
likes = db.relationship('SongLike', backref='songs', lazy='dynamic')
comments = db.relationship('Comments', backref='songs', lazy='dynamic')
def __repr__(self):
return f"Songs('{self.title}', '{self.date_posted}')"
class PlayCount(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_name = db.Column(db.Integer, db.ForeignKey("user.id"))
song_id = db.Column(db.Integer, db.ForeignKey("songs.id"))
class SongLike(db.Model):
__tablename__ = "song_like"
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey("user.id"))
song_id = db.Column(db.Integer, db.ForeignKey("songs.id"))
