def register():
msg = ""
if request.method == 'POST':
username = request.form['register_username']
u = User(username)
validUsername = u.validUser()
if u.newUser and validUsername:
u.add()
log("Adding new user {}".format(username))
url_root = request.url_root
t = multiprocessing.Process(target=u.sendCode, args=(url_root,))
t.start()
msg = 'We have successfully sent you a code. Please check your LYN inbox!'
category = "primary"
elif not validUsername:
msg = '{} is NOT a valid lowyat.net user.'.format(username)
category = "danger"
else:
msg = '{} is already a registered user.'.format(username)
category = "warning"
flash(msg, category)
return redirect(url_for("index"))
def init():
handler = pugsql.module('assets/sql')
handler.connect('sqlite:///memory')
settings.sql = handler
# Create table if they don't exist
handler.org_create()
handler.user_create()
handler.prof_create()
handler.fund_create()
handler.stat_create()
handler.value_create()
handler.role_create()
handler.role_user_create()
# Add test rows if they don't exist
org = handler.org_find(id=1)
if (org == None):
org = Organization.add('Test Company')
admin = User.add('admin', 'test123')
guest = User.add('guest', 'test123')
admin_role = Role.add('admin')
admin.add_to_role(role_id=admin_role.id)
profile = Profile.add('Test Profile', org.id)
fund = Fund.add('Test Fund', 'Test Manager', 2000, 0.00, 0.00,
profile.id)
# return the queries handler
return handler
def add(self, dict):
with Db.get() as self._db:
if "password" in dict:
dict["password"] = hashpw(dict["password"].encode('utf8'), gensalt())
user = User.add(self._db,dict)
self._db.commit()
self._db.refresh(user)
return user
def register():
form = RegisterForm()
if form.validate_on_submit():
user = User.find_byname(form.username.data)
if user is not None:
form.username.errors.append('Username is already taken')
else:
user = User.add(form.username.data, form.password.data)
login_user(user, remember=form.remember_me.data)
return redirect(url_for('organization_api.all'))
return render_template('accounts/register.html', form=form)
def sign_up():
error = None
if request.method == 'POST':
if User.exist(request.form['username']):
error = "username exist"
return render_template('sign_up.html', error=error)
else:
result = User.add(request.form['username'], request.form['password'])
flash('sign up success')
return redirect(url_for('index'))
else:
return render_template('sign_up.html', error=error)
