diff --git a/pynyaata/connectors/yggtorrent.py b/pynyaata/connectors/yggtorrent.py index d0a0358..0aa3742 100644 --- a/pynyaata/connectors/yggtorrent.py +++ b/pynyaata/connectors/yggtorrent.py @@ -15,8 +15,7 @@ class YggTorrent(ConnectorCore): color = 'is-success' title = 'YggTorrent' favicon = 'yggtorrent.png' - base_url = 'https://%s' % YGG_DOMAIN - hidden_url = 'http://%s' % YGG_IP + base_url = 'https://%s' % YGG_DOMAIN if not YGG_IP else 'http://%s' % YGG_IP is_light = False category = 2179 @@ -38,14 +37,9 @@ class YggTorrent(ConnectorCore): @ConnectorCache.cache_data def search(self): if self.category: - if YGG_IP: - params = self.get_full_search_url().split('?') - response = curl_content( - '%s/engine/search?%s' % (self.hidden_url, params[1]), - headers={"Host": YGG_DOMAIN}, - ) - else: - response = curl_content(self.get_full_search_url()) + response = curl_content( + self.get_full_search_url(), custom_host=YGG_DOMAIN if YGG_IP else None + ) if response['http_code'] == 200: html = BeautifulSoup(response['output'], 'html.parser')