Refactoring models outer main.py

main.py

@@ -1,81 +1,23 @@
 import argparse
 import asyncio
-import itertools
 import logging
 import os
 import shutil
-from files import util as ioutil
 from pathlib import Path
 from typing import Any
 
 from shazamio import Shazam
 
+from files import util as ioutil
+from models import TrackInfo
+
 logger = logging.getLogger(__name__)
 
 
-class Album:
-    def __init__(self, metadata: dict[str, Any]):
-        sections = metadata["track"]["sections"]
-        info = [x["metadata"] for x in sections if "metadata" in x]
-        items = list(itertools.chain(*info))
-
-        def extract_data(title: str) -> str | int:
-            for item in items:
-                if item["title"] == title:
-                    return item["text"]
-
-        self.__name: str = extract_data("Album")
-        self.__released: int = extract_data("Released")
-
-    @property
-    def name(self):
-        return self.__name
-
-    @property
-    def released(self):
-        return self.__released
-
-    def __str__(self):
-        return f"""Name: {self.__name}
-                   Released: {self.__released}"""
-
-
-class TrackInfo:
-    def __init__(self, metadata: dict[str, Any]):
-        track = metadata["track"]
-
-        self.__title: str = track["title"]
-        self.__artist: str = track["subtitle"]
-        self.__number: int = 0
-        self.__album: Album = Album(metadata)
-
-    @property
-    def title(self) -> str:
-        return self.__title
-
-    @property
-    def artist(self) -> str:
-        return self.__artist
-
-    @property
-    def album(self) -> Album:
-        return self.__album
-
-    @property
-    def track_number(self) -> int:
-        return self.__number
-
-    def __str__(self) -> str:
-        return f"""Artist: {self.__artist}
-            Title: {self.__title}
-            Album:  {self.__album}
-            Track Number: {self.__number}"""
-
-
 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 directory to scan files")
-    parser.add_argument("--dest", "-d", metavar="DEST", type=str, help="Destination directory to copy files")
+    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")
     return parser.parse_args()
 
 

models.py

@@ -0,0 +1,61 @@
+import itertools
+from typing import Any
+
+
+class Album:
+    def __init__(self, metadata: dict[str, Any]):
+        sections = metadata["track"]["sections"]
+        info = [x["metadata"] for x in sections if "metadata" in x]
+        items = list(itertools.chain(*info))
+
+        def extract_data(title: str) -> str | int:
+            for item in items:
+                if item["title"] == title:
+                    return item["text"]
+
+        self.__name: str = extract_data("Album")
+        self.__released: int = extract_data("Released")
+
+    @property
+    def name(self):
+        return self.__name
+
+    @property
+    def released(self):
+        return self.__released
+
+    def __str__(self):
+        return f"""Name: {self.__name}
+                   Released: {self.__released}"""
+
+
+class TrackInfo:
+    def __init__(self, metadata: dict[str, Any]):
+        track = metadata["track"]
+
+        self.__title: str = track["title"]
+        self.__artist: str = track["subtitle"]
+        self.__number: int = 0
+        self.__album: Album = Album(metadata)
+
+    @property
+    def title(self) -> str:
+        return self.__title
+
+    @property
+    def artist(self) -> str:
+        return self.__artist
+
+    @property
+    def album(self) -> Album:
+        return self.__album
+
+    @property
+    def track_number(self) -> int:
+        return self.__number
+
+    def __str__(self) -> str:
+        return f"""Artist: {self.__artist}
+            Title: {self.__title}
+            Album:  {self.__album}
+            Track Number: {self.__number}"""