Lossy will be set to Wanted Lossless during processing

Author: el133

headphones/postprocessor.py

@@ -405,12 +405,15 @@ def doPostProcessing(albumid, albumpath, release, tracks, downloaded_track_list,
     # Need to update the downloaded track list with the new location.
     # Could probably just throw in the "headphones-modified" folder,
     # but this is good to make sure we're not counting files that may have failed to move
+
     if new_folder:
         downloaded_track_list = []
         for r, d, f in os.walk(albumpath):
             for files in f:
                 if any(files.lower().endswith('.' + x.lower()) for x in headphones.MEDIA_FORMATS):
                     downloaded_track_list.append(os.path.join(r, files))
+    
+    all_files_lossless = all(file.lower().endswith(tuple('.' + format.lower() for format in headphones.LOSSLESS_MEDIA_FORMATS)) for file in downloaded_track_list)
 
     builder = metadata.AlbumMetadataBuilder()
     # Check if files are valid media files and are writable, before the steps
@@ -504,11 +507,19 @@ def doPostProcessing(albumid, albumpath, release, tracks, downloaded_track_list,
     if headphones.CONFIG.FILE_PERMISSIONS_ENABLED:
         updateFilePermissions(albumpaths)
 
-    myDB = db.DBConnection()
-    myDB.action('UPDATE albums SET status = "Downloaded" WHERE AlbumID=?', [albumid])
-    myDB.action(
-        'UPDATE snatched SET status = "Processed" WHERE Status NOT LIKE "Seed%" and AlbumID=?',
-        [albumid])
+    # If the downloaded tracks are lossy, but lossless is wanted, set status to wanted lossless
+    if not all_files_lossless and headphones.CONFIG.PREFERRED_QUALITY == 1:
+        myDB = db.DBConnection()
+        myDB.action('UPDATE albums SET status = "Wanted Lossless" WHERE AlbumID=?', [albumid])
+        myDB.action(
+            'UPDATE snatched SET status = "Processed" WHERE Status NOT LIKE "Seed%" and AlbumID=?',
+            [albumid])
+    else:
+        myDB = db.DBConnection()
+        myDB.action('UPDATE albums SET status = "Downloaded" WHERE AlbumID=?', [albumid])
+        myDB.action(
+            'UPDATE snatched SET status = "Processed" WHERE Status NOT LIKE "Seed%" and AlbumID=?',
+            [albumid])
 
     # Check if torrent has finished seeding
     if headphones.CONFIG.TORRENT_DOWNLOADER != 0:

headphones/soulseek.py

@@ -11,7 +11,10 @@
 def initialize_soulseek_client():
     host = headphones.CONFIG.SOULSEEK_API_URL
     api_key = headphones.CONFIG.SOULSEEK_API_KEY
-    return slskd_api.SlskdClient(host=host, api_key=api_key)
+    try:
+        return slskd_api.SlskdClient(host=host, api_key=api_key)
+    except:
+        logger.info("Something went wrong while connecting to the soulseek client")
 
     # Search logic, calling search and processing fucntions
 def search(artist, album, year, num_tracks, losslessOnly):
@@ -24,14 +27,14 @@ def search(artist, album, year, num_tracks, losslessOnly):
         return processed_results
 
     # Stage 2: If Stage 1 fails, search with artist, album, and num_tracks (excluding year)
-    logger.info("Soulseek search stage 1 did not meet criteria. Retrying without year...")
+    logger.debug("Soulseek search stage 1 did not meet criteria. Retrying without year...")
     results = execute_search(client, artist, album, None, losslessOnly)
     processed_results = process_results(results, losslessOnly, num_tracks)
     if processed_results:
         return processed_results
 
     # Stage 3: Final attempt, search only with artist and album
-    logger.info("Soulseek search stage 2 did not meet criteria. Final attempt with only artist and album.")
+    logger.debug("Soulseek search stage 2 did not meet criteria. Final attempt with only artist and album.")
     results = execute_search(client, artist, album, None, losslessOnly)
     processed_results = process_results(results, losslessOnly, num_tracks, ignore_track_count=True)
 
@@ -42,9 +45,8 @@ def execute_search(client, artist, album, year, losslessOnly):
     if year:
         search_text += f" {year}"
     if losslessOnly:
-        search_text += ".flac"
+        search_text += " .flac"
 
-    # Actual search
     search_response = client.searches.search_text(searchText=search_text, filterResponses=True)
     search_id = search_response.get('id')
 
@@ -112,7 +114,6 @@ def download(user, filelist):
     client = initialize_soulseek_client()
     client.transfers.enqueue(username=user, files=filelist)
 
-
 def download_completed():
     client = initialize_soulseek_client()
     all_downloads = client.transfers.get_all_downloads(includeRemoved=False)
@@ -166,10 +167,7 @@ def download_completed():
                         print(f"Failed to cancel download for file ID: {file_id}")
 
     # Clear completed/canceled/errored stuff from client downloads
-    try:
-        client.transfers.remove_completed_downloads()
-    except Exception as e:
-        print(f"Failed to remove completed downloads: {e}")
+    client.transfers.remove_completed_downloads()
 
     # Identify completed albums
     completed_albums = {album for album, counts in album_completion_tracker.items() if counts['total'] == counts['completed']}