models.py

from app import app, db

#the User model: each user has a username, and a playlist_id foreign key referring
#to the user's Playlist
class User(db.Model):
  id = db.Column(db.Integer, primary_key = True)
  username = db.Column(db.String(50), index = True, unique = True) 
  playlist_id = db.Column(db.Integer,  db.ForeignKey('playlist.id'))
  
  #representation method
  def __repr__(self):
        return "{}".format(self.username)

#create the Song model here + add a nice representation method
class Song(db.Model):
  id = db.Column(db.Integer, primary_key = True)
  artist = db.Column(db.String(30), index = True, unique = False)
  title = db.Column(db.String(30), index = True, unique = False)
  n = db.Column(db.Integer, index = False, unique = False)
  def __repr__(self):
        return "{} by {}".format(self.title,self.artist)


    
#create the Item model here + add a nice representation method
class Item(db.Model):
  id = db.Column(db.Integer, primary_key = True)
  song_id = db.Column(db.Integer, db.ForeignKey('song.id'))
  playlist_id = db.Column(db.Integer, db.ForeignKey('playlist.id'))
  #def __repr__(self):
  #  return "{} song is in {}'s playlist".format(Song.query.get(song_id=self.song_id).title, User.query.get(playlist_id=self.playlist_id).username)
    
#create the Playlist model here + add a nice representation method
class Playlist(db.Model):
  id = db.Column(db.Integer, primary_key = True)
  items = db.relationship('Item', backref='playlist', lazy='dynamic')
  #def __repr__(self):
  #  return "A {}'s playlist".format(User.query.get(playlist_id=self.id).username)


#adding this comment just to test a push from this cloned repository