Added file's index

main.py

@@ -7,7 +7,9 @@ from pathlib import Path
 from typing import Any
 
 from files.util import scan_folder, accepted_file_type
-from service.provider import search_song
+from models.track import TrackInfo
+from service import index
+from service.provider import search_song, SongRecognizeError
 
 logger = logging.getLogger(__name__)
 
@@ -16,29 +18,42 @@ def arg_parser():
     parser = argparse.ArgumentParser(description="Organize music from source folder to destination folder using Shazam")
     parser.add_argument("--src", "-s", metavar="SRC", type=str, help="source folder to scan")
     parser.add_argument("--dest", "-d", metavar="DEST", type=str, help="where it copy files")
+    parser.add_argument("--no-index", "-i", type=bool, default=False, help="disable music indexed")
     return parser.parse_args()
 
 
-async def main(src, dst) -> Any:
-    for song_file in scan_folder(src):
-        if accepted_file_type(song_file):
-            print(f"Found file {song_file}")
-            track_info = await search_song(song_file)
-            logging.info(f"Recognize file {song_file} as {track_info}")
-            album_path: str = os.path.join(dst, track_info.artist,
-                                           f"{track_info.album.released} - {track_info.album.name}")
-            if not os.path.exists(album_path):
-                Path(album_path).mkdir(mode=0o755, parents=True, exist_ok=True)
-
-            # I can copy to file now
-            shutil.copy(song_file,
-                        os.path.join(album_path, f"{track_info.track_number} - {track_info.title}.mp3"))
+async def main(options) -> Any:
+    for song_file in scan_folder(options.src):
+        if not accepted_file_type(song_file):
+            logger.info(f"Skipped file {song_file}")
         else:
-            print(f"Skipped file {song_file}")
+            logger.info(f"Found file {song_file}")
+            try:
+                if index.duplicated(song_file):
+                    logger.info(f"{song_file} already collected")
+                    continue
+
+                track_info: TrackInfo = await search_song(song_file)
+                logging.info(f"Recognize file {song_file}:\n{track_info}")
+                album_path: str = os.path.join(options.dest, track_info.artist,
+                                               f"{track_info.album.released} - {track_info.album.name}")
+                if not os.path.exists(album_path):
+                    Path(album_path).mkdir(mode=0o755, parents=True, exist_ok=True)
+
+                destination_path: str = os.path.join(album_path, f"{track_info.track_number} - {track_info.title}.mp3")
+                logger.debug(f"Copy {os.path.basename(song_file)} in {destination_path}")
+                shutil.copy(song_file, destination_path)
+                if not options.no_index and index.add_track(track_info=track_info, path=destination_path):
+                    logger.info(f"Added {destination_path} to index")
+
+            except SongRecognizeError as e:
+                logger.error(f"Error: {e}")
+                Path(f"{options.dest}/Unknown").mkdir(mode=0o755, parents=True, exist_ok=True)
+                shutil.copy(song_file, os.path.join(f"{options.dest}/Unknown/{os.path.basename(song_file)}"))
 
 
 if __name__ == '__main__':
-    logging.basicConfig(filename="music.log", level=logging.DEBUG)
+    logging.basicConfig(filename="/dev/stdout", level=logging.DEBUG)
     args = arg_parser()
     loop = asyncio.get_event_loop()
-    loop.run_until_complete(main(args.src, args.dest))
+    loop.run_until_complete(main(args))