Skip to content

Commit

Permalink
Merge pull request #378 from bill88t/master
Browse files Browse the repository at this point in the history
  • Loading branch information
RicterZ authored Jan 23, 2025
2 parents a83c571 + 04bd88a commit 4ab43da
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 12 deletions.
2 changes: 1 addition & 1 deletion nhentai/downloader.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ async def download(self, url, folder='', filename='', retried=0, proxy=None, len
logger.warning(f'Skipped download: {save_file_path} already exists')
return 1, url

response = await async_request('GET', url, timeout=self.timeout, proxies=proxy)
response = await async_request('GET', url, timeout=self.timeout, proxy=proxy)

if response.status_code != 200:
path = urlparse(url).path
Expand Down
16 changes: 6 additions & 10 deletions nhentai/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,24 +36,20 @@ def request(method, url, **kwargs):
return getattr(session, method)(url, verify=False, **kwargs)


async def async_request(method, url, proxies = None, **kwargs):
async def async_request(method, url, proxy = None, **kwargs):
headers = {
'Referer': constant.LOGIN_URL,
'User-Agent': constant.CONFIG['useragent'],
'Cookie': constant.CONFIG['cookie'],
}

if proxies is None:
proxies = constant.CONFIG['proxy']
if proxy is None:
proxy = constant.CONFIG['proxy']

if proxies.get('http') == '' and proxies.get('https') == '':
proxies = None
if proxy is not None and proxy.get('http') == '' and proxy.get('https') == '':
proxy = None

if proxies:
_proxies = {f'{k}://': v for k, v in proxies.items() if v}
proxies = _proxies

async with httpx.AsyncClient(headers=headers, verify=False, proxies=proxies, **kwargs) as client:
async with httpx.AsyncClient(headers=headers, verify=False, proxy=proxy, **kwargs) as client:
response = await client.request(method, url, **kwargs)

return response
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
httpx==0.27.2
httpx
requests
soupsieve
setuptools
Expand Down

0 comments on commit 4ab43da

Please sign in to comment.