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