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

fixy req

Browse Source
This commit is contained in:
zman 2025-04-06 18:38:32 -04:00
parent d3bd696d67
commit 3f53513c36
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/services/task.py
View File

@ -26,7 +26,7 @@ class TaskService:
def register_scheduled_tasks(self):
self.scheduler.add_job(self.hourly_pricing, 'cron', minute='36')
self.scheduler.add_job(self.hourly_orders, 'cron', hour='*', minute='26')
self.scheduler.add_job(self.hourly_orders, 'cron', hour='*', minute='42')
# every 5 hours on the 24th minute
#self.scheduler.add_job(self.inventory_pricing, 'cron', hour='*', minute='44')
self.logger.info("Scheduled tasks registered.")

12
app/services/util/_requests.py
View File

@ -61,10 +61,14 @@ class RequestHeaders:
'origin': Headers.SELLER_ORIGIN,
'referer': Headers.SELLER_REFERER
}
POST_HEADERS = {
'content-type': 'application/json'
}
class URLHeaders:
# combine base and seller headers
ORDER_HEADERS = {**RequestHeaders.BASE_HEADERS, **RequestHeaders.SELLER_HEADERS}
POST_HEADERS = {**RequestHeaders.BASE_HEADERS, **RequestHeaders.SELLER_HEADERS, **RequestHeaders.POST_HEADERS}
class RequestsUtil:
def __init__(self, browser_type: Browser = Browser.BRAVE):
@ -119,7 +123,7 @@ class RequestsUtil:
def send_request(self, url: str, method: str, cookies: dict, data=None) -> requests.Response:
"""Send a request with the specified cookies"""
headers = self.set_headers(url)
headers = self.set_headers(url, method)
if not headers:
raise ValueError("Headers not set")
@ -135,10 +139,12 @@ class RequestsUtil:
logger.error(f"Request failed: {str(e)}")
return None
def set_headers(self, url: str):
def set_headers(self, url: str, method: str) -> Dict:
# use tcgplayerendpoints enum to set headers where url partially matches enum value
for endpoint in TCGPlayerEndpoints:
if endpoint.value in url:
if endpoint.value in url and str.upper(method) == "POST":
return URLHeaders.POST_HEADERS
elif endpoint.value in url:
return URLHeaders.ORDER_HEADERS
else:
raise ValueError(f"Endpoint not found in TCGPlayerEndpoints: {url}")