diff options
author | rubenwardy <rw@rubenwardy.com> | 2018-05-15 19:35:59 +0100 |
---|---|---|
committer | rubenwardy <rw@rubenwardy.com> | 2018-05-15 19:35:59 +0100 |
commit | dbc56283ebb0d850fff65ecaf6753c0fa31e0c53 (patch) | |
tree | 78fc13668b963d2b6cd9cf3f48f18c7f155d00e6 /app/views/utils.py | |
parent | 976f83c5507e203107d5592a480b3b55343a0224 (diff) | |
download | cheatdb-dbc56283ebb0d850fff65ecaf6753c0fa31e0c53.tar.xz |
Add screenshot importing from github
Diffstat (limited to 'app/views/utils.py')
-rw-r--r-- | app/views/utils.py | 132 |
1 files changed, 0 insertions, 132 deletions
diff --git a/app/views/utils.py b/app/views/utils.py deleted file mode 100644 index 44647a1..0000000 --- a/app/views/utils.py +++ /dev/null @@ -1,132 +0,0 @@ -from flask import request, flash, abort, redirect -from flask_user import * -from flask_login import login_user, logout_user -from app.models import * -from app import app -import random, string, os - -def getExtension(filename): - return filename.rsplit(".", 1)[1].lower() if "." in filename else None - -def isFilenameAllowed(filename, exts): - return getExtension(filename) in exts - -def shouldReturnJson(): - return "application/json" in request.accept_mimetypes and \ - not "text/html" in request.accept_mimetypes - -def randomString(n): - return ''.join(random.choice(string.ascii_lowercase + \ - string.ascii_uppercase + string.digits) for _ in range(n)) - -def doFileUpload(file, allowedExtensions, fileTypeName): - if not file or file is None or file.filename == "": - flash("No selected file", "error") - return None - - ext = getExtension(file.filename) - if ext is None or not ext in allowedExtensions: - flash("Please upload load " + fileTypeName, "error") - return None - - filename = randomString(10) + "." + ext - file.save(os.path.join(app.config["UPLOAD_FOLDER"], filename)) - return "/uploads/" + filename - - -def _do_login_user(user, remember_me=False): - def _call_or_get(v): - if callable(v): - return v() - else: - return v - - # User must have been authenticated - if not user: - return False - - user.active = True - if not user.rank.atLeast(UserRank.NEW_MEMBER): - user.rank = UserRank.NEW_MEMBER - - db.session.commit() - - # Check if user account has been disabled - if not _call_or_get(user.is_active): - flash("Your account has not been enabled.", "error") - return False - - # Check if user has a confirmed email address - user_manager = current_app.user_manager - if user_manager.enable_email and user_manager.enable_confirm_email \ - and not current_app.user_manager.enable_login_without_confirm_email \ - and not user.has_confirmed_email(): - url = url_for("user.resend_confirm_email") - flash("Your email address has not yet been confirmed", "error") - return False - - # Use Flask-Login to sign in user - login_user(user, remember=remember_me) - signals.user_logged_in.send(current_app._get_current_object(), user=user) - - flash("You have signed in successfully.", "success") - - return True - -def loginUser(user): - user_mixin = None - if user_manager.enable_username: - user_mixin = user_manager.find_user_by_username(user.username) - - return _do_login_user(user_mixin, False) - -def rank_required(rank): - def decorator(f): - @wraps(f) - def decorated_function(*args, **kwargs): - if not current_user.is_authenticated: - return redirect(url_for("user.login")) - if not current_user.rank.atLeast(rank): - abort(403) - - return f(*args, **kwargs) - - return decorated_function - return decorator - -def getPackageByInfo(author, name): - user = User.query.filter_by(username=author).first() - if user is None: - abort(404) - - package = Package.query.filter_by(name=name, author_id=user.id).first() - if package is None: - abort(404) - - return package - -def is_package_page(f): - @wraps(f) - def decorated_function(*args, **kwargs): - if not ("author" in kwargs and "name" in kwargs): - abort(400) - - package = getPackageByInfo(kwargs["author"], kwargs["name"]) - - del kwargs["author"] - del kwargs["name"] - - return f(package=package, *args, **kwargs) - - return decorated_function - -def triggerNotif(owner, causer, title, url): - if owner.rank.atLeast(UserRank.NEW_MEMBER) and owner != causer: - Notification.query.filter_by(user=owner, url=url).delete() - notif = Notification(owner, causer, title, url) - db.session.add(notif) - -def clearNotifications(url): - if current_user.is_authenticated: - Notification.query.filter_by(user=current_user, url=url).delete() - db.session.commit() |