Пример #1
0
 def add_to_database(self, filename, song_info):
     # Just in case anything is None
     if song_info.title == None:
         r_title = ''
     else:
         r_title = song_info.title
     if song_info.artist == None:
         r_artist_name = ''
     else:
         r_artist_name = song_info.artist
     if song_info.album == None:
         r_album_name = ''
     else:
         r_album_name = song_info.album
     if song_info.length == None:
         r_length = 0
     else:
         r_length = song_info.length
     self.get_logger().error('Adding to database: %s by %s in %s', r_title, r_artist_name, r_album_name)
     # Ensure artist
     artist = self.ensure_or_create_artist(r_artist_name)
     ensure_artist_image(artist.id, r_artist_name)
     # Ensure album
     album = self.ensure_or_create_album(filename, r_artist_name, r_album_name)
     ensure_album_image(album.id, filename, r_album_name, r_artist_name)
     # Create song in database
     song = Song(filename, r_title, artist.id, album.id, r_length, song_info.year, song_info.track, song_info.disc_no)
     db.session.add(song)
     db.session.commit()
Пример #2
0
 def replace_in_database(self, filename, s, song_info):
     # Just in case anything is None
     if song_info.title == None:
         r_title = ''
     else:
         r_title = song_info.title
     if song_info.artist == None:
         r_artist_name = ''
     else:
         r_artist_name = song_info.artist
     if song_info.album == None:
         r_album_name = ''
     else:
         r_album_name = song_info.album
     if song_info.length == None:
         r_length = 0
     else:
         r_length = song_info.length
     # Save previous artist and album
     prev_artist_id = s.artistId
     prev_album_id = s.albumId
     # Ensure artist
     artist = self.ensure_or_create_artist(r_artist_name)
     ensure_artist_image(artist.id, r_artist_name)
     # Ensure album
     album = self.ensure_or_create_album(filename, r_artist_name, r_album_name)
     ensure_album_image(album.id, filename, r_album_name, r_artist_name)
     # Change song object
     s.title = r_title
     s.artistId = artist.id
     s.albumId = album.id
     s.length = r_length
     s.year = song_info.year
     s.track = song_info.track
     s.disc_no = song_info.disc_no
     db.session.commit()
     # Delete previous artist and album if not needed
     self.delete_artist_if_no_assoc(prev_artist_id)
     self.delete_album_if_no_assoc(prev_album_id)