def add_likes():
cursor, conn = open_connection()
cursor.execute('SELECT count(*) FROM "user";')
length_user = cursor.fetchone()[0]
cursor.execute('SELECT id FROM song;')
id_song = cursor.fetchall()
cursor.execute('SELECT id FROM album;')
id_album = cursor.fetchall()
cursor.execute('SELECT email FROM "user";')
id_user = cursor.fetchall()
cursor.execute('SELECT name FROM "group";')
id_group = cursor.fetchall()
cursor.execute('SELECT id FROM playlist;')
id_playlist = cursor.fetchall()
for i in range(0, int(length_user / 4)):
cursor.execute(
'insert into like_song ("user",song) VALUES (%s, %s);commit;',
(random.choice(id_user), random.choice(id_song)))
cursor.execute(
'insert into like_album ("user",album) VALUES (%s, %s);commit;',
(random.choice(id_user), random.choice(id_album)))
cursor.execute(
'insert into like_group ("user","group") VALUES (%s, %s);commit;',
(random.choice(id_user), random.choice(id_group)))
cursor.execute(
'insert into like_playlist ("user",playlist) VALUES (%s, %s);commit;',
(random.choice(id_user), random.choice(id_playlist)))
close_connection(cursor, conn)
def add_users(n):
cursor, conn = open_connection()
gen_users = UserGenerator()
gen_users.get_primary_key_data(n)
for u in gen_users.data:
params = gen_users.get_params()
print(u, params, sep=': ')
cursor.execute(
'insert into "user" (email, name, icon_path, password) values(%s, %s, %s, %s);commit;',
(u, params.get('name'), params.get('icon_path'),
params.get('password')))
print(n, "Records added", sep=' ')
close_connection(cursor, conn)
def add_groups(n):
cursor, conn = open_connection()
gen_groups = GroupGenerator()
gen_groups.get_primary_key_data(n)
for g in gen_groups.data:
params = gen_groups.get_params()
print(g, params, sep=': ')
cursor.execute(
'insert into "group"(name, "desc", country, icon_path) values(%s, %s, %s, %s);commit;',
(g, params.get('desc'), params.get('country'),
params.get('icon_path')))
print(n, "Records added", sep=' ')
close_connection(cursor, conn)
def add_artists(n):
cursor, conn = open_connection()
gen_artists = ArtistGenerator()
gen_artists.get_primary_key_data(n)
for a in gen_artists.data:
params = gen_artists.get_params()
print(a, params, sep=': ')
cursor.execute(
'insert into artist (name, "desc", country, icon_path) values (%s, %s, %s, %s );commit;',
(a, params.get('desc'), params.get('country'),
params.get('icon_path')))
print(n, "Artists added", sep=' ')
close_connection(cursor, conn)
def add_songs(n):
cursor, conn = open_connection()
song_generator = SongGenerator()
data = song_generator.get_primary_key_data(n)
for s in data:
params = song_generator.get_params()
print('id: ', s, params, sep=': ')
cursor.execute(
'insert into song (id, name, path, album, year) values (%s, %s, %s, %s, %s);commit;',
(s, params.get('name'), params.get('path'), params.get('album'),
params.get('year')))
print(n, "Songs added", sep=' ')
close_connection(cursor, conn)
def add_prizers(n):
cursor, conn = open_connection()
gen_prizers = PrizeGenerator()
data = gen_prizers.get_primary_key_data(n)
for id in data:
params = gen_prizers.get_params()
print('id: ', id, params, sep=': ')
cursor.execute(
'insert into prize (id, name, year, description) values (%s, %s, %s, %s);commit;',
(id, params.get('name'), params.get('year'),
params.get('description')))
print(n, "Prizes added", sep=' ')
close_connection(cursor, conn)
def add_playlists_table():
cursor, conn = open_connection()
cursor.execute('SELECT id FROM playlist;')
id_playlist = cursor.fetchall()
cursor.execute('SELECT id FROM song;')
id_song = cursor.fetchall()
cursor.execute('SELECT count(*) FROM song;')
length_song = cursor.fetchone()[0]
for i in range(0, int(length_song / 4)):
cursor.execute(
'insert into playlists_table (playlist, song) VALUES (%s, %s);',
(random.choice(id_playlist), random.choice(id_song)))
close_connection(cursor, conn)
def add_playlists(n):
cursor, conn = open_connection()
gen_playlist = PlaylistGenerator()
data = gen_playlist.get_primary_key_data(n)
for d in data:
params = gen_playlist.get_params()
print('id: ', d, params, sep=': ')
cursor.execute(
'insert into playlist (id, name, quantity, author, duration, icon_path) VALUES (%s, %s, %s, %s, %s, '
'%s);commit;', (d, params.get('name'), params.get('quantity'),
params.get('author'), params.get('duration'),
params.get('icon_path')))
print(n, "Playlists added", sep=' ')
close_connection(cursor, conn)
def add_albums(n):
cursor, conn = open_connection()
gen_albums = AlbumGenerator()
data = gen_albums.get_primary_key_data(n)
for d in data:
params = gen_albums.get_params()
print('id: ', d, params, sep=': ')
cursor.execute(
'insert into album (id, name, year, duration, quantity, icon_path, type) values (%s, %s, %s, %s, %s, %s, %s);commit;',
(d, params.get('name'), params.get('year'), params.get('duration'),
params.get('quantity'), params.get('icon_path'),
params.get('type')))
print(n, "Albums added", sep=' ')
close_connection(cursor, conn)
def add_groups_table():
cursor, conn = open_connection()
cursor.execute('SELECT count(*) FROM song;')
length_song = cursor.fetchone()[0]
cursor.execute('SELECT id FROM song;')
id_song = cursor.fetchall()
cursor.execute('SELECT name FROM "group";')
id_group = cursor.fetchall()
for i in range(0, int(length_song / 4)):
cursor.execute(
'insert into groups_table (creator, song) VALUES (%s, %s);commit;',
(random.choice(id_group), random.choice(id_song)))
close_connection(cursor, conn)
def add_history_artist(n):
cursor, conn = open_connection()
cursor.execute('SELECT count(*) FROM artist;')
artist_length = cursor.fetchone()[0]
cursor.execute('SELECT name FROM artist;')
id_artist = cursor.fetchall()
cursor.execute('SELECT name FROM "group";')
id_group = cursor.fetchall()
for i in range(0, int(artist_length / 4)):
cursor.execute(
'insert into history_artist_table (artist, "group", start_date, end_date) values (%s, %s, %s, %s);commit;',
(random.choice(id_artist), random.choice(id_group),
random.randint(1500, 2020), random.randint(1500, 2020)))
print(n, "History added", sep=' ')
close_connection(cursor, conn)
class SongGenerator(BaseGenerator):
cursor, conn = open_connection()
cursor.execute('SELECT id FROM album;')
id_albums = cursor.fetchall()
close_connection(cursor, conn)
def get_primary_key_data(self, num):
return range(0, num - 1)
def get_params(self):
song = {
'name': self.person.first_name(),
'year': self.generic.datetime.year(),
'path': self.generic.path.project_dir(),
}
if bool(random.getrandbits(1)):
if bool(self.id_albums) is True:
song.update({'album': random.choice(self.id_albums)})
else:
song.update({'album': None})
print(song)
return song