def createuser():
form = CreateUserForm()
error = None
if (request.method == "POST"):
if form.validate() == False:
flash_errors(form)
return redirect(url_for('createuser', error=error))
else:
uid = random.randint(1, 1000)
firstname = form.firstname.data
lastname = form.lastname.data
gender = form.gender.data
email = form.email.data
location = form.location.data
bio = form.bio.data
file = request.files['image']
image = secure_filename(file.filename)
created_on = datetime.now().strftime("%a %d %b %Y")
file.save(os.path.join("app/static/images", image))
user = UserProfile(uid, firstname, lastname, gender, email,
location, bio, image, created_on)
db.session.add(user)
db.session.commit()
flash('USER CREATED SUCESSFULLY', 'success')
return redirect(url_for('createuser', error=error))
flash_errors(form)
return render_template('createuser.html', form=form, error=error)
def createUser():
createUserForm = CreateUserForm(request.form)
if request.method == 'POST' and createUserForm.validate():
print('SimpleApp Ln 115')
usersDict = {}
db = shelve.open('storage.db', 'c')
try:
print('SimpleApp Ln 119')
usersDict = db['Users']
except IOError:
print("IOError")
except:
print("Error in retrieving Users from storage.db.")
finally:
user = User.User(createUserForm.firstName.data,
createUserForm.lastName.data,
createUserForm.DOB.data,
createUserForm.gender.data,
createUserForm.email.data, createUserForm.pw.data,
createUserForm.confirmpw.data)
usersDict[user.get_email()] = user
db['Users'] = usersDict
db.close()
return redirect(url_for('home'))
return render_template('createUser.html', form=createUserForm)
def signup():
form = CreateUserForm()
if request.method == 'POST' and form.validate():
#generate user id
userid = createID()
#gets today's date
profile_add_on = timeinfo()
#creating confirmaton code
ccode = form.email.data[:3]+userid
#creating a default user photo
defaultimg = 'img/'+userid+'.png'
shutil.copyfile(UPLOAD_FOLDER+'noprofileimage.png', UPLOAD_FOLDER+userid+'.png')
# Saving profile to database and setting to inactive
user = Profiles(userid=userid, password=form.password.data, email=form.email.data, \
profile_add_on=profile_add_on, code=ccode, highscore1=0, highscore2=0, tdollars=0, image=defaultimg)
db.session.add(user)
db.session.commit()
# sending confirmation email
sendcode(form.email.data, ccode)
flash('Please complete registration by verifying your email')
return render_template('signup.html', form=form)
else:
return render_template('signup.html', form=form)
def createUser():
createUserForm = CreateUserForm(request.form)
if request.method == 'POST' and createUserForm.validate():
usersDict = {}
db = shelve.open('storage.db', 'c')
try:
usersDict = db['Users']
except:
print("Error in retrieving Users from storage.db.")
user = User.User(createUserForm.firstName.data,
createUserForm.lastName.data,
createUserForm.membership.data,
createUserForm.gender.data,
createUserForm.remarks.data)
usersDict[user.get_userID()] = user
db['Users'] = usersDict
db.close()
return redirect(url_for('retrieveUsers'))
return redirect(url_for('home'))
return render_template('createUser.html', form=createUserForm)
def admin_user():
create_form = CreateUserForm()
create_form.roles.query = Role.query.all()
delete_form = DeleteUserForm()
delete_form.users.query = User.query.all()
update_form = UpdateUserForm()
update_form.users.query = User.query.all()
update_form.roles.query = Role.query.all()
if request.method == 'POST':
if create_form.data['create'] and create_form.validate():
form = create_form
user = User.query.filter_by(user_id=form.create_id.data).first()
role = form.roles.data
if user == None:
user = User(user_id=form.create_id.data,
user_name=form.create_name.data)
user.password = '******'
user.role = role
db.session.add(user)
db.session.commit()
session['status'] = u'success'
flash(u'成功添加登录用户信息!')
return redirect(url_for('admin_user'))
elif delete_form.data['delete'] and delete_form.validate():
form = delete_form
user = form.users.data
db.session.delete(user)
db.session.commit()
session['status'] = u'warning'
flash(u'成功删除登录用户信息!')
return redirect(url_for('admin_user'))
elif update_form.data['update'] and update_form.validate():
form = update_form
user = form.users.data
role = form.roles.data
name = form.update_name.data
password = form.update_passwd.data
user.role = role
if password != '':
user.password = password
elif name != '':
user.user_name = name
db.session.add(user)
db.session.commit()
session['status'] = u'info'
flash(u'成功修改登录用户信息!')
return redirect(url_for('admin_user'))
return render_template('admin-user.html',
create_form=create_form,
delete_form=delete_form,
update_form=update_form)
def admin_user():
create_form = CreateUserForm()
create_form.roles.query = Role.query.all()
delete_form = DeleteUserForm()
delete_form.users.query = User.query.all()
update_form = UpdateUserForm()
update_form.users.query = User.query.all()
update_form.roles.query = Role.query.all()
if request.method == 'POST':
if create_form.data['create'] and create_form.validate():
form = create_form
user = User.query.filter_by(user_id=form.create_id.data).first()
role = form.roles.data
if user == None:
user = User(
user_id=form.create_id.data,
user_name=form.create_name.data)
user.password = '******'
user.role = role
db.session.add(user)
db.session.commit()
status = u'success'
message = u'成功添加登录用户'
session['message']=messages(status, message)
return redirect(url_for('admin'))
elif delete_form.data['delete'] and delete_form.validate():
form = delete_form
user = form.users.data
db.session.delete(user)
db.session.commit()
status = u'warning'
message = u'成功删除登录用户'
session['message']=messages(status, message)
return redirect(url_for('admin'))
elif update_form.data['update'] and update_form.validate():
form = update_form
user = form.users.data
role = form.roles.data
name = form.update_name.data
password = form.update_passwd.data
user.role = role
if password != '':
user.password = password
elif name != '':
user.user_name = name
db.session.add(user)
db.session.commit()
status = u'info'
message = u'成功修改登录用户信息'
session['message']=messages(status, message)
return redirect(url_for('admin'))
return render_template('admin-user.html', create_form=create_form, delete_form=delete_form, update_form=update_form)
