def login(): form = LoginForm() if request.method == 'GET': return render_template('login.html', form=form, variables=variables, captcha=captcha.create()) if form.validate_on_submit(): username = request.form.get('username') password = request.form.get('password') captcha_hash = request.form.get('captcha-hash') captcha_text = request.form.get('captcha_text') redirect_url = request.args.get(variables['redirect_url_param_name'], default='/') if not captcha.verify(captcha_text, captcha_hash): flash('Captcha is not valid.', 'error') return redirect( f'{url_for("login")}?{variables["redirect_url_param_name"]}={redirect_url}' ) user = User.get(username=username, password=password) if user: if login_user(user): return redirect(redirect_url) flash('Username or password is invalid.', 'error') return redirect( f'{url_for("login")}?{variables["redirect_url_param_name"]}={redirect_url}' )
def ajax(): username = request.form["username"] exists = False if User.get(username): exists = True else: exists = False return jsonify({"exists": exists})
def update(self, id): data = json.loads(request.data) u = User.get(Key(id)) if 'current_cells' in data: u.current_cells = data['current_cells'] if 'is_admin' in data: if data['is_admin']: u.role = "admin" else: u.role = "editor" u.put() return Response(u.as_json(), mimetype='application/json')
def buy(): user = User.get(user_id=session["user_id"]) stock = None setSessionStock("buystock") if "symbol" in session["buystock"]: stock = Stock.get(session["buystock"]["symbol"]) print("stock set to: {}".format(stock.name)) # Stock will be None unless search has been made. But user won't get option to do anything with stock unless there's been a search. form = BuyForm(user=user, stock=stock) if request.method == "POST": if form.validate_on_submit(): return redirect(url_for("transactions.buy")) print(form.errors) return render_template("/buy.html", form=form)
def get(self, id): u = User.get(Key(id)) return Response(u.as_json(), mimetype='application/json')
def uniqueUser(form, field): if User.get(field.data): raise ValidationError('Username is already taken')
def existingUser(form, field): if not User.get(field.data): raise ValidationError('User does not exist')
def load_user(user_id): return User.get(int(user_id))