zman27/giga_tcg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

aaa

Browse Source
This commit is contained in:
zman 2025-02-07 21:23:32 -05:00
parent 13c96b1643
commit f44d5740fc
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
app/services/tcgplayer.py
View File

@ -118,6 +118,16 @@ class TCGPlayerService:
logger.error(f"Failed to get browser cookies: {str(e)}") logger.error(f"Failed to get browser cookies: {str(e)}")
return None return None
def is_in_docker():
"""Check if we're running inside a Docker container"""
try:
with open('/proc/1/cgroup', 'r') as f:
return 'docker' in f.read()
except:
# If we can't read the file, check for common Docker env vars
return os.environ.get('DOCKER_CONTAINER', False) or os.environ.get('IN_DOCKER', False)
def _send_request(self, url: str, method: str, data=None, except_302=False) -> requests.Response: def _send_request(self, url: str, method: str, data=None, except_302=False) -> requests.Response:
"""Send a request with the specified cookies""" """Send a request with the specified cookies"""
# if previous request was made less than 10 seconds ago, wait until current time is 10 seconds after previous request # if previous request was made less than 10 seconds ago, wait until current time is 10 seconds after previous request
@ -128,10 +138,11 @@ class TCGPlayerService:
time.sleep(10 - time_diff) time.sleep(10 - time_diff)
headers = self._set_headers(method) headers = self._set_headers(method)
# only get cookies on mac os not in docker container # only get cookies on mac os not in docker container
if not self.cookies and os.name != 'nt': if not self.cookies:
self.cookies = self._get_browser_cookies() if self.is_in_docker():
else:
self.cookies = self.get_cookies_from_file() self.cookies = self.get_cookies_from_file()
else:
self.cookies = self._get_browser_cookies()
if not self.cookies: if not self.cookies:
raise ValueError("Failed to retrieve browser cookies") raise ValueError("Failed to retrieve browser cookies")