Xefir/PyNyaaTa
Archived
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Michel Roux 2023-01-06 00:25:38 +01:00
parent 6c51312e09
commit 198af3c631
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pynyaata/connectors/yggtorrent.py
View File

@ -15,7 +15,8 @@ class YggTorrent(ConnectorCore):
color = 'is-success' color = 'is-success'
title = 'YggTorrent' title = 'YggTorrent'
favicon = 'yggtorrent.png' 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 is_light = False
category = 2179 category = 2179
@ -37,9 +38,14 @@ class YggTorrent(ConnectorCore):
@ConnectorCache.cache_data @ConnectorCache.cache_data
def search(self): def search(self):
if self.category: if self.category:
if YGG_IP:
params = self.get_full_search_url().split('?')
response = curl_content( 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: if response['http_code'] == 200:
html = BeautifulSoup(response['output'], 'html.parser') html = BeautifulSoup(response['output'], 'html.parser')