def select(self, id):
artist = None
sql = "SELECT * FROM artists WHERE id = %s"
results = SqlRunner.run(sql, (id, ))
row = results[0]
artist = Artist(row['name'], row['id'])
return artist
def select(self, id):
album = None
sql = "SELECT * FROM albums WHERE id = %s"
results = SqlRunner.run(sql,(id,))
row = results[0]
artist = artist_repository.select(row['artist_id'])
album = Album(row['title'], artist, row['quantity'], row['id'])
return album
def select_all(self):
artists = []
sql = "SELECT * FROM artists"
results = SqlRunner.run(sql)
for row in results:
artist = Artist(row['name'], row['id'])
artists.append(artist)
return artists
def select_for_artist(self, id):
artist = artist_repository.select(id)
albums = []
sql = "SELECT * FROM albums where artist_id = %s"
results = SqlRunner.run(sql,(id,))
for row in results:
album = Album(row['title'], artist, row['quantity'], row['id'])
albums.append(album)
return albums
def select_all(self):
albums = []
sql = "SELECT * FROM albums"
results = SqlRunner.run(sql)
for row in results:
artist = artist_repository.select(row['artist_id'])
album = Album(row['title'], artist, row['quantity'], row['id'])
albums.append(album)
return albums
def delete(self, id):
sql = "DELETE FROM albums WHERE id = %s"
SqlRunner.run(sql,(id,)).count
def update(self, album):
sql = "UPDATE albums SET (title, artist_id, quantity) = (%s, %s, %s) WHERE id = %s"
SqlRunner.run(sql,(album.title, album.artist.id, album.quantity, album.id))
def save(self, album):
sql = "INSERT INTO albums (title, artist_id, quantity) VALUES (%s, %s, %s) RETURNING id"
results = SqlRunner.run(sql, (album.title, album.artist.id, album.quantity))
id = results[0]['id']
album.id = id
return album
def update(self, artist):
sql = "UPDATE artists SET (name) = (%s) WHERE id = %s"
SqlRunner.run(sql, (artist.name, artist.id))
def save(self, artist):
sql = "INSERT INTO artists (name) VALUES (%s) RETURNING id"
results = SqlRunner.run(sql, (artist.name, ))
id = results[0]['id']
artist.id = id
return artist