2021-12-28 08:39:12 +00:00
|
|
|
import flask
|
2022-10-27 18:10:08 +00:00
|
|
|
from flask import Blueprint
|
|
|
|
from flask import current_app as app
|
|
|
|
from flask import g
|
2021-12-27 07:54:55 +00:00
|
|
|
from flask.views import View
|
2021-12-28 12:26:56 +00:00
|
|
|
from flask_login import current_user, login_required, login_user, logout_user
|
2022-04-29 11:10:44 +00:00
|
|
|
from sqlalchemy import or_
|
2021-12-27 07:54:55 +00:00
|
|
|
|
2022-04-11 15:16:20 +00:00
|
|
|
from ereuse_devicehub import __version__, messages
|
|
|
|
from ereuse_devicehub.db import db
|
2022-10-27 18:10:08 +00:00
|
|
|
from ereuse_devicehub.forms import LoginForm, PasswordForm
|
2022-04-29 11:10:44 +00:00
|
|
|
from ereuse_devicehub.resources.action.models import Trade
|
|
|
|
from ereuse_devicehub.resources.lot.models import Lot
|
2022-10-27 18:10:08 +00:00
|
|
|
from ereuse_devicehub.resources.user.models import User
|
2021-12-28 08:39:12 +00:00
|
|
|
from ereuse_devicehub.utils import is_safe_url
|
2021-12-22 23:29:44 +00:00
|
|
|
|
|
|
|
core = Blueprint('core', __name__)
|
|
|
|
|
|
|
|
|
2022-04-05 08:49:27 +00:00
|
|
|
@core.route("/")
|
|
|
|
def index():
|
|
|
|
return flask.redirect(flask.url_for('core.login'))
|
|
|
|
|
|
|
|
|
2021-12-27 08:15:06 +00:00
|
|
|
class LoginView(View):
|
2021-12-28 08:39:12 +00:00
|
|
|
methods = ['GET', 'POST']
|
2021-12-27 08:15:06 +00:00
|
|
|
template_name = 'ereuse_devicehub/user_login.html'
|
|
|
|
|
|
|
|
def dispatch_request(self):
|
2021-12-28 08:39:12 +00:00
|
|
|
form = LoginForm()
|
|
|
|
if form.validate_on_submit():
|
|
|
|
# Login and validate the user.
|
|
|
|
# user should be an instance of your `User` class
|
|
|
|
user = User.query.filter_by(email=form.email.data).first()
|
2021-12-29 07:10:26 +00:00
|
|
|
login_user(user, remember=form.remember.data)
|
2021-12-28 08:39:12 +00:00
|
|
|
|
|
|
|
next_url = flask.request.args.get('next')
|
|
|
|
# is_safe_url should check if the url is safe for redirects.
|
|
|
|
# See http://flask.pocoo.org/snippets/62/ for an example.
|
|
|
|
if not is_safe_url(flask.request, next_url):
|
|
|
|
return flask.abort(400)
|
|
|
|
|
2022-03-15 13:33:27 +00:00
|
|
|
return flask.redirect(next_url or flask.url_for('inventory.devicelist'))
|
2022-10-27 18:10:08 +00:00
|
|
|
|
|
|
|
url_register = "#"
|
2022-11-11 16:42:29 +00:00
|
|
|
url_reset_password = "#"
|
|
|
|
|
2022-10-27 18:10:08 +00:00
|
|
|
if 'register' in app.blueprints.keys():
|
|
|
|
url_register = flask.url_for('register.user-registration')
|
|
|
|
|
2022-11-11 16:42:29 +00:00
|
|
|
if 'reset_password' in app.blueprints.keys():
|
|
|
|
url_reset_password = flask.url_for('reset_password.reset-password')
|
|
|
|
|
|
|
|
context = {
|
|
|
|
'form': form,
|
|
|
|
'version': __version__,
|
|
|
|
'url_register': url_register,
|
|
|
|
'url_reset_password': url_reset_password,
|
|
|
|
}
|
2022-10-27 18:10:08 +00:00
|
|
|
|
|
|
|
return flask.render_template(self.template_name, **context)
|
2021-12-27 08:15:06 +00:00
|
|
|
|
|
|
|
|
2021-12-28 12:05:41 +00:00
|
|
|
class LogoutView(View):
|
|
|
|
def dispatch_request(self):
|
|
|
|
logout_user()
|
|
|
|
return flask.redirect(flask.url_for('core.login'))
|
|
|
|
|
|
|
|
|
2022-05-16 15:52:31 +00:00
|
|
|
class GenericMixin(View):
|
2022-05-18 09:01:58 +00:00
|
|
|
methods = ['GET']
|
2021-12-28 08:44:04 +00:00
|
|
|
decorators = [login_required]
|
2021-12-27 07:54:55 +00:00
|
|
|
|
2022-04-29 11:10:44 +00:00
|
|
|
def get_lots(self):
|
|
|
|
return (
|
|
|
|
Lot.query.outerjoin(Trade)
|
|
|
|
.filter(
|
|
|
|
or_(
|
|
|
|
Trade.user_from == g.user,
|
|
|
|
Trade.user_to == g.user,
|
|
|
|
Lot.owner_id == g.user.id,
|
|
|
|
)
|
|
|
|
)
|
|
|
|
.distinct()
|
|
|
|
)
|
|
|
|
|
|
|
|
def get_context(self):
|
|
|
|
self.context = {
|
|
|
|
'lots': self.get_lots(),
|
2022-03-15 10:46:32 +00:00
|
|
|
'version': __version__,
|
2021-12-28 12:26:56 +00:00
|
|
|
}
|
2022-04-11 15:16:20 +00:00
|
|
|
|
2022-04-29 11:10:44 +00:00
|
|
|
return self.context
|
|
|
|
|
|
|
|
|
2022-05-16 15:52:31 +00:00
|
|
|
class UserProfileView(GenericMixin):
|
2022-04-29 11:10:44 +00:00
|
|
|
decorators = [login_required]
|
|
|
|
template_name = 'ereuse_devicehub/user_profile.html'
|
|
|
|
|
|
|
|
def dispatch_request(self):
|
|
|
|
self.get_context()
|
|
|
|
self.context.update(
|
|
|
|
{
|
|
|
|
'current_user': current_user,
|
|
|
|
'password_form': PasswordForm(),
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
|
|
|
return flask.render_template(self.template_name, **self.context)
|
2021-12-27 07:54:55 +00:00
|
|
|
|
|
|
|
|
2022-04-11 17:48:59 +00:00
|
|
|
class UserPasswordView(View):
|
|
|
|
methods = ['POST']
|
|
|
|
decorators = [login_required]
|
|
|
|
|
|
|
|
def dispatch_request(self):
|
|
|
|
form = PasswordForm()
|
|
|
|
db.session.commit()
|
|
|
|
if form.validate_on_submit():
|
|
|
|
form.save(commit=False)
|
|
|
|
messages.success('Reset user password successfully!')
|
|
|
|
else:
|
|
|
|
messages.error('Error modifying user password!')
|
|
|
|
|
|
|
|
db.session.commit()
|
|
|
|
return flask.redirect(flask.url_for('core.user-profile'))
|
|
|
|
|
|
|
|
|
2021-12-27 08:15:06 +00:00
|
|
|
core.add_url_rule('/login/', view_func=LoginView.as_view('login'))
|
2021-12-28 12:05:41 +00:00
|
|
|
core.add_url_rule('/logout/', view_func=LogoutView.as_view('logout'))
|
2021-12-27 07:54:55 +00:00
|
|
|
core.add_url_rule('/profile/', view_func=UserProfileView.as_view('user-profile'))
|
2022-04-11 17:48:59 +00:00
|
|
|
core.add_url_rule('/set_password/', view_func=UserPasswordView.as_view('set-password'))
|