def add_users(cursor: RealDictCursor, users_data):
f = IteratorFile(("{}\t{}\t{}\t{}".format(*user) for user in users_data))
cursor.copy_from(
f,
'user_account',
columns=(['first_name', 'last_name', 'email', 'address_id']))
def add_albums(cursor: RealDictCursor, albums_data):
f = IteratorFile(
("{}\t{}\t{}\t{}".format(*album) for album in albums_data))
cursor.copy_from(f,
'album',
columns=(['name', 'date', 'studio_id', 'band_id']))
def add_bands(cursor: RealDictCursor, bands_amount: int, bands):
f = IteratorFile(("{}\t{}".format(bands[randint(1,
len(bands) - 1)],
get_random_date())
for _ in range(bands_amount)))
cursor.copy_from(f, 'band', columns=(['name', 'band_establishment']))
def add_subscriptions(cursor: RealDictCursor, subscriptions):
f = IteratorFile(
("{}\t{}\t{}".format(*subscription) for subscription in subscriptions))
cursor.copy_from(f,
'subscription',
columns=(['date', 'user_id', 'band_id']))
def add_playlists(cursor: RealDictCursor, playlists_amount: int, playlists):
f = IteratorFile(
("{}\t{}".format(playlists[randint(1,
len(playlists) - 1)],
get_random_date()) for _ in range(playlists_amount)))
cursor.copy_from(f, 'playlist', columns=(['name', 'date']))
def add_artists(cursor: RealDictCursor, artists_data):
f = IteratorFile(
("{}\t{}\t{}\t{}".format(*artist) for artist in artists_data))
cursor.copy_from(
f,
'artist',
columns=(['first_name', 'last_name', 'gender', 'instrument']))
def add_songs(cursor: RealDictCursor, songs_data):
f = IteratorFile(
("{}\t{}\t{}\t{}\t{}".format(*song) for song in songs_data))
cursor.copy_from(
f,
'song',
columns=(['name', 'length', 'views', 'album_id', 'genre_id']))
def add_addresses(cursor: RealDictCursor, addresses_amount: int,
countries_amount: int, cities_amount: int,
street_amount: int):
max_local_number = 1000
f = IteratorFile(("{}\t{}\t{}\t{}".format(randint(1, street_amount),
randint(1, max_local_number),
randint(1, cities_amount),
randint(1, countries_amount))
for _ in range(addresses_amount)))
cursor.copy_from(
f,
'address',
columns=(['street_id', 'local_number', 'city_id', 'country_id']))
def add_studios(cursor: RealDictCursor, studios_data):
f = IteratorFile(("{}\t{}".format(*studio) for studio in studios_data))
cursor.copy_from(f, 'studio', columns=(['name', 'address_id']))
def add_streets(cursor: RealDictCursor, streets):
f = IteratorFile(("{}".format(x) for x in streets))
cursor.copy_from(f, 'street', columns=(['name']))
def add_genres(cursor: RealDictCursor, genres):
f = IteratorFile(("{}".format(x) for x in genres))
cursor.copy_from(f, 'genre', columns=(['type']))
def add_countries(cursor: RealDictCursor, countries):
f = IteratorFile(("{}".format(x) for x in countries))
cursor.copy_from(f, 'country', columns=(['name']))
def add_members_to_bands(cursor: RealDictCursor, bands_members):
f = IteratorFile(
("{}\t{}".format(*band_member) for band_member in bands_members))
cursor.copy_from(f, 'artist_band', columns=(['artist_id', 'band_id']))
def add_songs_to_playlists(cursor: RealDictCursor, songs_playlists):
f = IteratorFile(
("{}\t{}".format(*user_playlist) for user_playlist in songs_playlists))
cursor.copy_from(f, 'song_playlist', columns=(['song_id', 'playlist_id']))
def add_user_playlists(cursor: RealDictCursor, user_playlists):
f = IteratorFile(
("{}\t{}".format(*user_playlist) for user_playlist in user_playlists))
cursor.copy_from(f, 'user_playlist', columns=(['user_id', 'playlist_id']))
