49 lines
2.1 KiB
Python
49 lines
2.1 KiB
Python
import httpx
|
|
from utils import *
|
|
from config import *
|
|
from sanic.log import logger
|
|
from metrics import *
|
|
import traceback
|
|
import asyncio
|
|
|
|
async def get(url, baseUrl=base_url_event, headers=headers, expectedStatusCodes=[200]) -> httpx.Response:
|
|
async def func(client : httpx.AsyncClient) -> httpx.Request:
|
|
return await client.get(join(baseUrl, url), headers=headers)
|
|
return await doReq(url, func, incPretixRead, expectedStatusCodes, "GETing")
|
|
|
|
async def post(url, content=None, json=None, baseUrl=base_url_event, headers=headers, expectedStatusCodes=[200]) -> httpx.Response:
|
|
async def func(client : httpx.AsyncClient) -> httpx.Request:
|
|
return await client.post(join(baseUrl, url), headers=headers, content=content, json=json)
|
|
return await doReq(url, func, incPretixWrite, expectedStatusCodes, "POSTing")
|
|
|
|
async def patch(url, json, baseUrl=base_url_event, headers=headers, expectedStatusCodes=[200]) -> httpx.Response:
|
|
async def func(client : httpx.AsyncClient) -> httpx.Request:
|
|
return await client.patch(join(baseUrl, url), headers=headers, json=json)
|
|
return await doReq(url, func, incPretixWrite, expectedStatusCodes, "PATCHing")
|
|
|
|
|
|
|
|
async def doReq(url, httpxFunc, metricsFunc, expectedStatusCodes, opLogString) -> httpx.Response:
|
|
res = None
|
|
async with httpx.AsyncClient(timeout=PRETIX_REQUESTS_TIMEOUT) as client:
|
|
requests = 0
|
|
for requests in range(PRETIX_REQUESTS_MAX):
|
|
try:
|
|
metricsFunc()
|
|
res = await httpxFunc(client)
|
|
|
|
if expectedStatusCodes is not None and res.status_code not in expectedStatusCodes:
|
|
incPretixErrors()
|
|
logger.warning(f"[PRETIX] Got an unexpected status code ({res.status_code}) while {opLogString} '{url}'. Allowed status codes: {', '.join(expectedStatusCodes)}")
|
|
continue
|
|
break
|
|
except Exception as e:
|
|
incPretixErrors()
|
|
logger.warning(f"[PRETIX] An error ({requests}) occurred while {opLogString} '{url}':\n{traceback.format_exc()}")
|
|
|
|
requests += 1
|
|
else:
|
|
logger.error(f"[PRETIX] Reached PRETIX_REQUESTS_MAX ({PRETIX_REQUESTS_MAX}) while {opLogString} '{url}'. Aborting")
|
|
raise httpx.TimeoutException(f"PRETIX_REQUESTS_MAX reached while {opLogString} to pretix.")
|
|
|
|
return res |