From a5463bd2fe2cd055ed11fff08ca59a3c80606e99 Mon Sep 17 00:00:00 2001
From: Sergio <jaildesigner@gmail.com>
Date: Thu, 3 Aug 2023 16:15:06 +0200
Subject: [PATCH] =?UTF-8?q?A=C3=B1adido=20zx=5Fgame=5Fcollector.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 zx_game_collector.py | 101 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 101 insertions(+)
 create mode 100644 zx_game_collector.py

diff --git a/zx_game_collector.py b/zx_game_collector.py
new file mode 100644
index 0000000..d2d3361
--- /dev/null
+++ b/zx_game_collector.py
@@ -0,0 +1,101 @@
+import os
+import re  # regexp
+import shutil
+from pathlib import Path
+
+
+def first_letter(x):
+    if len(x) == 0:
+        return "0-9"
+    if x[0] in ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"]:
+        return "0-9"
+    else:
+        return x[0].upper()
+
+
+source_path = [Path("/home/sergio/zx/Games"), Path("/home/sergio/zx/Pokes")]
+destination_path = Path("/home/sergio/tmp/final")
+
+opt_print = "no"
+opt_create_dirs = "yes"
+
+# Elimina el directorio de destino
+try:
+    print("Directory: {} -> deleting...".format(destination_path))
+    shutil.rmtree(destination_path)
+    print("Directory: {} -> removed successfully".format(destination_path))
+except OSError as o:
+    print(f"Error, {o.strerror}: {destination_path}")
+
+# Crea el directorio de destino
+try:
+    os.mkdir(destination_path)
+    print("Directory: {} -> created successfully".format(destination_path))
+except OSError as error:
+    print(error)
+
+# Variables
+paths = []  # Ruta donde se encuentra el fichero
+files = []  # Nombre del fichero
+names = []  # Nombre del juego
+years = []  # Año del juego
+companies = []  # Compañía o distribuidora del juego
+
+# Obtiene la lista de ficheros desde los directorios de origen
+for path in source_path:
+    for file_name in os.listdir(path):
+        if os.path.isfile(os.path.join(path, file_name)):  # Comprueba si es un fichero
+            paths.append(path)  # Añade la ruta
+            files.append(file_name)  # Añade el nombre del fichero
+
+# Extrae los datos del juego
+regex_year = r"\(\d.*?\)"
+regex_company = r"^\(.*?\)"
+
+for i in files:
+    # Año
+    match = re.search(regex_year, i)  # Busca el año en el nombre del fichero
+    if match:
+        years.append(match.group())  # Añade el año con los parentesis a la lista
+        pos = i.find(years[-1])  # Busca el caracter donde empieza el año
+        names.append(
+            i[0:pos].strip()
+        )  # Añade como nombre el texto que hay desde el principio hasta el año
+        match_company = re.search(
+            regex_company, i[pos + len(years[-1]) :]
+        )  # Busca la compañia en lo que queda despues del año
+        if match_company:
+            companies.append(match_company.group())
+        else:
+            companies.append("-")
+
+        # Limpia los parentesis del año y la compañía
+        years[-1] = years[-1][1:5]
+        if companies[-1] != "-":
+            companies[-1] = companies[-1][1:-1]
+    else:
+        years.append("0")
+        names.append(i)
+        companies.append("-")
+
+# Lista los resultados
+if opt_print == "yes":
+    for i, item in enumerate(files):
+        print(
+            "File:    {}\nName:    {}\nYear:    {}\nCompany: {}\n".format(
+                item, names[i], years[i], companies[i]
+            )
+        )
+
+# Copia los archivos
+total_files = len(files)
+if opt_create_dirs == "yes":
+    for i in range(total_files):
+        print("({} de {}) {}".format(i + 1, total_files, files[i]))
+        game_dir = names[i] + " (" + years[i] + ")"
+        dst_path = os.path.join(destination_path, first_letter(names[i]), game_dir)
+        if not os.path.exists(dst_path):
+            os.makedirs(dst_path)
+        src = os.path.join(paths[i], files[i])
+        dst = os.path.join(dst_path, files[i])
+        shutil.copyfile(src, dst)