Unit3Dup 0.8.14__py3-none-any.whl → 0.8.16__py3-none-any.whl

common/command.py

@@ -43,11 +43,10 @@ class CommandLine:
         parser.add_argument("-duplicate", "--duplicate", action="store_true", help="Find duplicates")
         parser.add_argument("-personal", "--personal", action="store_true", help="Set to personal release")
 
-        parser.add_argument("-pw", "--pw", type=str, help="")
         parser.add_argument("-ftp", "--ftp", action="store_true", help="Connect to FTP")
 
         # optional
-        parser.add_argument("-dump", "--dump", type=str, help="Download all torrent files")
+        parser.add_argument("-dump", "--dump", action="store_true", help="Download all torrent titles")
         parser.add_argument("-s", "--search", type=str, help="Search for torrent")
         parser.add_argument("-i", "--info", type=str, help="Get info on torrent")
         parser.add_argument("-up", "--uploader", type=str, help="Search by uploader")

common/database.py

@@ -1,5 +1,7 @@
 import json
+import os
 import sqlite3
+from unit3dup import config_settings
 
 # Torrent attributes
 create_table_sql = ('\n'
@@ -53,14 +55,17 @@ class Database:
 
     def __init__(self, db_file):
         self.filename = db_file
-        self.database = sqlite3.connect(f"{db_file}.db")
+        self.CACHE_PATH = config_settings.user_preferences.CACHE_PATH
+        self.database = sqlite3.connect(os.path.join(self.CACHE_PATH,f"{db_file}.db"))
         self.cursor = self.database.cursor()
         self.build()
 
     def build(self):
+        self.cursor.execute("DROP TABLE IF EXISTS torrents")
         self.cursor.execute(create_table_sql)
         self.database.commit()
 
+
     def write(self, data: dict):
         for key, value in data.items():
             if isinstance(value, (dict,list)):
@@ -69,26 +74,6 @@ class Database:
         keys = ', '.join(data.keys())
         placeholders = ', '.join(['?'] * len(data))
         values = tuple(data.values())
-
         sql = f'''INSERT INTO torrents ({keys}) VALUES ({placeholders})'''
         self.cursor.execute(sql, values)
-        self.database.commit()
-
-    def search(self, query: str):
-        # Search a substring in 'name'
-        self.cursor.execute("SELECT name FROM torrents WHERE name LIKE ?", ('%' + query + '%',))
-        results = self.cursor.fetchall()
-        # print the results
-        for r in results:
-            print(f"[database]{r}")
-        input("[DATABASE] Press Enter to continue...")
-
-
-
-
-
-
-
-
-
-
+        self.database.commit()

common/settings.py

@@ -13,7 +13,7 @@ from common.utility import ManageTitles
 from common import trackers
 
 config_file = "Unit3Dbot.json"
-version = "0.8.14"
+version = "0.8.16"
 
 if os.name == "nt":
     PW_TORRENT_ARCHIVE_PATH: Path = Path(os.getenv("LOCALAPPDATA", ".")) / "Unit3Dup_config" / "pw_torrent_archive"

unit3dup/__main__.py

@@ -125,14 +125,6 @@ def main():
         bot.watcher(duration=config.user_preferences.WATCHER_INTERVAL, watcher_path=config.user_preferences.WATCHER_PATH,
                     destination_path = config.user_preferences.WATCHER_DESTINATION_PATH)
 
-    # Pw
-    if cli.args.pw:
-        print("Not yet implemented")
-        # bot = Bot(path=cli.args.pw,cli=cli.args, trackers_name_list=tracker_name_list)
-        # bot.pw()
-        return
-
-
     # ftp and upload
     if cli.args.ftp:
         bot = Bot(path='', cli=cli.args, mode="folder", trackers_name_list=tracker_name_list)
@@ -153,13 +145,14 @@ def main():
 
     # Search by different criteria
     if cli.args.search:
-        torrent_info.view_search(cli.args.search)
+        print("NOT YET IMPLEMENTED")
+        # torrent_info.view_search(cli.args.search)
         return
 
     # Dump
     if cli.args.dump:
-        print("Not yet implemented")
-        # torrent_info.view_search(cli.args.dump,inkey=False)
+        print("NOT YET IMPLEMENTED")
+        # torrent_info.view_search(" ", inkey=False)
         return
 
     if cli.args.info:

unit3dup/bot.py

@@ -11,7 +11,6 @@ from unit3dup.media_manager.ContentManager import ContentManager
 from unit3dup.media_manager.TorrentManager import TorrentManager
 
 from common.external_services.ftpx.core.models.list import FTPDirectory
-from common.external_services.Pw.pw_manager import PwManager
 from common.external_services.ftpx.core.menu import Menu
 from common.external_services.ftpx.client import Client
 from common.extractor import Extractor
@@ -55,6 +54,10 @@ class Bot:
         """
         custom_console.panel_message("Analyzing your media files... Please wait")
 
+        if not os.path.exists(self.path):
+            custom_console.bot_error_log("Path doesn't exist")
+            return False
+
         # Get a Files list with basic attributes and create a content object for each
         self.content_manager: ContentManager = ContentManager(path=self.path, mode=self.mode, cli=self.cli)
         contents = self.content_manager.process()
@@ -64,10 +67,6 @@ class Bot:
             custom_console.bot_error_log("There are no Media to process")
             return False
 
-        if not os.path.exists(self.path):
-            custom_console.bot_error_log("Path doesn't exist")
-            return False
-
         # we got a handled exception
         if contents is None:
             exit(1)
@@ -179,19 +178,6 @@ class Bot:
             custom_console.bot_log("Exiting...")
         return True
 
-    def pw(self)-> bool:
-        """
-        Interacts with the PW service to search for torrent files
-
-        This method performs a search query and logs the results for torrents with
-        a certain number of seeders
-        """
-        # PW service
-        pw_manager = PwManager(cli=self.cli)
-        pw_manager.process()
-        custom_console.panel_message("Searching... Please wait")
-        return True
-
 
     def ftp(self)-> None:
         """

unit3dup/torrent.py

@@ -147,6 +147,7 @@ class View(Torrent):
 
         # Load the constant tracker
         self.tracker_data = TRACKData.load_from_module(tracker_name=tracker_name)
+        self.tracker_name = tracker_name
         print()
 
     def get_unique_id(self, media_info: str) -> str:
@@ -182,12 +183,12 @@ class View(Torrent):
                 else:
                     release_year = item['attributes']['release_year']
 
-                media = f"[TRACKER] TMDB: {item['attributes']['tmdb_id']} - {release_year}"
+                media = f"{self.tracker_name} - TMDB: {item['attributes']['tmdb_id']} - {release_year}"
 
             elif item['attributes']['igdb_id'] !=0:
-                    media = f"[TRACKER] IGDB: {item['attributes']['igdb_id']}"
+                    media = f"{self.tracker_name} IGDB: {item['attributes']['igdb_id']}"
             else:
-                media = f"[TRACKER] DOC:"
+                media = f"{self.tracker_name} DOC:"
 
             # Print a data to the console
             custom_console.bot_log(f"\n {media} - {item['attributes']['name']}")

{unit3dup-0.8.14.dist-info → unit3dup-0.8.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Unit3Dup
-Version: 0.8.14
+Version: 0.8.16
 Summary: An uploader for the Unit3D torrent tracker
 Author: Parzival
 License-Expression: MIT

{unit3dup-0.8.14.dist-info → unit3dup-0.8.16.dist-info}/RECORD

@@ -1,14 +1,14 @@
 common/__init__.py,sha256=M82CDZm5KKWf8V3a10FKCy1wtFrJd1TMEhBib1Xtk1g,76
 common/bdinfo_string.py,sha256=dfrU5T90fvkO_fklvVNWSoCgd5qvrVvqEZ_HKEsHAU8,3457
 common/bittorrent.py,sha256=MPgMykWjmhGox7jf1MQn53Ew4NxpjzEc4FCF6oA2oYs,315
-common/command.py,sha256=7QmtTFdawq2S70p9JP2dr2bGbAUdiTVnGFiNrBcz3R4,6029
+common/command.py,sha256=BJNOdnfS5FQpZdCib-q0Tq9S9IoPtHlOEF9kxDQCzls,5978
 common/constants.py,sha256=tdflS7ZJwAcI1ZYUhBBb60QybVT7TmVS-EYxS4bNFp0,352
-common/database.py,sha256=Lp6i3Wu5vza87atWD2NGQRCchVLBcY-k8E0AAuAQxls,3232
+common/database.py,sha256=XOjh-X3XZBHjln9scRgzArpTlfgfR7P3ZlwahPbeUSU,3056
 common/extractor.py,sha256=WKZwt2kQfKO2VJ1rtwE_j6Zl84zICnowZq_Ql16wmRc,4564
 common/frames.py,sha256=p_jsaXII5tZTVt5ymu-w1hk2c_UMeOn3PZeuVR-wXWY,7973
 common/mediainfo.py,sha256=U2r1jJejBsV8GP3iPk4O8_NkHO5RQ9Kkh2bKwVNUBmg,6229
 common/mediainfo_string.py,sha256=8vuWlF2bqWRKpDbn81bV2fPA7hbl7RwOnxN2i4E3zNE,3958
-common/settings.py,sha256=9UCaHT1ZC0c3U0vwOxxpNNkNpyMo93nXmopXdEkn3hg,32782
+common/settings.py,sha256=Lpr3MRybihcsdh27bEnSg1VxCIXRrpKZMn39mtY1Gw8,32782
 common/title.py,sha256=nFainfUBTYW9ML4Y-CB9ZFD_Es-OZXcAMPUo6D09u3k,3793
 common/torrent_clients.py,sha256=NOIpYtLG_bA19HwcH9ahGFmGNtRkoMO6nAjma-JzDfs,12040
 common/utility.py,sha256=r5MuFrafAzSJbe7DLx8uMKSG1qQi8k6oA43Y_qJhH7A,8647
@@ -16,15 +16,6 @@ common/external_services/__init__.py,sha256=rU7HPEcZ7WQFD8eqDzuye2xHPBjxXPwPqpt7
 common/external_services/imageHost.py,sha256=BpdtGZUtZHl4QJwFKyJs_BI6z4gHI4wu_Beu-fnfkDw,9588
 common/external_services/imdb.py,sha256=AIo8CO-SfP_uNocDeNY08hpvCAnuotoI7hYUytDiMQA,579
 common/external_services/mediaresult.py,sha256=Yz-h7QROkfNHwaSSXZ1JRDSmG2YIZkmtQO5vmpkYWZ4,677
-common/external_services/Pw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-common/external_services/Pw/pw_manager.py,sha256=a8s3YgzHf0iw45Fh5BeMRSX5aGlatBJbOYamSrWFj2U,2225
-common/external_services/Pw/pw_service.py,sha256=_-AQgffqiuoS0xcHc7sEgQBuXidU9RdJZyZnRa-ATZc,1151
-common/external_services/Pw/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-common/external_services/Pw/core/pw_api.py,sha256=C1XzN-hJwJvzjZwiNTL39mu4gJGiEKKZYjt9JAStIHk,3234
-common/external_services/Pw/core/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-common/external_services/Pw/core/models/indexers.py,sha256=7nVqZd-_nbwL40HLGEFauI8zNeC-w66py2tK7h9rGV4,1947
-common/external_services/Pw/core/models/search.py,sha256=fmoEccxgRDG26nmEcpJR0e5twBLTOEzDzaeQFG2TAc4,969
-common/external_services/Pw/core/models/torrent_client_config.py,sha256=zmPqcYoRSIeXoxLvK0Q5wAj_gkSd1y1u1oEj5g6BvGM,1025
 common/external_services/ftpx/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 common/external_services/ftpx/client.py,sha256=D_23Cw9zeQQIiVhUIUE13ENXNVyP4-MMx_Rgqg67U7Y,10713
 common/external_services/ftpx/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -72,9 +63,9 @@ common/trackers/itt.py,sha256=cB0D5ZpMbGgEOAKloET9dS_fAFclhzwPPwRFt-LStX0,4019
 common/trackers/sis.py,sha256=AahQH-FxAqS19vgEFylOJzeY0_SHr27_Tibs7lraSdI,3480
 common/trackers/trackers.py,sha256=sts5l_x27V2Op1Y6Au5FoQujspSkWMPWyEYMfcmVYLA,1467
 unit3dup/__init__.py,sha256=seXz3lHgdrUBiOnhC6Je47npS66UZ0c62VFuoH3z5Mk,78
-unit3dup/__main__.py,sha256=UIQ0s9deamfOSS7-O65ojeuN9Cx6zpPGD5H1HXlfrMM,9256
+unit3dup/__main__.py,sha256=DBhzZdUi19N1-WTBQJbps9wQKNcNhwreaXwXQtFORHs,9088
 unit3dup/automode.py,sha256=HIJrE8qEHp7DZQbppNtB_8jHY3Q94d6frp7BXnZuEac,4387
-unit3dup/bot.py,sha256=TM8Htnq-udJDnN-7TwaPP6CiwWX0wb_Nud9p8ej8GRo,9250
+unit3dup/bot.py,sha256=2P24z_8UM58binKqrM0eIXxPTiWEduyK2bdfVkmxPEQ,8759
 unit3dup/duplicate.py,sha256=Ji1Y9vVLmhFQ5rgkUU-s7uXl-jC8akAa8OiQOfjFa9g,10428
 unit3dup/exceptions.py,sha256=DhlcbbZEuvEYk5VSZTnle8oRg5I8Pq6Y6fxXTdOOOfo,4078
 unit3dup/media.py,sha256=5suH2XeekB80zNoIAM8PERZBO746JDlpcIvHz-4jY9E,13888
@@ -82,7 +73,7 @@ unit3dup/pvtDocu.py,sha256=ZLMaal2fn6ZcFPLtET0LTmYEPjhJmLitEjkICHrm9CQ,4594
 unit3dup/pvtTorrent.py,sha256=cItlsCpcUJL23iXQHy0YzrrvV3JSl54UlBgm8_UROAs,2559
 unit3dup/pvtTracker.py,sha256=U4xMmhiqMNZaIOgEi3pnP-3H75e7Gagndh8GDFEP7Cs,18647
 unit3dup/pvtVideo.py,sha256=w_T6wEeGrsHkuR3KObJc4mNpJgg9hhHh_9LoxFlaqjM,3900
-unit3dup/torrent.py,sha256=JEcrVG8dlYASBvL51GldU1ouKlfieKccKX2KZ_4UKBY,19883
+unit3dup/torrent.py,sha256=tWnAxZjPKGr_ktqfVmimnSgdnWU0efNbB-FIsLUMmY8,19957
 unit3dup/upload.py,sha256=gjB8u2oP2FPXZ8t9SMIzkUEr52hKRy_zSR8OWJlW2xc,6075
 unit3dup/media_manager/ContentManager.py,sha256=68k8sqJgL1a2ir_rpQhkB1ntePp01IpblJYcQCdev2Y,7358
 unit3dup/media_manager/DocuManager.py,sha256=oFt7jlxj-gIUty9PADBQV5a24bsv3yhjKhwI6niOhf4,3116
@@ -95,11 +86,13 @@ unit3dup/media_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 unit3dup/media_manager/common.py,sha256=hG2zOw7ocQfZyI1dhusbehxswpIrZK7T2aAbCNwULqA,10138
 unit3dup/web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unit3dup/web/main.py,sha256=BzjKDgAjKZMnoQwx7nDDbs_64kCrFO1VYpbHmsGiFVc,1267
-unit3dup-0.8.14.dist-info/licenses/LICENSE,sha256=GNAZDLhU0xz8QPbIyHAOYlVnQYDvKWk2N9fZJMhqaG8,1090
+unit3dup-0.8.16.dist-info/licenses/LICENSE,sha256=GNAZDLhU0xz8QPbIyHAOYlVnQYDvKWk2N9fZJMhqaG8,1090
 view/__init__.py,sha256=XIzb6rl58HmYPnksD73cYMFF88dn6FHa3u7bOHFbChk,81
 view/custom_console.py,sha256=OITmkEoQH9N_uE5ElPaSdc8XvaLzE9wcwTbOHtcMvrI,5629
-unit3dup-0.8.14.dist-info/METADATA,sha256=Zqj0fjP9zuTWL1BDAJy6dxRciIYke0m3GP7z2lcArRg,4864
-unit3dup-0.8.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-unit3dup-0.8.14.dist-info/entry_points.txt,sha256=fxXSyI6-r6jy9_v-C5ZHm03q1aC3tE9EvCQZxC1NQnI,52
-unit3dup-0.8.14.dist-info/top_level.txt,sha256=19NVMnQNkJxBUKebRNaYCRs56A5CO4U1L67GMQCPiLU,21
-unit3dup-0.8.14.dist-info/RECORD,,
+view/custom_console2.py,sha256=ipNoZWz1YO5iOp-vIH3PrWoBzpiAxdW6gdsijKSEg7c,6137
+view/settings.py,sha256=dSabnCfUGqPS8w1a2B_O_GF_eT9GsxBDRp7Yzf6RDF8,1289
+unit3dup-0.8.16.dist-info/METADATA,sha256=6b9LonXp-zhs5cbxezMrA9RErDDuB4wvoDLsjAsNjRM,4864
+unit3dup-0.8.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+unit3dup-0.8.16.dist-info/entry_points.txt,sha256=fxXSyI6-r6jy9_v-C5ZHm03q1aC3tE9EvCQZxC1NQnI,52
+unit3dup-0.8.16.dist-info/top_level.txt,sha256=19NVMnQNkJxBUKebRNaYCRs56A5CO4U1L67GMQCPiLU,21
+unit3dup-0.8.16.dist-info/RECORD,,

view/custom_console2.py

@@ -0,0 +1,139 @@
+from nicegui import ui
+from typing import List, Callable
+from common import config_settings
+from datetime import datetime
+
+
+class NiceGUIConsole:
+    def __init__(self):
+        self.init_footer_log()
+        self.log_area = ui.column().style('max-height: 400px; overflow-y: auto; border: 1px solid #ccc; padding: 10px')
+        self.counter_label = ui.label()
+
+    def init_footer_log(self):
+        with ui.element('div').style('''
+            position: fixed;
+            bottom: 0;
+            left: 0;
+            width: 100%;
+            background-color: #2e3b4e;
+            text-align: center;
+            padding: 4px 8px;  /* meno padding verticale */
+            font-size: 14px;
+            line-height: 18px;
+            z-index: 1000;
+        ''') as footer:
+            self.footer_label = ui.label("Ready.").style("color: white; margin: 0")
+
+    def log_message(self, message: str, color: str = "#ffffff"):
+        timestamp = datetime.now().strftime("%H:%M:%S")
+        """
+        with self.log_area:
+            ui.label(f"[{timestamp}] {message}").style(f'color: {color}; white-space: pre-wrap')
+        """
+        self.footer_label.text = f"${timestamp}$ {message}"
+        self.footer_label.style(f"color: {color}")
+
+
+    def panel_message(self, message: str, color: str = "#ccc"):
+        with ui.card().style(f'background-color: {color}; color: white'):
+            ui.label(message).style("text-align: center")
+
+    def bot_log(self, message: str):
+        self._add_log(message, config_settings.console_options.NORMAL_COLOR)
+
+    def bot_error_log(self, message: str):
+        self._add_log(f'❌ {message}', config_settings.console_options.ERROR_COLOR)
+
+    def bot_warning_log(self, message: str):
+        self._add_log(f'⚠️ {message}', config_settings.console_options.QUESTION_MESSAGE_COLOR)
+
+    def bot_input_log(self, message: str):
+        self._add_log(message, config_settings.console_options.NORMAL_COLOR)
+
+    def bot_question_log(self, message: str):
+        self._add_log(message, config_settings.console_options.QUESTION_MESSAGE_COLOR)
+
+    def bot_counter_log(self, message: str):
+        self.counter_label.set_text(message)
+        self.counter_label.style(f'color: {config_settings.console_options.QUESTION_MESSAGE_COLOR}')
+
+    def bot_process_table_log(self, content: List):
+        ui.label("Here is your files list" if content else "There are no files here")
+        ui.table(columns=[
+            {'name': 'pack', 'label': 'Torrent Pack', 'field': 'pack'},
+            {'name': 'category', 'label': 'Media', 'field': 'category'},
+            {'name': 'path', 'label': 'Path', 'field': 'path'},
+        ], rows=[
+            {
+                'pack': 'Yes' if item.torrent_pack else 'No',
+                'category': item.category,
+                'path': item.torrent_path,
+            } for item in content
+        ])
+
+    def bot_process_table_pw(self, content: List):
+        ui.label("Here is your files list" if content else "There are no files here")
+        ui.table(columns=[
+            {'name': 'category', 'label': 'Category', 'field': 'category'},
+            {'name': 'indexer', 'label': 'Indexer', 'field': 'indexer'},
+            {'name': 'title', 'label': 'Title', 'field': 'title'},
+            {'name': 'size', 'label': 'Size', 'field': 'size'},
+            {'name': 'seeders', 'label': 'Seeders', 'field': 'seeders'},
+        ], rows=[
+            {
+                'category': item.categories[0]['name'],
+                'indexer': item.indexer,
+                'title': item.title,
+                'size': str(item.size),
+                'seeders': str(item.seeders),
+            } for item in content
+        ])
+
+    def bot_tmdb_table_log(self, result, title: str, media_info_language: List[str]):
+        media_info_audio_languages = ", ".join(media_info_language).upper()
+        self.panel_message(f"Results for {title.upper()}")
+
+        ui.table(columns=[
+            {'name': 'tmdb_id', 'label': 'TMDB ID', 'field': 'tmdb_id'},
+            {'name': 'language', 'label': 'LANGUAGE', 'field': 'language'},
+            {'name': 'poster', 'label': 'TMDB POSTER', 'field': 'poster'},
+            {'name': 'backdrop', 'label': 'TMDB BACKDROP', 'field': 'backdrop'},
+        ], rows=[{
+            'tmdb_id': str(result.video_id),
+            'language': media_info_audio_languages,
+            'poster': result.poster_path,
+            'backdrop': result.backdrop_path,
+        }])
+
+    def wait_for_user_confirmation(self, message: str, on_confirm: Callable):
+        self.bot_error_log(message)
+        with ui.row():
+            ui.button("OK", on_click=on_confirm)
+            ui.button("Cancel",
+                      on_click=lambda: self.bot_error_log("Operation cancelled. Please update your config file."))
+
+    def user_input(self, message: str, on_submit: Callable[[int], None]):
+        with ui.row():
+            input_field = ui.input(label=message, placeholder="Enter TMDB ID").props("type=number")
+            ui.button("Submit", on_click=lambda: self._validate_and_submit(input_field.value, on_submit))
+
+    def user_input_str(self, message: str, on_submit: Callable[[str], None]):
+        with ui.row():
+            input_field = ui.input(label=message, placeholder="Enter text")
+            ui.button("Submit", on_click=lambda: on_submit(input_field.value or "0"))
+
+    def _add_log(self, message: str, color: str = "#fff"):
+        timestamp = datetime.now().strftime("%H:%M:%S")
+        with self.log_area:
+            ui.label(f"[{timestamp}] {message}").style(f'color: {color}; white-space: pre-wrap')
+
+    def _validate_and_submit(self, value: str, on_submit: Callable[[int], None]):
+        try:
+            if value.isdigit():
+                user_id = int(value)
+                on_submit(user_id if user_id < 9999999 else 0)
+            else:
+                self.bot_error_log("Please enter a valid number.")
+        except Exception as e:
+            self.bot_error_log(f"Error: {str(e)}")

view/settings.py

@@ -0,0 +1,31 @@
+from nicegui import ui
+
+class SettingsPanel:
+    def __init__(self):
+        with ui.column().classes('items-center q-pa-xl'):
+
+            # Header
+            ui.label("Settings").style('font-size: 32px; font-weight: bold; color: #2e3b4e')
+
+            # Settings card
+            with ui.card().style('max-width: 400px; width: 100%; padding: 20px; box-shadow: 0 2px 12px rgba(0, 0, 0, 0.1)'):
+
+                self._setting_group("Account Settings", [
+                    ("Enable notifications", True),
+                    ("Allow email updates", False),
+                ])
+
+                ui.separator().style("margin: 20px 0")
+
+                self._setting_group("Privacy", [
+                    ("Location tracking", False),
+                    ("Data sharing", True),
+                ])
+
+    def _setting_group(self, title: str, settings: list[tuple[str, bool]]):
+        ui.label(title).style('font-size: 18px; font-weight: 500; margin-bottom: 10px; color: #333')
+
+        for label_text, default_value in settings:
+            with ui.row().style('justify-content: space-between; align-items: center; margin-bottom: 12px'):
+                ui.label(label_text).style("font-size: 14px; color: #555")
+                ui.switch(value=default_value)

common/external_services/Pw/core/models/indexers.py

@@ -1,80 +0,0 @@
-# -*- coding: utf-8 -*-
-
-from dataclasses import dataclass, field
-
-
-@dataclass
-class SubCategory:
-    id: int = 0
-    name: str = ''
-    subCategories: list[any] = field(default_factory=list)
-
-
-@dataclass
-class Category:
-    id: int = 0
-    name: str = ''
-    subCategories: list[SubCategory] = field(default_factory=list)
-
-
-@dataclass
-class Capability:
-    bookSearchParams: list[str] = field(default_factory=list)
-    categories: list[Category] = field(default_factory=list)
-
-
-@dataclass
-class FieldOption:
-    hint: str = ''
-    name: str = ''
-    order: int = 0
-    value: int = 0
-
-
-@dataclass
-class Field:
-    advanced: bool = False
-    hidden: str = ''
-    isFloat: bool = False
-    name: str = ''
-    order: int = 0
-    privacy: str = ''
-    type: str = ''
-    value: str = ''
-    helpText: str = ''
-    label: str = ''
-    selectOptionsProviderAction: str = ''
-    selectOptions: list[FieldOption] = field(default_factory=list)
-    unit: str = ''
-
-
-@dataclass
-class Indexer:
-    added: str = ''
-    appProfileId: int = 0
-    capabilities: Capability = field(default_factory=Capability)
-    configContract: str = ''
-    definitionName: str = ''
-    description: str = ''
-    downloadClientId: int = 0
-    enable: bool = True
-    encoding: str = ''
-    fields: list[Field] = field(default_factory=list)
-    id: int = 0
-    implementation: str = ''
-    implementationName: str = ''
-    indexerUrls: list[str] = field(default_factory=list)
-    infoLink: str = ''
-    language: str = ''
-    legacyUrls: list[str] = field(default_factory=list)
-    name: str = ''
-    priority: int = 0
-    privacy: str = ''
-    protocol: str = ''
-    redirect: bool = False
-    sortName: str = ''
-    supportsPagination: bool = False
-    supportsRedirect: bool = False
-    supportsRss: bool = False
-    supportsSearch: bool = False
-    tags: list[int] = field(default_factory=list)

common/external_services/Pw/core/models/search.py

@@ -1,39 +0,0 @@
-# -*- coding: utf-8 -*-
-
-from dataclasses import dataclass, field
-
-
-@dataclass
-class Search:
-    """
-    Get a result from the endpoint /search?query
-
-    """
-
-    age: int = 0
-    ageHours: float = 0.0
-    ageMinutes: float = 0.0
-    categories: list[dict[str, any]] = field(default_factory=list)
-    downloadUrl: None | str = None
-    fileName: None | str = None
-    guid: None | str = None
-    imdbId: int = 0
-    indexer: None | str = None
-    indexerFlags: list[str] = field(default_factory=list)
-    indexerId: int = 0
-    infoUrl: None | str = None
-    leechers: int = 0
-    protocol: str = "torrent"
-    publishDate: None | str = None
-    seeders: int = 0
-    size: int = 0
-    sortTitle: None | str = None
-    title: None | str = None
-    tmdbId: int = 0
-    tvMazeId: int = 0
-    tvdbId: int = 0
-    files: int | None = None
-    grabs: int | None = None
-    posterUrl: str | None = None
-    infoHash: str | None = None
-

common/external_services/Pw/core/models/torrent_client_config.py

@@ -1,42 +0,0 @@
-# -*- coding: utf-8 -*-
-
-from dataclasses import dataclass, field
-
-
-@dataclass
-class SelectOption:
-    hint: str
-    name: str
-    order: int
-    value: int
-
-
-@dataclass
-class Field:
-    advanced: bool = False
-    isFloat: bool = False
-    label: str = ""
-    name: str = ""
-    order: int = 0
-    privacy: str = "normal"
-    type: str = "textbox"
-    value: str | int | bool | None = None
-    helpText: str | None = None
-    selectOptions: list[SelectOption] | None = None
-
-
-@dataclass
-class TorrentClientConfig:
-    categories: list[str] = field(default_factory=list)
-    configContract: str | None = None
-    enable: bool = True
-    fields: list[Field] = field(default_factory=list)
-    id: int = 0
-    implementation: str | None = None
-    implementationName: str | None = None
-    infoLink: str | None = None
-    name: str | None = None
-    priority: int = 0
-    protocol: str | None = None
-    supportsCategories: bool = False
-    tags: list[str] = field(default_factory=list)

common/external_services/Pw/core/pw_api.py

@@ -1,96 +0,0 @@
-# -*- coding: utf-8 -*-
-import os.path
-
-import httpx
-
-from common.external_services.Pw.core.models.torrent_client_config import (
-    TorrentClientConfig,
-)
-from common.external_services.Pw.core.models.indexers import Indexer
-from common.external_services.Pw.core.models.search import Search
-from common.external_services.sessions.session import MyHttp
-from common.external_services.sessions.agents import Agent
-from common import config_settings
-
-from view import custom_console
-
-class PwAPI(MyHttp):
-
-    def __init__(self):
-        """
-        Initialize the PwApi instance
-        """
-        headers = Agent.headers()
-        headers.update(
-            {"X-Api-Key": config_settings.options.PW_API_KEY, "Content-Type": "application/json"}
-        )
-
-        super().__init__(headers)
-        self.http_client = self.session
-        self.base_url = config_settings.options.PW_URL
-        self.api_key = config_settings.options.PW_API_KEY
-        self.dataclass = {f"{self.base_url}/indexer": Indexer}
-
-        if not config_settings.options.PW_URL:
-            custom_console.bot_question_log("No PW_URL provided\n")
-            exit(1)
-
-        if not config_settings.options.PW_API_KEY:
-            custom_console.bot_question_log("No PW_API_KEY provided\n")
-            exit(1)
-
-    def get_indexers(self) -> list[type[[Indexer]]]:
-        """Get all indexers."""
-
-        response = self.get_url(url=f"{self.base_url}/indexer", params={})
-
-        if response.status_code == 200:
-            indexers_list = response.json()
-            return [Indexer(**indexer) for indexer in indexers_list]
-        else:
-            return [Indexer]
-
-    def get_torrent_url(self, url: str, filename: str)-> httpx.Response :
-        return self.get_url(url=url)
-
-    def search(self, query: str) -> list[Search] | None:
-        """Get search queue."""
-
-        params = {"query": query}
-        url = f"{self.base_url}/search?"
-
-        response = self.get_url(url=url, params=params)
-        if response:
-            if response.status_code == 200:
-                results_list = response.json()
-                return [Search(**result) for result in results_list]
-            else:
-                return []
-        return None
-
-    def get_torrent_client_ids(self) -> list["TorrentClientConfig"]:
-        """Get a list of torrent client configurations"""
-
-        url = f"{self.base_url}/downloadclient"
-        response = self.get_url(url=url, params={})
-
-        if response.status_code == 200:
-            configurations_list = response.json()
-            return [
-                TorrentClientConfig(**client_config)
-                for client_config in configurations_list
-            ]
-        else:
-            return []
-
-    def send_torrent_to_client(self, payload):
-        """send torrent to client"""
-
-        url = f"{self.base_url}/downloadclient/1"
-        response = self.get_url(url=url, body=payload, get_method=False)
-
-        # TODO: Test again - get_url() updated 21/09/2024
-        if response.status_code == 202 or response.status_code == 200:
-            result = response.json()
-        else:
-            return []

common/external_services/Pw/pw_manager.py

@@ -1,63 +0,0 @@
-# -*- coding: utf-8 -*-
-
-import argparse
-import os
-
-from common.external_services.Pw.pw_service import PwService
-from common.utility import ManageTitles
-from common.database import Database
-from common import config_settings
-from unit3dup.media import Media
-
-
-from qbittorrent import Client
-from view import custom_console
-
-class PwManager:
-
-    def __init__(self,cli: argparse.Namespace):
-        # Keyword
-        self.search = cli.pw
-
-        # filename for the new download
-        self.filename = ManageTitles.normalize_filename(self.search)
-
-        # Select the tracker database
-        self.database = Database(db_file=cli.tracker)
-
-
-    def process(self):
-
-        # a new qbittorrent instance
-        qb = Client(f"http://{config_settings.torrent_client_config.QBIT_HOST}:{config_settings.torrent_client_config.QBIT_PORT}/")
-        # a new pw instance
-        pw_service = PwService()
-        # Query the indexers
-        search = pw_service.search(query=self.search)
-
-        content = []
-        if search:
-            for index, s in enumerate(search):
-                    if s.seeders > 0:
-                        category = s.categories[0]['name']
-                        if category in ['Movies','TV','TV/HD']:
-                            content.append(s)
-            custom_console.bot_process_table_pw(content=content)
-
-
-            for c in content:
-                test = Media(folder=f"c:\\test\\{c.fileName}", subfolder=f"c:\\test\\{c.fileName}")
-                print(f"[Prowlarr] {c.fileName}")
-                self.database.search(test.guess_title)
-
-
-            """"
-            qb.login(username=config_settings.torrent_client_config.QBIT_USER,
-                     password=config_settings.torrent_client_config.QBIT_PASS)
-
-            for torrent in content:
-                filename = str(os.path.join(config_settings.options.PW_TORRENT_ARCHIVE_PATH,torrent.fileName))
-                print(filename)
-                magnet = pw_service.get_torrent_from_pw(torrent_url=torrent.downloadUrl,download_filename=filename)
-                qb.download_from_link(magnet, savepath=config_settings.options.PW_DOWNLOAD_PATH)
-            """

common/external_services/Pw/pw_service.py

@@ -1,34 +0,0 @@
-# -*- coding: utf-8 -*-
-
-from common.external_services.Pw.core.pw_api import PwAPI
-from common.external_services.Pw.core.models.indexers import Indexer
-from common.external_services.Pw.core.models.search import Search
-from common.external_services.Pw.core.models.torrent_client_config import TorrentClientConfig
-
-
-class PwService:
-
-    def __init__(self):
-        self.pw_api = PwAPI()
-
-    def get_indexers(self) -> [Indexer]:
-        return self.pw_api.get_indexers()
-
-    def search(self, query: str) -> list[Search]:
-        return self.pw_api.search(query=query)
-
-    def get_torrent_client_ids(self) -> list[TorrentClientConfig]:
-        return self.pw_api.get_torrent_client_ids()
-
-    def send_torrent_to_client(self, payload):
-        return self.pw_api.send_torrent_to_client(payload)
-
-    def get_torrent_from_pw(self, torrent_url: str, download_filename: str)-> str | None:
-        response = self.pw_api.get_torrent_url(url=torrent_url, filename=download_filename)
-        # Redirect (PW)
-        if response.status_code == 301:
-            return response.headers.get('Location')
-        return None
-
-
-