mirror of
https://github.com/aljazceru/CTFd.git
synced 2026-01-07 08:04:25 +01:00
76e5ad08a8
* Remove Python 2 specific code * Require imports to have a proper isort-supported order * Only test/lint on Python 3 * Bump most dependencies to latest supported version
63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
import cmarkgfm
|
|
from flask import current_app as app
|
|
|
|
# isort:imports-firstparty
|
|
from CTFd.cache import cache
|
|
from CTFd.models import Configs, db
|
|
|
|
string_types = (str,)
|
|
text_type = str
|
|
binary_type = bytes
|
|
|
|
|
|
def markdown(md):
|
|
return cmarkgfm.markdown_to_html_with_extensions(
|
|
md, extensions=["autolink", "table", "strikethrough"]
|
|
)
|
|
|
|
|
|
def get_app_config(key, default=None):
|
|
value = app.config.get(key, default)
|
|
return value
|
|
|
|
|
|
@cache.memoize()
|
|
def _get_config(key):
|
|
config = db.session.execute(
|
|
Configs.__table__.select().where(Configs.key == key)
|
|
).fetchone()
|
|
if config and config.value:
|
|
value = config.value
|
|
if value and value.isdigit():
|
|
return int(value)
|
|
elif value and isinstance(value, string_types):
|
|
if value.lower() == "true":
|
|
return True
|
|
elif value.lower() == "false":
|
|
return False
|
|
else:
|
|
return value
|
|
# Flask-Caching is unable to roundtrip a value of None.
|
|
# Return an exception so that we can still cache and avoid the db hit
|
|
return KeyError
|
|
|
|
|
|
def get_config(key, default=None):
|
|
value = _get_config(key)
|
|
if value is KeyError:
|
|
return default
|
|
else:
|
|
return value
|
|
|
|
|
|
def set_config(key, value):
|
|
config = Configs.query.filter_by(key=key).first()
|
|
if config:
|
|
config.value = value
|
|
else:
|
|
config = Configs(key=key, value=value)
|
|
db.session.add(config)
|
|
db.session.commit()
|
|
cache.delete_memoized(_get_config, key)
|
|
return config
|