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"
