pingu_98/MrDraw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added decorator for before firstrun only calls.

Browse source
This commit is contained in:
Teja 2017-09-08 16:06:36 +02:00
parent 4c6dfed137
commit c7e72d21c9
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/octoprint/server/util/flask.py
View file

@ -1108,6 +1108,23 @@ def restricted_access(func):
return decorated_view return decorated_view
def firstrun_only_access(func):
"""
If you decorate a view with this, it will ensure that first setup has _not_ been
done for OctoPrint's Access Control. Otherwise it
will cause a HTTP 403 status code to be returned by the decorated resource.
"""
@functools.wraps(func)
def decorated_view(*args, **kwargs):
# if OctoPrint has been set up yet, abort
if settings().getBoolean(["server", "firstRun"]) and (octoprint.server.userManager is None or not octoprint.server.userManager.hasBeenCustomized()):
return func(*args, **kwargs)
else:
return flask.make_response("OctoPrint is already setup, this resource is not longer available.", 403)
return decorated_view
class AppSessionManager(object): class AppSessionManager(object):
VALIDITY_UNVERIFIED = 1 * 60 # 1 minute VALIDITY_UNVERIFIED = 1 * 60 # 1 minute