aljaz/CTFd
1
0
Fork 0
mirror of https://github.com/aljazceru/CTFd.git synced 2025-12-17 22:14:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Properly checking for config values and setting them if they don't exist

Browse Source
This commit is contained in:
Kevin Chung
2015-01-18 19:33:48 -05:00
parent 4a128e83dc
commit 1882d8231d
1 changed files with 25 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
CTFd/admin.py
View File

@@ -74,9 +74,31 @@ def init_admin(app):
db.session.commit()
return redirect('/admin/config')
start = Config.query.filter_by(key="start").first().value
end = Config.query.filter_by(key="end").first().value
view_challenges_unregistered = (Config.query.filter_by(key='view_challenges_unregistered').first().value == '1')
start = Config.query.filter_by(key="start").first()
if start:
start = start.value
else:
start = Config('start', None)
db.session.add(start)
end = Config.query.filter_by(key="end").first()
if end:
end = end.value
else:
end = Config('end', None)
db.session.add(end)
view_challenges_unregistered = Config.query.filter_by(key='view_challenges_unregistered').first()
if view_challenges_unregistered:
view_challenges_unregistered = (view_challenges_unregistered.value == '1')
else:
view_challenges_unregistered = Config('view_challenges_unregistered', None)
db.session.add(view_challenges_unregistered)
db.session.commit()
db.session.close()
return render_template('admin/config.html', start=start, end=end, view_challenges_unregistered=view_challenges_unregistered)
@app.route('/admin/pages', defaults={'route': None}, methods=['GET', 'POST'])