From 4dae1d037f5d6f356b70300111f914139e6891e4 Mon Sep 17 00:00:00 2001
From: Lukas Pupka-Lipinski <lukas.pupkalipinski@lpl-mind.de>
Date: Sat, 17 May 2025 21:31:16 +0200
Subject: [PATCH] backup

---
 __pycache__/Loader.cpython-310.pyc | Bin 0 -> 2702 bytes
 main.py                            |  65 ++++++++++++++++-------------
 2 files changed, 35 insertions(+), 30 deletions(-)
 create mode 100644 __pycache__/Loader.cpython-310.pyc

diff --git a/__pycache__/Loader.cpython-310.pyc b/__pycache__/Loader.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..90bd0ff09e40718107a6f3d397f26825f76f8602
GIT binary patch
literal 2702
zcma)7&2JP(7VoP5oSyM`Fvb`RNr#xkv#TDVU3OQp5E}~{;SI47b|4uUjhd<Qxb6NJ
zSJxP()w5znu93J#StR?&W&e;!x#d*Xoc53tmnhP%^Qy*U8x*8;t6qJ*SM}cSeN;L&
z=4$YL^PjU@-&Zv4M>#qDnK1bTT2=rd8bOiP9QLS*)Eqal>h-1$J&ugnY??^&=#drM
zO*?j)j+z@$DR!GK(x|>&CMM7dv55V(4l83++piLbln(H7)Ep-+H8(LSlgd{{vqsRG
zRv-H>XrQgskuVpMFs5}ZYWME1K3HB{y5G99v;f_jm{?wUuyA*Iv2|}@{Z8xForU{r
zAW#Zerx(+NXL*IBPm(ANNFz!+dHw2KkhQryE%SWu>Rd$kXp{uU^O$B?(1G2$E}YFU
zqH<4|icMIXDT@Q%tSzoST)Dftu-IB({%mRW^Y!oxtZBjwq?&Cc)js+p*5lA_LCZKW
z722Zq0zJAuKm_drdPM&U*0#~2_8c9c0WQ!sCdLkCvjzTEBPPssVts268@LQuab8*j
zJvpm(ONv|Wx{9y-2A?y~xw&oa%k%9*KfnV6=8jq`6~+M~6*BfUMjChb%cNSs=`ZmC
z0?&;XXukqdWAe@eOlsr|nILD$WMPnVWQv^cln18V=hZ!p^?6l}FMwr_N&~Af_s7X}
zVRh75yD6`2pJD%&cK{ax3zXK-E1jVqS$2V31Vm;OBIXbe(5NIe+Qs>d((;njl3aeN
z%T+$HeYP-3g-??AWK1cJ_fPD22y%wxI@kXJY4Qh-DdO=-IO79#+m*L&15}Xq<w)*)
z|EE)OA1aC2w<MI6M-tO7Q2%eo5*Nr-rKR+N!h2wo$^h`+zgXA@;PFS(gA$+FzeGMN
zAPZg~HUoQ?k9=m!H7WlB?ONpHfg}3@Xf`)G0mgrU@q6;LBkSi1d*nr_(-X)_$fWv;
z(B2432t-YnAt~D>Ape3O0q3CtacvJV2*U>e*DfUbn)V7o4l)C{AOmsZnlRd10ehvf
zDlGkSadCcw(Qdi{Ir=1J5!tv4$wpbD+ZUx99h%UmU3POyUgQA5&kSt>8Hf4IabDht
z(smGKHyg*?Or}s?f&~36-3zHF@U=hX22AI2v)+ii*VSZAprt0h_UCm-c!xd>Gu}#f
z^7HpVn}CH$$K$Ctx@6wVtv>fj)XgV+f0G3rSsi>vISXmlbi!maOhQicDc^6WNxR3G
z+#fM!0$tCspXRpj@Ag8<^D*B~QvXT761XYLUElArw8Nl!<Q3l!2=TY5gdx}K^PWK7
zyWyV&TNV6c5`Lof5UX&lK+jPEHR8k#b-co4IeCSIo^kdHGbrDhFhJ{Axnf9UwP7_N
zH=$+HYxvm}EkC!$1I8sPBS<7Xx!!0rz^i6^i?(-!Eem7X<N4&`QPK3;Y21w{rvwnZ
zaJ(w{hzFcru*oU&a@Wf!;6}3fcU$^;8kA9%+5-qPY2k@=tPGn>h6$5wzh%%JSbse<
zXiCe^FEN%fkM$Di0jXykB%tLxfo3w!f5$cdkZhNM<d=`_4HK}E^}2wA_qgx%GKlYd
zW}`6&k;aQYCs83s{%r`X4f4nO?Ct!@eHy2Gz)#~Y@1LMfLPpy>W&L@t{=P7_(=ZVx
z4|zm|mC+ze6JgVCn5BdYCkm2I51u1|(@YrMfNzN!px*8A*6}&QiGv*q(@fYv)q7>b
zTh{?tAUcxGRE&4%NT3yk$&PR!SocB*PbRB?u;^1x6T;-%FDhvg^;<`H3ODOL2Ie*d
zw$K4=HY0he>%%}<dfEkZh%n)#l!=L3K@?GPZ^*A4B+h(F`;XIrk>wGF!ahnHVMma(
zptHbiXgzCh35zM<gvnyqs=KTNH?j|aG|y%5!tyYuq@|wPmcp%BK7vF(*T)kDmOV1V
z8fK$#PtX%in6;F1L`h1vByw-Xop46kG70x*{{-S97a7PobPWqzP{<5y{ogpWY&jmy
z@nOxHLN$!_D)7o5tKaz9Emu(uS=c}xl4a01#0JI_xP~TC75xF)EHoEYK&FbTXc{@N
z>w=8&48JzMV}Az~>Sx%8(6PTt(vV~hh!_V<R0VFt5Ij-FWR*X)3i-Jic9h3eH9xfl
z!3!Zh^ndwrkcI7Aidh)TE4NpL4)HH62qK#7sQ^%FoWe@J1WZ;Q;oN}Bdl9`UOSpPk
W9L+)UuYi$YmW~x1*8ghu1pXDLSHt`O

literal 0
HcmV?d00001

diff --git a/main.py b/main.py
index 0b68455..5750e38 100644
--- a/main.py
+++ b/main.py
@@ -33,40 +33,20 @@ class Loader:
                 logging.debug(f"Found {len(mp4_files)} .mp4 files in {root_folder_name}")
                 yield root_folder_name, mp4_files
         for dir in self.__find_empty_folders():
-            relative_path = os.path.relpath(dir, self.directory)
-            logging.debug(f"Found no .mp4 files in {relative_path}")
-            yield relative_path, []
-        for dir in self.__find_folders_without_mp4():
-            relative_path = os.path.relpath(dir, self.directory)
-            logging.debug(f"Found no .mp4 files in {relative_path}")
-            yield relative_path, []
+            logging.debug(f"Found no .mp4 files in {dir}")
+            yield dir, []
 
     def __find_empty_folders(self):
-        """Finds and returns a list of empty folders in the given directory."""
-        empty_folders = []
+        """Yield folder names that do not contain any mp4 files in a given directory."""
+        for folder in os.listdir(self.directory):
+            folder_path = os.path.join(self.directory, folder)
 
-        for root, dirs, files in os.walk(self.directory):
-            if not files and not dirs:  # If the folder contains no files or subdirectories
-                empty_folders.append(root)
+            if os.path.isdir(folder_path):  # Ensure it's a directory
+                has_mp4 = any(file.endswith(".mp4") for file in os.listdir(folder_path))
 
-        return empty_folders
-    def __find_folders_without_mp4(self):
-        """
-        Finds folders that are either empty or contain no .mp4 files.
+                if not has_mp4:
+                    yield folder  # Yield the folder name if no mp4 files found
 
-        :param root_dir: The directory to search within.
-        :return: List of folders that are empty or have no .mp4 files.
-        """
-        result_folders = []
-
-        for foldername, subfolders, filenames in os.walk(self.directory):
-            # Check if the folder is empty or has no MP4 files
-            folder_name = os.path.relpath(foldername, self.directory)
-            if not any(file.lower().endswith('.mp4') for file in filenames):
-                if (folder_name != "."):
-                    result_folders.append(foldername)
-
-        return result_folders
     def __remove_year(self, input_string: str):
         cleaned_string = re.sub(r'\(\d{4}\)', '', input_string).strip()
         logging.debug(f"Removed year from '{input_string}' -> '{cleaned_string}'")
@@ -95,11 +75,36 @@ class Loader:
             season = match.group(1)
             episode = match.group(2)
             logging.debug(f"Extracted season {season}, episode {episode} from '{filename}'")
-            return season, episode
+            return int(season), int(episode)
         else:
             logging.error(f"Failed to find season/episode pattern in '{filename}'")
             raise MatchNotFoundError("Season and episode pattern not found in the filename.")
 
+    def __GetEpisodesAndSeasons(self, mp4_files: []):
+        episodes_dict = {}
+
+        for file in mp4_files:
+            season, episode = self.__GetEpisodeAndSeason(file)
+
+            if season in episodes_dict:
+                episodes_dict[season].append(episode)
+            else:
+                episodes_dict[season] = [episode]
+
+        return episodes_dict
+
+    def __GetMissingEpisodesAndSeason(self, key: str, mp4_files: []):
+        expected_dict = get_season_episode_count(key) # key season , value count of episodes
+        filedict = self.__GetEpisodesAndSeasons(mp4_files)
+
+        for season, expected_count in expected_dict.items():
+            existing_episodes = filedict.get(season, [])
+            missing_episodes = [ep for ep in range(1, expected_count + 1) if ep not in existing_episodes]
+
+            if missing_episodes:
+                yield season, missing_episodes
+
+
     def LoadMissing(self):
         logging.info("Starting process to load missing episodes")
         result = self.__find_mp4_files()