def change_availability(): artist_name = ui.get_artist_name() artist_id = artwork_store._get_artist_id(artist_name) artworks_list = Artwork.get_all_artist_artwork(artist_id) if artworks_list: # If artwork ID found in db artwork_name = ui.get_artwork_name() artwork_to_change = Artwork.get_artwork_by_name(artwork_name) availability_status = ui.get_artwork_availability() # if else statement to determine if for sale or not if not availability_status: sale_status = '"Sold"' else: sale_status = '"For Sale"' artwork_to_change.for_sale = availability_status artwork_to_change.update_artwork() print('Artwork status has changed to', sale_status) else: # If artwork ID not found in db print('That artist has no artworks.') option = input( 'Return to main menu? Y for main menu or Enter to quit. ').upper() if option == 'Y': print() else: quit_program()
def show_all_artist_artwork(): artist_name = ui.get_artist_name() artist_id = artwork_store._get_artist_id(artist_name) artworks = Artwork.get_all_artist_artwork(artist_id) ui.show_artworks_by_artist(artworks)