MarseyWorld/files/helpers/cloudflare.py

import json
from typing import List, Union, Optional
from files.helpers.const import *
import requests

CLOUDFLARE_API_URL = "https://api.cloudflare.com/client/v4"
CLOUDFLARE_REQUEST_TIMEOUT_SECS = 5
DEFAULT_CLOUDFLARE_ZONE = 'blahblahblah'

def _request_from_cloudflare(url:str, method:str, post_data_str) -> bool:
	if CF_ZONE == DEFAULT_CLOUDFLARE_ZONE: return False
	try:
		res = str(requests.request(method, f"{CLOUDFLARE_API_URL}/zones/{CF_ZONE}/{url}", headers=CF_HEADERS, data=post_data_str, timeout=CLOUDFLARE_REQUEST_TIMEOUT_SECS))
	except:
		return False
	return res == "<Response [200]>"

def get_security_level() -> Optional[str]:
	res = None
	try:
		res = requests.get(f'{CLOUDFLARE_API_URL}/zones/{CF_ZONE}/settings/security_level', headers=CF_HEADERS, timeout=CLOUDFLARE_REQUEST_TIMEOUT_SECS).json()['result']['value']
	except:
		pass
	return res

def set_security_level(under_attack="high") -> bool:
	return _request_from_cloudflare("settings/security_level", "PATCH", f'{{"value":"{under_attack}"}}')

def clear_cloudflare_cache() -> bool:
	return _request_from_cloudflare("clear_cloudflare_cache", "POST", '{"purge_everything":true}')

def purge_files_in_cache(files:Union[List[str],str]) -> bool:
	if CF_ZONE == DEFAULT_CLOUDFLARE_ZONE: return False
	if isinstance(files, str):
		files = [files]
	post_data = {"files": files}
	res = None
	try:
		res = requests.post(f'{CLOUDFLARE_API_URL}/zones/{CF_ZONE}/purge_cache', headers=CF_HEADERS, data=json.dumps(post_data), timeout=CLOUDFLARE_REQUEST_TIMEOUT_SECS)
	except:
		return False
	return res == "<Response [200]>"