public/Zdalne-systemy-z-kit-uVPN
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

dostosowanie do code review

Browse Source
This commit is contained in:
Mateusz779 2023-04-24 08:33:51 +02:00
parent ac3e7f7edf
commit d397dfa11f
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
app.py
View File

@ -27,6 +27,13 @@ def login_required(f):
return f(*args, **kwargs) return f(*args, **kwargs)
return login_function return login_function
def is_logged(auth_token):
if auth_token != "" or auth_token is not None:
if db.get_user_bytoken(auth_token) is not None:
return True
return False
@app.route('/') @app.route('/')
@login_required @login_required
def main(): def main():
@ -35,20 +42,20 @@ def main():
@app.route('/login') @app.route('/login')
@login_required
def login(): def login():
return render_template('index.html', ssh_port=config.webssh_port, machines=machines_all.machines) if is_logged(request.cookies.get('auth_token')) is True:
machines_all = db.get_machines()
return render_template('index.html', ssh_port=config.webssh_port, machines=machines_all.machines)
return render_template('login.html')
@app.route('/logout') @app.route('/logout')
def logout(): def logout():
auth_token = request.cookies.get('auth_token') if is_logged(request.cookies.get('auth_token')) is True:
if auth_token != "" or auth_token is not None: db.del_auth_token(request.cookies.get('auth_token'))
if db.get_user_bytoken(auth_token) is not None: response = make_response(redirect('/'))
db.del_auth_token(auth_token) response.delete_cookie('auth_token')
response = make_response(redirect('/')) return response
response.delete_cookie('auth_token')
return response
return render_template('login.html') return render_template('login.html')