plexflow 0.0.64__py3-none-any.whl

plexflow/core/torrents/providers/torrentquest/utils.py

@@ -0,0 +1,70 @@
+from datetime import datetime
+from plexflow.core.torrents.results.torrent import Torrent
+from plexflow.utils.imdb.imdb_codes import IMDbCode
+from plexflow.utils.torrent.hash import extract_torrent_hash
+
+class TorrentQuestSearchResult(Torrent):
+    def __init__(self, **kwargs):
+        super().__init__()
+        self._magnet = kwargs.get('magnet_link')
+        self._name = kwargs.get('name')
+        self._date = kwargs.get('date')
+        self._type = kwargs.get('type')
+        self._size = kwargs.get('size_bytes')
+        self._seeds = kwargs.get('seeds')
+        self._peers = kwargs.get('peers')
+        self._files = kwargs.get('files')
+        self._link = kwargs.get('link')
+        self.src = 'torrentquest'
+
+    @property
+    def source(self) -> str:
+        return self.src
+
+    @property
+    def magnet(self) -> str:
+        return self._magnet
+
+    @property
+    def date(self) -> datetime:
+        return self._date
+
+    @property
+    def seeds(self) -> int:
+        return self._seeds
+
+    @property
+    def peers(self) -> int:
+        return self._peers
+
+    @property
+    def n_files(self) -> int:
+        return self._files
+
+    @property
+    def size_bytes(self) -> int:
+        return self._size
+
+    @property
+    def imdb_code(self) -> IMDbCode:
+        return None
+
+    @property
+    def release_name(self) -> str:
+        return self._name
+
+    @property
+    def hash(self) -> str:
+        return extract_torrent_hash(self._magnet)
+
+    @property
+    def uploader(self) -> str:
+        return None
+
+    @property
+    def url(self) -> str:
+        return self._link
+
+    @property
+    def category(self) -> str:
+        return self._type

plexflow/core/torrents/providers/tpb/tpb.py

@@ -0,0 +1,17 @@
+from plexflow.utils.api.rest.restful import Restful
+from plexflow.core.torrents.providers.tpb.utils import TPBSearchResult
+
+class TPB(Restful):
+    def __init__(self, base_url: str = 'https://apibay.org'):
+        super().__init__(base_url=base_url)
+    
+    def search(self, query: str):
+        response = self.get('/q.php', query_params={
+            'q': query,
+        })
+        
+        response.raise_for_status()
+        
+        data = response.json()
+        
+        return list(map(lambda x: TPBSearchResult(**x), data))

plexflow/core/torrents/providers/tpb/utils.py

@@ -0,0 +1,139 @@
+from datetime import datetime
+from plexflow.core.torrents.results.torrent import Torrent
+from plexflow.utils.imdb.imdb_codes import IMDbCode
+
+class TPBSearchResult(Torrent):
+    added: str
+    category: str
+    id: str
+    imdb: str
+    info_hash: str
+    leechers: str
+    name: str
+    num_files: str
+    seeders: str
+    size: str
+    status: str
+    username: str
+
+    def __init__(self, **kwargs):
+        super().__init__()
+        self.added = kwargs.get("added")
+        self.category = kwargs.get("category")
+        self.id = kwargs.get("id")
+        self.imdb = kwargs.get("imdb")
+        self.info_hash = kwargs.get("info_hash")
+        self.leechers = kwargs.get("leechers")
+        self.name = kwargs.get("name")
+        self.num_files = kwargs.get("num_files")
+        self.seeders = kwargs.get("seeders")
+        self.size = kwargs.get("size")
+        self.status = kwargs.get("status")
+        self.username = kwargs.get("username")
+        self.src = "tpb"
+
+    @property
+    def magnet(self) -> str:
+        return f'magnet:?xt=urn:btih:{self.info_hash}'
+
+    @property
+    def date(self) -> datetime:
+        return datetime.fromtimestamp(int(self.added))
+
+    @property
+    def seeds(self) -> int:
+        return int(self.seeders)
+
+    @property
+    def peers(self) -> int:
+        return int(self.leechers)
+
+    @property
+    def n_files(self) -> int:
+        return int(self.num_files)
+
+    @property
+    def size_bytes(self) -> int:
+        return int(self.size)
+
+    @property
+    def imdb_code(self) -> IMDbCode:
+        return IMDbCode(self.imdb)
+
+    @property
+    def release_name(self) -> str:
+        return self.name
+
+    @property
+    def hash(self) -> str:
+        return self.info_hash
+
+    @property
+    def uploader(self) -> str:
+        return self.username
+
+    @property
+    def torrent_id(self) -> str:
+        return self.id
+
+    @property
+    def url(self) -> str:
+        return f'https://apibay.org/t.php?id={self.id}'
+
+    @property
+    def category_name(self) -> str:
+        categories = {
+            '100': 'Audio',
+            '101': 'Music',
+            '102': 'Audio books',
+            '103': 'Sound clips',
+            '104': 'FLAC',
+            '199': 'Other',
+            '200': 'Video',
+            '201': 'Movies',
+            '202': 'Movies DVDR',
+            '203': 'Music videos',
+            '204': 'Movie clips',
+            '205': 'TV shows',
+            '206': 'Handheld',
+            '207': 'HD - Movies',
+            '208': 'HD - TV shows',
+            '209': '3D',
+            '299': 'Other',
+            '300': 'Applications',
+            '301': 'Windows',
+            '302': 'Mac',
+            '303': 'UNIX',
+            '304': 'Handheld',
+            '305': 'IOS (iPad/iPhone)',
+            '306': 'Android',
+            '399': 'Other',
+            '400': 'Games',
+            '401': 'PC',
+            '402': 'Mac',
+            '403': 'PSx',
+            '404': 'XBOX360',
+            '405': 'Wii',
+            '406': 'Handheld',
+            '407': 'IOS (iPad/iPhone)',
+            '408': 'Android',
+            '499': 'Other',
+            '500': 'Porn',
+            '501': 'Movies',
+            '502': 'Movies DVDR',
+            '503': 'Pictures',
+            '504': 'Games',
+            '505': 'HD - Movies',
+            '506': 'Movie clips',
+            '599': 'Other',
+            '600': 'Other',
+            '601': 'E-books',
+            '602': 'Comics',
+            '603': 'Pictures',
+            '604': 'Covers',
+            '605': 'Physibles',
+            '699': 'Other'
+        }
+        return categories.get(self.category, 'Unknown')
+
+    

plexflow/core/torrents/providers/yts/utils.py

@@ -0,0 +1,57 @@
+from datetime import datetime
+from plexflow.core.torrents.results.torrent import Torrent
+from plexflow.utils.imdb.imdb_codes import IMDbCode
+
+class YTSSearchResult(Torrent):
+    def __init__(self, **kwargs):
+        super().__init__()
+        self._url = kwargs.get("url")
+        self._hash = kwargs.get("hash")
+        self._quality = kwargs.get("quality")
+        self._type = kwargs.get("type")
+        self._is_repack = kwargs.get("is_repack")
+        self._video_codec = kwargs.get("video_codec")
+        self._bit_depth = kwargs.get("bit_depth")
+        self._audio_channels = kwargs.get("audio_channels")
+        self._seeds = kwargs.get("seeds")
+        self._peers = kwargs.get("peers")
+        self._size = kwargs.get("size")
+        self._size_bytes = kwargs.get("size_bytes")
+        self._date_uploaded = kwargs.get("date_uploaded")
+        self._name = kwargs.get("name")
+        self._imdb_code = kwargs.get("imdb_code")
+        self._date_uploaded_unix = kwargs.get("date_uploaded_unix")
+        self.src = "yts"
+
+    @property
+    def magnet(self) -> str:
+        return f'magnet:?xt=urn:btih:{self._hash}'
+
+    @property
+    def date(self) -> datetime:
+        return datetime.fromtimestamp(int(self._date_uploaded_unix))
+
+    @property
+    def seeds(self) -> int:
+        return self._seeds
+
+    @property
+    def peers(self) -> int:
+        return self._peers
+
+    @property
+    def size_bytes(self) -> int:
+        return self._size_bytes
+
+    @property
+    def imdb_code(self) -> IMDbCode:
+        return IMDbCode(self._imdb_code)
+
+    @property
+    def release_name(self) -> str:
+        return self._name
+
+    @property
+    def hash(self) -> str:
+        return self._hash
+

plexflow/core/torrents/providers/yts/yts.py

@@ -0,0 +1,31 @@
+from plexflow.utils.api.rest.restful import Restful
+from plexflow.core.torrents.providers.yts.utils import YTSSearchResult
+
+class YTS(Restful):
+    def __init__(self, base_url: str = 'https://yts.mx'):
+        super().__init__(base_url=base_url)
+    
+    def search(self, query: str):
+        response = self.get(url='/api/v2/list_movies.json', query_params={
+            'query_term': query,
+        })
+        
+        response.raise_for_status()
+        
+        data = response.json()
+        
+        data = data.get("data", {})
+        movies = data.get("movies", [])
+        
+        results = []
+        
+        for m in movies:
+            torrents = m.get("torrents", [])
+            torrents = map(lambda t: YTSSearchResult(**{
+                **t,
+                "name": "_".join(filter(lambda p: p, [m.get("slug"), t.get("type"), t.get("quality"), t.get("video_codec")])) + "-YTS",
+                "imdb_code": m.get("imdb_code")
+                }), torrents)
+            results.extend(torrents)
+            
+        return results

plexflow/core/torrents/results/torrent.py

@@ -0,0 +1,165 @@
+from abc import ABC, abstractmethod
+from datetime import datetime
+import PTN
+from typing import List, Union
+from plexflow.utils.imdb.imdb_codes import IMDbCode
+
+class Torrent(ABC):
+    """
+    This is an abstract base class that represents a Torrent.
+    Any class that inherits from this must implement all the properties.
+    """
+
+    @property
+    @abstractmethod
+    def seeds(self) -> int:
+        """
+        This property represents the number of seeds for the torrent.
+        It must be implemented by any class that inherits from this class.
+        """
+        pass
+    
+    @property
+    @abstractmethod
+    def peers(self) -> int:
+        """
+        This property represents the number of peers for the torrent.
+        It must be implemented by any class that inherits from this class.
+        """
+        pass
+    
+    @property
+    @abstractmethod
+    def release_name(self) -> str:
+        """
+        This property represents the release name of the torrent.
+        It must be implemented by any class that inherits from this class.
+        """
+        pass
+    
+    @property
+    @abstractmethod
+    def size_bytes(self) -> int:
+        """
+        This property represents the size of the torrent in bytes.
+        It must be implemented by any class that inherits from this class.
+        """
+        pass
+    
+    @property
+    @abstractmethod
+    def magnet(self) -> str:
+        """
+        This property represents the magnet link of the torrent.
+        It must be implemented by any class that inherits from this class.
+        """
+        pass
+    
+    @property
+    @abstractmethod
+    def hash(self) -> str:
+        """
+        This property represents the hash of the torrent.
+        It must be implemented by any class that inherits from this class.
+        """
+        pass
+    
+    @property
+    def uploader(self) -> str:
+        """
+        This property represents the uploader of the torrent.
+        It must be implemented by any class that inherits from this class.
+        """
+        return self.source
+    
+    @property
+    @abstractmethod
+    def date(self) -> datetime:
+        """
+        This property represents the date of the torrent upload.
+        It must be implemented by any class that inherits from this class.
+        """
+        pass
+    
+    @property
+    @abstractmethod
+    def imdb_code(self) -> IMDbCode:
+        """
+        This property represents the IMDB code of the torrent.
+        It must be implemented by any class that inherits from this class.
+        """
+        pass
+    
+    @property
+    def source(self) -> str:
+        """
+        This property represents the source of the torrent.
+        It must be implemented by any class that inherits from this class.
+        """
+        return self.src
+    
+    @property
+    def torrent_id(self) -> str:
+        """
+        This property represents the id of the torrent.
+        It must be implemented by any class that inherits from this class.
+        """
+        return self.source + "_" + self.hash
+
+    @property
+    def parsed_release_name(self) -> dict:
+        return PTN.parse(self.release_name)
+    
+    @property
+    def encoder_name(self) -> str:
+        parts = self.parsed_release_name
+        return parts.get("encoder")
+
+    @property
+    def season(self) -> Union[int, List[int]]:
+        parts = self.parsed_release_name
+        return parts.get("season")
+    
+    @property
+    def episode(self) -> Union[int, List[int]]:
+        parts = self.parsed_release_name
+        return parts.get("episode")
+
+    @property
+    def has_multiple_episodes(self):
+        tmp = self.episode
+        return isinstance(tmp, list) and len(tmp) > 1
+
+    @property
+    def has_multiple_seasons(self):
+        tmp = self.season
+        return isinstance(tmp, list) and len(tmp) > 1
+        
+    @property
+    def title(self) -> str:
+        parts = self.parsed_release_name
+        return parts.get("title")
+        
+    @property
+    def year(self) -> int:
+        parts = self.parsed_release_name
+        return parts.get("year")
+
+    @property
+    def quality(self) -> str:
+        parts = self.parsed_release_name
+        return parts.get("quality")
+
+    @property
+    def is_bad_quality(self):
+        parsed_title = self.parsed_release_name
+        bad_qualities = ['CAM', 'HDCAM', 'TS', 'HDTS', 'TELESYNC', 
+                         'PDVD', 'PREDVDRIP', 'WP', 'WORKPRINT', 'TC', 
+                         'HDTC', 'TELECINE', 'PPV', 'PPVRIP', 'SCR', 'SCREENER', 
+                         'DVDSCR', 'DVDSCREENER', 'BDSCR', 'WEBSCREENER', 'DDC', 
+                         'R5', 'R5.LINE', 'R5.AC3.5.1.HQ']
+        return parsed_title.get('quality', '').upper() in bad_qualities
+
+    @property
+    def url(self) -> str:
+        return None