def add_categories(): categories = json.loads(request.data) db.getset( "user:%s:categories" % current_user.user, json.dumps(categories, sort_keys=True, indent=4, separators=(",", ": ")), ) return "Success"
def signup(): if current_user is not None and current_user.is_authenticated(): return redirect(url_for("index")) form = LoginForm() if form.validate_on_submit(): user = User(username=form.email.data, password=form.password.data) if user.is_valid_login(): login_user(user) return redirect(url_for("index")) elif form.new_account.data: login_user(user) # add login, add password db.getset("user:%s:login" % user.user, user.username) db.getset("user:%s:password" % user.user, user.password) # add to email list db.sadd("user:users", user.user) # add "untrained" to models db.getset("user:%s:trained" % user.user, "false") # add category category = [{"id": 1, "emails": 3, "split": 0, "name": "inbox", "class": "category-inbox"}] categoryJSON = json.dumps(category, sort_keys=True, indent=4, separators=(",", ": ")) db.getset("user:%s:categories" % user.user, categoryJSON) return redirect(url_for("index")) return render_template("signup.html", form=form, title="Sign In")
def train_models(): retrain_models(current_user.user) db.getset("user:%s:trained" % current_user.user, "true") return "Success"