34115477ab
These changes will make it easier to read through verbose logs with some tactful colors (yellow)
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
import subprocess
|
|
import os
|
|
import json
|
|
|
|
class Server:
|
|
def __init__(self, meta: dict):
|
|
self.url = meta.get('url')
|
|
self.wsurl = meta.get('wsurl')
|
|
self.serv_name = meta.get('name')
|
|
|
|
def __str__(self) -> str:
|
|
fields = {
|
|
'url': self.url,
|
|
'wsurl': self.wsurl,
|
|
'name': self.serv_name
|
|
}
|
|
return str(fields)
|
|
|
|
|
|
class Admin:
|
|
def __init__(self, user: dict, server: dict):
|
|
self.id = user.get('id')
|
|
self.name = user.get('name')
|
|
self.permissions = user.get('permissions')
|
|
self.secret = user.get('secret')
|
|
self.status = user.get('status')
|
|
self.jwt = None
|
|
|
|
self.server = Server(server)
|
|
|
|
def __str__(self) -> str:
|
|
return f'{self.name}#{self.id}'
|
|
|
|
def create_admin() -> Admin :
|
|
CARGO_BIN = os.getenv('CARGO_BIN')
|
|
|
|
proc = subprocess.run(
|
|
f'cargo run --release -- -c python-tester'.split(),
|
|
text=True, capture_output=True
|
|
)
|
|
try:
|
|
raw = json.loads(proc.stdout)
|
|
user = raw.get('user')
|
|
server = raw.get('server')
|
|
print(f'Raw data to use as user: {raw}')
|
|
if user is None or server is None:
|
|
return None
|
|
else:
|
|
return Admin(user, server)
|
|
except:
|
|
return None
|
|
|