sudacode/owocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use urllib for Google Lens web

Browse Source
This commit is contained in:
AuroraWright
2025-04-09 22:26:23 +02:00
parent 748ed9dd84
commit 8dac199a8f
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
owocr/ocr.py
View File

@@ -319,7 +319,6 @@ class GoogleLensWeb:
if 'pyjson5' not in sys.modules: if 'pyjson5' not in sys.modules:
logger.warning('pyjson5 not available, Google Lens (web) will not work!') logger.warning('pyjson5 not available, Google Lens (web) will not work!')
else: else:
self.regex = re.compile(r'(\w+)=([^&]+)')
self.requests_session = requests.Session() self.requests_session = requests.Session()
self.available = True self.available = True
logger.info('Google Lens (web) ready') logger.info('Google Lens (web) ready')
@@ -363,13 +362,18 @@ class GoogleLensWeb:
if res.status_code != 303: if res.status_code != 303:
return (False, 'Unknown error!') return (False, 'Unknown error!')
location_params = dict(self.regex.findall(res.headers['Location'])) redirect_url = res.headers.get('Location')
if not redirect_url:
return (False, 'Error getting redirect URL!')
if ('vsrid' not in location_params) or ('gsessionid' not in location_params): parsed_url = urlparse(redirect_url)
query_params = parse_qs(parsed_url.query)
if ('vsrid' not in query_params) or ('gsessionid' not in query_params):
return (False, 'Unknown error!') return (False, 'Unknown error!')
try: try:
res = self.requests_session.get(f"https://lens.google.com/qfmetadata?vsrid={location_params['vsrid']}&gsessionid={location_params['gsessionid']}", timeout=20) res = self.requests_session.get(f"https://lens.google.com/qfmetadata?vsrid={query_params['vsrid'][0]}&gsessionid={query_params['gsessionid'][0]}", timeout=20)
except requests.exceptions.Timeout: except requests.exceptions.Timeout:
return (False, 'Request timeout!') return (False, 'Request timeout!')
except requests.exceptions.ConnectionError: except requests.exceptions.ConnectionError: