Don't disclose YGG_IP
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Michel Roux 2023-01-06 00:25:38 +01:00
parent 6c51312e09
commit 198af3c631

View File

@ -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:
if YGG_IP:
params = self.get_full_search_url().split('?')
response = curl_content(
self.get_full_search_url(), custom_host=YGG_DOMAIN if YGG_IP else None
'%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')