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))