Move inventory api key to const

This commit is contained in:
Madis Mägi 2023-08-18 03:54:11 +03:00
parent f1cb8e6cfe
commit dc34c66c6e
2 changed files with 2 additions and 2 deletions

View File

@ -9,7 +9,6 @@ from common import CustomForm, build_query, flatten, format_name, spam, User
page_api = Blueprint("api", __name__) page_api = Blueprint("api", __name__)
db = MongoClient(const.MONGO_URI).get_default_database() db = MongoClient(const.MONGO_URI).get_default_database()
api_key = os.getenv("INVENTORY_API_KEY")
def check_api_key(f): def check_api_key(f):
@wraps(f) @wraps(f)
@ -18,7 +17,7 @@ def check_api_key(f):
if not request_key: if not request_key:
return "nope", 403 return "nope", 403
found_key = re.search(r"Basic (.*)", request_key).group(1) found_key = re.search(r"Basic (.*)", request_key).group(1)
if not found_key or found_key != api_key: if not found_key or found_key != const.INVENTORY_API_KEY:
return "nope", 403 return "nope", 403
return f(*args, **kwargs) return f(*args, **kwargs)
return decorated_function return decorated_function

View File

@ -19,3 +19,4 @@ MONGO_URI = os.environ["MONGO_URI"]
MEMBERS_HOST = os.environ["MEMBERS_HOST"] MEMBERS_HOST = os.environ["MEMBERS_HOST"]
SLACK_VERIFICATION_TOKEN = os.environ["SLACK_VERIFICATION_TOKEN"] # used to verify (deprecated) incoming requests from slack SLACK_VERIFICATION_TOKEN = os.environ["SLACK_VERIFICATION_TOKEN"] # used to verify (deprecated) incoming requests from slack
SLACK_DOORLOG_CALLBACK = os.environ["SLACK_DOORLOG_CALLBACK"] # used for sending logs to private channel SLACK_DOORLOG_CALLBACK = os.environ["SLACK_DOORLOG_CALLBACK"] # used for sending logs to private channel
INVENTORY_API_KEY = os.environ["INVENTORY_API_KEY"]