mirror of
https://github.com/norohind/FDEV-CAPI-Handler.git
synced 2025-06-08 03:12:28 +03:00
code style fixes
This commit is contained in:
parent
318bbb8626
commit
24950cb8d7
7
main.py
7
main.py
@ -87,6 +87,7 @@ class Authorize:
|
|||||||
sql_connection.execute(
|
sql_connection.execute(
|
||||||
"insert into authorizations (code_verifier, state, timestamp_init) values "
|
"insert into authorizations (code_verifier, state, timestamp_init) values "
|
||||||
"(?, ?, ?);", [code_verifier.decode('utf-8'), state_string, int(time.time())])
|
"(?, ?, ?);", [code_verifier.decode('utf-8'), state_string, int(time.time())])
|
||||||
|
|
||||||
sql_connection.close()
|
sql_connection.close()
|
||||||
resp.content_type = falcon.MEDIA_HTML
|
resp.content_type = falcon.MEDIA_HTML
|
||||||
resp.text = REDIRECT_HTML_TEMPLATE.format(link=redirect_user_to)
|
resp.text = REDIRECT_HTML_TEMPLATE.format(link=redirect_user_to)
|
||||||
@ -107,6 +108,7 @@ class FDEV_redirect:
|
|||||||
if code_verifier is None:
|
if code_verifier is None:
|
||||||
# Somebody got here not by frontier redirect
|
# Somebody got here not by frontier redirect
|
||||||
raise falcon.HTTPBadRequest(description="Don't show here without passing frontier authorization!")
|
raise falcon.HTTPBadRequest(description="Don't show here without passing frontier authorization!")
|
||||||
|
|
||||||
code_verifier = code_verifier[0]
|
code_verifier = code_verifier[0]
|
||||||
|
|
||||||
with sql_connection:
|
with sql_connection:
|
||||||
@ -121,8 +123,10 @@ class FDEV_redirect:
|
|||||||
if token_request.status_code != 200:
|
if token_request.status_code != 200:
|
||||||
# Something went wrong
|
# Something went wrong
|
||||||
print(f"error_id: {error_id}\n{token_request.text}")
|
print(f"error_id: {error_id}\n{token_request.text}")
|
||||||
|
|
||||||
with sql_connection:
|
with sql_connection:
|
||||||
sql_connection.execute("delete from authorizations where state = ?;", [state])
|
sql_connection.execute("delete from authorizations where state = ?;", [state])
|
||||||
|
|
||||||
sql_connection.close()
|
sql_connection.close()
|
||||||
raise falcon.HTTPBadRequest(description=f'Something went wrong, your error id is {error_id}')
|
raise falcon.HTTPBadRequest(description=f'Something went wrong, your error id is {error_id}')
|
||||||
|
|
||||||
@ -146,6 +150,7 @@ class FDEV_redirect:
|
|||||||
try:
|
try:
|
||||||
sql_connection.execute("update authorizations set nickname = ? where state = ?;", [nickname, state])
|
sql_connection.execute("update authorizations set nickname = ? where state = ?;", [nickname, state])
|
||||||
sql_connection.commit()
|
sql_connection.commit()
|
||||||
|
|
||||||
except sqlite3.IntegrityError:
|
except sqlite3.IntegrityError:
|
||||||
already_saved_state = sql_connection.execute(
|
already_saved_state = sql_connection.execute(
|
||||||
"select state from authorizations where nickname = ?", [nickname]).fetchone()[0]
|
"select state from authorizations where nickname = ?", [nickname]).fetchone()[0]
|
||||||
@ -160,6 +165,7 @@ class FDEV_redirect:
|
|||||||
print(f"error_id: {error_id}\n{e}")
|
print(f"error_id: {error_id}\n{e}")
|
||||||
with sql_connection:
|
with sql_connection:
|
||||||
sql_connection.execute("delete from authorizations where state = ?;", [state])
|
sql_connection.execute("delete from authorizations where state = ?;", [state])
|
||||||
|
|
||||||
else: # all fine
|
else: # all fine
|
||||||
resp.text = f"All right, {nickname}, we have your tokens\nSave and keep this key: {state}"
|
resp.text = f"All right, {nickname}, we have your tokens\nSave and keep this key: {state}"
|
||||||
sql_connection.close()
|
sql_connection.close()
|
||||||
@ -208,6 +214,7 @@ class Admin:
|
|||||||
raise falcon.HTTPNotFound(description="You aren't an admin")
|
raise falcon.HTTPNotFound(description="You aren't an admin")
|
||||||
|
|
||||||
users_html = ""
|
users_html = ""
|
||||||
|
|
||||||
for user in sql_connection.execute(
|
for user in sql_connection.execute(
|
||||||
'select nickname, state from authorizations where nickname is not null;').fetchall():
|
'select nickname, state from authorizations where nickname is not null;').fetchall():
|
||||||
user_name = user[0]
|
user_name = user[0]
|
||||||
|
@ -58,6 +58,7 @@ def refresh_one_token(_state: str, _force: bool = False):
|
|||||||
sql_connection.commit()
|
sql_connection.commit()
|
||||||
sql_connection.close()
|
sql_connection.close()
|
||||||
raise e
|
raise e
|
||||||
|
|
||||||
sql_connection.close()
|
sql_connection.close()
|
||||||
|
|
||||||
|
|
||||||
@ -68,4 +69,5 @@ def refresh_all_suitable_tokens():
|
|||||||
for state in suitable_tokens.fetchall():
|
for state in suitable_tokens.fetchall():
|
||||||
state = state[0]
|
state = state[0]
|
||||||
refresh_one_token(state)
|
refresh_one_token(state)
|
||||||
|
|
||||||
sql_connection.close()
|
sql_connection.close()
|
||||||
|
Loading…
x
Reference in New Issue
Block a user