github/qBittorrent
1
0
Fork 0
mirror of https://github.com/qbittorrent/qBittorrent synced 2025-07-31 03:50:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Provide SSL context field

Browse source 
The allows the caller to provide proper SSL parameters and avoid dirty monkey patching to
suppress SSL errors.
This commit is contained in:
Chocobo1 2024-12-27 02:15:35 +08:00
parent 90e457a671
commit cc31a90931
No known key found for this signature in database
GPG key ID: 210D9C873253A68C
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/searchengine/nova3/helpers.py
View file

@ -35,6 +35,7 @@ import os
import re import re
import socket import socket
import socks import socks
import ssl
import sys import sys
import tempfile import tempfile
import urllib.error import urllib.error
@ -76,12 +77,12 @@ if "sock_proxy" in os.environ and len(os.environ["sock_proxy"].strip()) > 0:
htmlentitydecode = html.unescape htmlentitydecode = html.unescape
def retrieve_url(url: str, custom_headers: Mapping[str, Any] = {}, request_data: Optional[Any] = None) -> str: def retrieve_url(url: str, custom_headers: Mapping[str, Any] = {}, request_data: Optional[Any] = None, ssl_context: Optional[ssl.SSLContext] = None) -> str:
""" Return the content of the url page as a string """ """ Return the content of the url page as a string """
request = urllib.request.Request(url, request_data, {**headers, **custom_headers}) request = urllib.request.Request(url, request_data, {**headers, **custom_headers})
try: try:
response = urllib.request.urlopen(request) response = urllib.request.urlopen(request, context=ssl_context)
except urllib.error.URLError as errno: except urllib.error.URLError as errno:
print(f"Connection error: {errno.reason}", file=sys.stderr) print(f"Connection error: {errno.reason}", file=sys.stderr)
return "" return ""
@ -104,14 +105,14 @@ def retrieve_url(url: str, custom_headers: Mapping[str, Any] = {}, request_data:
return dataStr return dataStr
def download_file(url: str, referer: Optional[str] = None) -> str: def download_file(url: str, referer: Optional[str] = None, ssl_context: Optional[ssl.SSLContext] = None) -> str:
""" Download file at url and write it to a file, return the path to the file and the url """ """ Download file at url and write it to a file, return the path to the file and the url """
# Download url # Download url
request = urllib.request.Request(url, headers=headers) request = urllib.request.Request(url, headers=headers)
if referer is not None: if referer is not None:
request.add_header('referer', referer) request.add_header('referer', referer)
response = urllib.request.urlopen(request) response = urllib.request.urlopen(request, context=ssl_context)
data = response.read() data = response.read()
# Check if it is gzipped # Check if it is gzipped