hswaw
/
bitvend
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remember login by default, fix horizontal scrolling

feature/cython
informatic 2017-06-02 15:41:00 +02:00
parent 55dab93eb8
commit 8f1d906e9c
3 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
bitvend/admin.py
View File

@ -1,5 +1,5 @@
from flask import Blueprint, render_template, redirect, request, flash, url_for
from flask_login import current_user
from flask_login import current_user, fresh_login_required
from bitvend.models import db, Transaction
from bitvend.forms import ManualForm
@ -10,6 +10,7 @@ admin_required = cap_required('staff')
bp = Blueprint('admin', __name__)
@bp.route('/manual', methods=['GET', 'POST'])
@fresh_login_required
@admin_required
def manual():
form = ManualForm()
@ -24,6 +25,7 @@ def manual():
@bp.route('/transactions/', defaults={'page': 1})
@bp.route('/transactions/p/<int:page>')
@fresh_login_required
@admin_required
def transactions(page):
return render_template('admin/transactions.html',

7
bitvend/auth.py
View File

@ -9,6 +9,11 @@ from sqlalchemy import func
login_manager = LoginManager()
login_manager.refresh_view = "bitvend.login"
login_manager.needs_refresh_message = (
u"To protect your account, please reauthenticate to access this page."
)
login_manager.needs_refresh_message_category = "info"
@login_manager.user_loader
def load_user(user_id):
@ -28,7 +33,7 @@ def try_login(username, password):
db.session.add(u)
db.session.commit()
login_user(u)
login_user(u, remember=True)
return True

8
bitvend/templates/base.html
View File

@ -36,6 +36,9 @@
opacity: 0.5;
font-style: italic;
}
.vend-item code {
word-wrap: break-word;
}
</style>
</head>
<body>
@ -48,6 +51,11 @@
<span class="icon-bar"></span>
<span class="icon-bar"></span>
</button>
{% if current_user.is_authenticated %}
<a href="{{ url_for('bitvend.logout') }}" class="navbar-brand pull-right hidden-md hidden-lg"><small>Logout</small></a>
{% else %}
<a href="{{ url_for('bitvend.login') }}" class="navbar-brand pull-right hidden-md hidden-lg"><small>Login</small></a>
{% endif %}
</div>
<div class="navbar-collapse collapse" id="navbar-main">
<p class="navbar-text navbar-nav">