def check_service(host: str) -> Verdict:
try:
with build_session() as session:
api = Api(host, session)
resp = api.register_user(Randomizer.user())
if resp.status_code != 201:
return Verdict.MUMBLE("Can't register user", "Can't register user")
file_in_zip, *file = create_zip()
resp = api.upload_zip(file)
if resp.status_code != 202:
return Verdict.MUMBLE("Can't upload file", "Can't upload file")
resp = api.search_file(file_in_zip)
if file_in_zip not in resp.text:
return Verdict.MUMBLE("Can't find file from zip", "Can't find file from zip")
resp = api.create_note(get_text(), True)
if resp.status_code != 201:
return Verdict.MUMBLE("Can't create note", "Can't create note")
return Verdict.OK()
except Exception as e:
return Verdict.DOWN("Can't connect to service", str(e))
def put_note_flag(host: str, flag_id: str, flag: str) -> Verdict:
try:
with build_session() as session:
api = Api(host, session)
user = Randomizer.user()
resp = api.register_user(user)
if resp.status_code != 201:
return Verdict.MUMBLE("Can't register user", "Can't register user")
resp = api.create_note(flag, False)
if resp.status_code != 201:
return Verdict.MUMBLE("Can't create note", "Can't create note")
return Verdict.OK(f"{user['login']}:{user['pwd']}")
except Exception as e:
return Verdict.DOWN("Can't connect to service", str(e))
