furizon_webint/pretixClient.py

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 : httpx.Response = 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(map(str, expectedStatusCodes))}")
					logger.debug(f"Response: '{res.text}'")
					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