Unit3Dup 0.8.13__py3-none-any.whl → 0.8.15__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.13"
+version = "0.8.15"
 
 if os.name == "nt":
     PW_TORRENT_ARCHIVE_PATH: Path = Path(os.getenv("LOCALAPPDATA", ".")) / "Unit3Dup_config" / "pw_torrent_archive"

unit3dup/__main__.py

@@ -119,20 +119,12 @@ def main():
 
     # Watcher
     if cli.args.watcher:
-        bot = Bot(path=cli.args.watcher, cli=cli.args, mode="auto", trackers_name_list=tracker_name_list,
+        bot = Bot(path='', cli=cli.args, mode="auto", trackers_name_list=tracker_name_list,
                   torrent_archive_path=tracker_archive)
 
         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

unit3dup/media.py

@@ -14,7 +14,7 @@ class Media:
         self.folder: str = folder
         self.subfolder: str = subfolder
         self.title: str = os.path.basename(os.path.join(self.folder, self.subfolder))
-        self._is_tv: bool = False
+        self._not_title_lang = False
 
         # // Media
         self._crew_list: list[str] | None = None
@@ -53,6 +53,10 @@ class Media:
         self._igdb_id: int | None = None
         self._generate_title: str | None = None
 
+    @property
+    def no_title_lang(self) -> bool:
+        return self._not_title_lang
+
     @property
     def title_sanitized(self)-> str:
         if not self._title_sanitized:
@@ -279,6 +283,8 @@ class Media:
                 if converted_code := ManageTitles.convert_iso(code):
                     self._audio_languages = converted_code
                     return self._audio_languages
+            # no language found in title
+            self._not_title_lang = True
             # get from the audio track
             self._audio_languages = self.languages
         return self._audio_languages

unit3dup/media_manager/ContentManager.py

@@ -51,6 +51,10 @@ class ContentManager:
             content = self.get_data(media=media)
             if content:
                 contents.append(content)
+
+            # Add language to the title from the media file when it's absent
+            if not media._not_title_lang:
+                media.display_name = " ".join([media.display_name] + media.audio_languages)
         return contents
 
     def get_data(self, media: Media) -> Media | bool:

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.13.dist-info → unit3dup-0.8.15.dist-info}/METADATA

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

{unit3dup-0.8.13.dist-info → unit3dup-0.8.15.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=Y18iSTmFJg-OPSD5Z7wicGJpiWJ_l5p4R6siyE3DLl0,32782
+common/settings.py,sha256=pGdw3qAC_fqSncyxqRNRYec3N75EhiUWTS5wy8JV8CE,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,19 +63,19 @@ 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=CDazL8WFGv_Y6y875s3Ej-0QK5MPytbTJm_vsPHWB-A,9270
+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=dgmFaXHb_F992EF67tvXqHJ0nvfhpE2CwIjrGT-O9wo,9188
 unit3dup/duplicate.py,sha256=Ji1Y9vVLmhFQ5rgkUU-s7uXl-jC8akAa8OiQOfjFa9g,10428
 unit3dup/exceptions.py,sha256=DhlcbbZEuvEYk5VSZTnle8oRg5I8Pq6Y6fxXTdOOOfo,4078
-unit3dup/media.py,sha256=9WxBBE9051W8mKbClCABPw3ngSfdCFgzXynz2uUa5Ao,13710
+unit3dup/media.py,sha256=5suH2XeekB80zNoIAM8PERZBO746JDlpcIvHz-4jY9E,13888
 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=ReZ1Fg3d2QXHVKnCv6VvYXcAcaDyejA4YDvruqf41qQ,7142
+unit3dup/media_manager/ContentManager.py,sha256=68k8sqJgL1a2ir_rpQhkB1ntePp01IpblJYcQCdev2Y,7358
 unit3dup/media_manager/DocuManager.py,sha256=oFt7jlxj-gIUty9PADBQV5a24bsv3yhjKhwI6niOhf4,3116
 unit3dup/media_manager/GameManager.py,sha256=9EmPeNrirOwaVOj-vkLr29Xo7daIA4ssqrrt0WyMl30,4090
 unit3dup/media_manager/MediaInfoManager.py,sha256=0NO9dgD7seJM67B3DRnwvRIdoy7bfquBUox-PnHInK8,1081
@@ -95,11 +86,11 @@ 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.13.dist-info/licenses/LICENSE,sha256=GNAZDLhU0xz8QPbIyHAOYlVnQYDvKWk2N9fZJMhqaG8,1090
+unit3dup-0.8.15.dist-info/licenses/LICENSE,sha256=GNAZDLhU0xz8QPbIyHAOYlVnQYDvKWk2N9fZJMhqaG8,1090
 view/__init__.py,sha256=XIzb6rl58HmYPnksD73cYMFF88dn6FHa3u7bOHFbChk,81
 view/custom_console.py,sha256=OITmkEoQH9N_uE5ElPaSdc8XvaLzE9wcwTbOHtcMvrI,5629
-unit3dup-0.8.13.dist-info/METADATA,sha256=m33LV7uVhDsgmFLX62hagVe35rDjIFuJQgxQHqeTANQ,4864
-unit3dup-0.8.13.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-unit3dup-0.8.13.dist-info/entry_points.txt,sha256=fxXSyI6-r6jy9_v-C5ZHm03q1aC3tE9EvCQZxC1NQnI,52
-unit3dup-0.8.13.dist-info/top_level.txt,sha256=19NVMnQNkJxBUKebRNaYCRs56A5CO4U1L67GMQCPiLU,21
-unit3dup-0.8.13.dist-info/RECORD,,
+unit3dup-0.8.15.dist-info/METADATA,sha256=wWNaUq7iZ_jCVHSHMNBDdQMaCSwlwORirOzhSi69nUg,4864
+unit3dup-0.8.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+unit3dup-0.8.15.dist-info/entry_points.txt,sha256=fxXSyI6-r6jy9_v-C5ZHm03q1aC3tE9EvCQZxC1NQnI,52
+unit3dup-0.8.15.dist-info/top_level.txt,sha256=19NVMnQNkJxBUKebRNaYCRs56A5CO4U1L67GMQCPiLU,21
+unit3dup-0.8.15.dist-info/RECORD,,

{unit3dup-0.8.13.dist-info → unit3dup-0.8.15.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.8.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

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
-
-
-