You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.6 KiB
52 lines
1.6 KiB
import aiohttp, os, sys
|
|
from colors import *
|
|
import traceback
|
|
from steam import SteamID
|
|
|
|
class BackendClient:
|
|
up = False
|
|
|
|
def __init__(self):
|
|
if not os.getenv("BACKEND_URL", ""):
|
|
error("BACKEND_URL not setted in env")
|
|
sys.exit(10)
|
|
if not os.getenv("SECRET_KEY", ""):
|
|
error("SECRET_KEY not setted in env")
|
|
sys.exit(11)
|
|
self.secret_key = os.getenv("SECRET_KEY")
|
|
self.pulse_url = f"{os.getenv('BACKEND_URL')}/api/pulse"
|
|
self.vip_url = f"{os.getenv('BACKEND_URL')}/api/external/vip"
|
|
|
|
async def pulse(self):
|
|
async with aiohttp.ClientSession(cookies={
|
|
"secretkey":self.secret_key}) as session:
|
|
async with session.get(self.pulse, ssl=False) as response:
|
|
try:
|
|
await response.text()
|
|
self.up = True
|
|
except:
|
|
error("Backend not respond")
|
|
traceback.print_exc()
|
|
self.up = False
|
|
return self.up
|
|
|
|
async def vip(self, steamid: SteamID, amount: int, extra: str):
|
|
async with aiohttp.ClientSession(cookies={
|
|
"secretkey":self.secret_key}) as session:
|
|
async with session.post(self.vip_url + f"?steam={steamid.id64}&amount={amount}&service=steam&extra={extra}", ssl=False) as response:
|
|
try:
|
|
result = int(await response.text())
|
|
if result == 0:
|
|
warning(f"[S64:{steamid.id64}] VIP as not be added, maybe permition already exists")
|
|
return 99
|
|
elif result > 0:
|
|
info(f"[S64:{steamid.id64}] VIP has be added!")
|
|
return 100
|
|
elif result < 0:
|
|
info(f"[S64:{steamid.id64}] VIP has be extends!")
|
|
return 101
|
|
except:
|
|
error(f"[S64:{steamid.id64}] Backend returned error")
|
|
traceback.print_exc()
|
|
return False
|
|
return False
|
|
|