def edit(id=None):
# редактирование пользователя
user = User.query.get(id)
form = AddUserForm()
if form.validate_on_submit():
user.first_name = form.first_name.data
user.last_name = form.last_name.data
user.date_of_birth = form.date_of_birth.data
user.address = form.address.data
print user.first_name
db.session.commit()
flash('Успешно изменен!')
return redirect(url_for('index'))
else:
if user:
form.first_name.data = user.first_name
form.last_name.data = user.last_name
form.date_of_birth.data = user.date_of_birth
form.address.data = user.address
if user:
return render_template('edit.html', user=user, form=form)
else:
return render_template('edit.html')
def register_user():
"""Show form to register/create user."""
form = AddUserForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
email = form.email.data
first_name = form.first_name.data
last_name = form.last_name.data
flash(f"{username} register!")
user = User.register(username=username,
password=password,
email=email,
first_name=first_name,
last_name=last_name)
db.session.add(user)
db.session.commit()
return redirect(f'/users/{user.username}')
else:
return render_template("add_user_form.html", form=form)
def register_view():
if 'user' in session:
flash(f"{session['user']}, you are already logged in.", "warning")
return redirect(url_for('home_view'))
form = AddUserForm()
if form.validate_on_submit():
data = {
field: value for field, value in form.data.items()
if field in User.__table__.columns.keys()
}
try:
new_user = User.register(**data)
db.session.add(new_user)
db.session.commit()
except exc.IntegrityError:
flash("Username/email already exist!", "danger")
return render_template('register.html', form=form,
submit_button="Register")
# log user in
session['user'] = new_user.username
return redirect(url_for('user_detail_view', username=new_user.username))
return render_template('register.html', form=form, submit_button="Register")
def registration():
"""Register user."""
form = AddUserForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
email = form.email.data
first_name = form.first_name.data
last_name = form.last_name.data
user = User.register(username, password, email, first_name, last_name)
try:
db.session.add(user)
db.session.commit()
except IntegrityError:
# in case a unique constraint was violated
form.username.errors = ["Username already exists."]
return render_template("register.html", form=form)
# log in the user to the session
session["user_id"] = user.username
# redirect to secret page for newly-registered user
return redirect(f"/users/{user.username}")
else:
return render_template("register.html", form=form)
def users():
addUser = AddUserForm()
if addUser.validate_on_submit():
login_db.add([addUser.username.data, addUser.password.data, addUser.type.data])
return redirect(url_for("users"))
return render_template("user_management.html", subheading="User Management", addUser=addUser, table=user_table(),
accounts=login_db.read(), page="user-management")
def show_registration_form():
# create new user
form = AddUserForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
email = form.email.data
first_name = form.first_name.data
last_name = form.last_name.data
new_user = User.register(username=username,
password=password,
email=email,
first_name=first_name,
last_name=last_name)
db.session.add(new_user)
db.session.commit()
session['username'] = username
flash('New user added!')
return redirect(f'/users/{username}')
else:
return render_template('register.html', form=form)
def user_register():
"""Register new user."""
form = AddUserForm()
if form.validate_on_submit():
try:
new_user = User.signup(
username=form.username.data,
password=form.password.data,
email=form.email.data,
)
db.session.commit()
except IntegrityError:
db.session.rollback()
if User.query.filter_by(username=form.username.data).first():
form.username.errors = ['Username taken']
if User.query.filter_by(email=form.email.data).first():
form.email.errors = ['Email taken']
return render_template('signup.html', form=form)
db.session.commit()
session[CURR_USER_KEY] = new_user.id
return redirect('/')
return render_template('signup.html', form=form)
def signup(): # create def signup():
"""Handle user signup.
Create new user and add user to database. Redirect to home page."""
form = AddUserForm()
if form.validate_on_submit():
try:
user = User.signup(
username=form.username.data,
password=form.password.data,
first_name=form.first_name.data,
last_name=form.last_name.data,
email=form.email.data,
)
db.session.commit()
do_login(user)
flash(f"Hello, {user.first_name}!", "success")
session['user_id'] = user.id
session['username'] = user.username
return redirect('/')
except IntegrityError:
flash("Username already taken", 'danger')
return redirect('/signup')
else:
return render_template('users/signup.html', form=form)
def add_user():
if g.user.is_authenticated() and g.user.is_admin():
form = AddUserForm()
if request.method == 'POST' and form.validate_on_submit():
user_email = form.user_email.data
registered_user = User.query.filter_by(
user_email=user_email).first()
if registered_user is not None:
flash("The email already exists", "error")
return redirect(url_for('add_user'))
password = '******' if form.password.data == '' else form.password.data
user = User(form.name.data, form.user_email.data, password,
form.role.data)
db.session.add(user)
db.session.commit()
flash("Added User : "******" with Email : " +
form.user_email.data + ", password : "******", role : " + form.role.data)
if form.add_user.data:
return redirect(url_for('index'))
else:
return redirect(url_for('test'))
return render_template("addUser.html", title="Add User", form=form)
flash("You are not authorised to view the page")
return redirect(url_for('index'))
def signup():
"""Handle user signup.
Create new user and add to DB. Redirect to home page.
If form not valid, present form.
If the there already is a user with that username: flash message
and re-present form.
"""
form = AddUserForm()
if form.validate_on_submit():
try:
user = User.signup(
username=form.username.data,
password=form.password.data,
email=form.email.data,
)
db.session.commit()
except IntegrityError:
flash("Username already taken", 'danger')
return render_template('signup.html', form=form)
do_login(user)
flash("Welcome to iSports! Start adding items to follow below.",
'success')
return redirect(f"/user")
else:
return render_template('signup.html', form=form)
def show_new_user_registration_form():
"""Attempts to create a new user based on form submission."""
if CURR_USER_ID in session:
flash('You have been logged out', 'info')
log_out_procedures()
form = AddUserForm()
if form.validate_on_submit():
name = form.name.data
email_address = form.email_address.data
password = form.password.data
new_user = User.register(name, email_address, password)
# send_confirm_email_link(email_address)
# flash('Welcome!', 'success')
flash(f" Welcome, {new_user.name}! You have successfully registered for an account. Please log in to confirm your password.'", 'success')
# session[CURR_USER_ID] = new_user.id
# g.user = new_user
# next_url = request.form.get("next")
# if next_url:
# return redirect(next_url)
# else:
# return redirect('/')
return redirect('/login')
else:
return render_template('register.html', form=form)
def add_user():
form = AddUserForm()
if form.validate_on_submit():
try:
user = User(form.name.data, form.password.data, form.email.data)
db.session.add(user)
db.session.commit()
return redirect(url_for('view_frontend'))
except IntegrityError:
form.email.errors.append('Email address is not unique')
return render_template('user_add.html', form=form)
def add_user():
form = AddUserForm()
if form.validate_on_submit():
try:
user = User(form.name.data, form.password.data, form.email.data)
db.session.add(user)
db.session.commit()
return redirect(url_for('view_frontend'))
except IntegrityError:
form.email.errors.append('Email address is not unique')
return render_template('user_add.html', form=form)
def addUser():
form = AddUserForm()
if form.validate_on_submit():
new = Doctor(username=form.username.data,
email=form.email.data,
password_hash=form.password.data)
db.session.add(new)
db.session.commit()
flash('New {} added!!'.format(new.username))
return redirect(url_for('dashboard'))
return render_template('addUser.html', form=form)
def add_user():
if not current_user.is_administrator:
abort(403)
form = AddUserForm()
if form.validate_on_submit():
user = Users(username=form.username.data,
password=form.password.data,
is_administrator=form.is_administrator.data)
db.session.add(user)
flash('添加成功')
return redirect(url_for('main.index'))
return render_template('adduser.html', form=form)
def team_management():
form = AddUserForm()
allTeamMembers = teamMembers()
allTeamMembers = allTeamMembers.get("members")
if form.validate_on_submit():
member_email = form.member_email.data
member_given_name = form.member_given_name.data
member_surname = form.member_surname.data
member_external_id = form.member_external_id.data
newguy = addMember(member_email, member_given_name, member_surname, member_external_id)
return redirect('/')
return render_template('main/team_management.html', form=form, teamMembers=allTeamMembers)
def add():
form = AddUserForm()
if form.validate_on_submit():
user = User(first_name=form.first_name.data,
last_name=form.last_name.data,
date_of_birth=form.date_of_birth.data,
address=form.address.data)
db.session.add(user)
db.session.commit()
flash('Успешно добавлен!')
return redirect(url_for('index'))
return render_template('add.html', form=form)
def addUser():
form = AddUserForm()
customer_choices = [(a.id, a.name) for a in Customer.query.filter_by(customer_type=CUSTOMER_TYPES['TYPE_CUSTOMER']).all()]
customer_choices = [(0, '')] + customer_choices
maker_choices = [(a.id, a.name) for a in Maker.query.all()]
maker_choices = [(0, '')] + maker_choices
form.customer.choices = customer_choices
form.maker.choices = maker_choices
if form.validate_on_submit():
if len(User.query.filter_by(nickname=form.nickname.data).all()) > 0:
flash(gettext("Selected username already exists!"))
return redirect(url_for('users'))
new_email = form.email.data
check_mail = User.query.filter_by(email=new_email).all()
#user mail already exists
if len(check_mail) > 0:
flash(gettext('Selected email is already in use!'))
return redirect(url_for('users'))
user = User()
user.nickname = form.nickname.data
user.password = generate_password_hash(form.password.data)
user.email = form.email.data
user.role = form.role.data
if int(form.role.data) == USER_ROLES['ROLE_CUSTOMER']:
if form.customer.data and form.customer.data != '' and form.customer.data != 0:
user.customer_id = form.customer.data
else:
user.customer_id = None
elif int(form.role.data) == USER_ROLES['ROLE_MAKER']:
if form.maker.data and form.maker.data != '' and form.maker.data != 0:
user.maker_id = form.maker.data
else:
user.maker_id = None
else:
user.customer_id = None
user.maker_id = None
user.language = form.language.data
db.session.add(user)
db.session.commit()
flash(gettext("New user successfully added."))
return redirect(url_for("users"))
return render_template('settings/addUser.html',
title=gettext("Add New User"),
USER_ROLES=USER_ROLES,
form=form)
def new_user():
form = AddUserForm(request.form, csrf_enabled=False)
if form.validate_on_submit():
new_user = Users(form.firstname.data, form.lastname.data, form.username.data, form.password.data, session['user_id'], 0)
try:
db.session.add(new_user)
db.session.commit()
flash('New user was successfully added. Thanks.')
except IntegrityError:
error = 'Oh no! That username and/or email already exist. Please try again.'
return render_template('register.html', form=form, error=error)
else:
# flash("All fields are required. Please try again.")
flash_errors(form)
return redirect(url_for('users'))
def adduser():
form = AddUserForm()
if form.validate_on_submit():
user = User.query.filter_by(username=form.user.data).first()
if user is None:
user = User()
user.username = form.user.data
user.password = form.password.data
user.role_id = form.role.data
current_app.logger.info('add user %s role %s', user.username,
user.role_id)
user.save()
return redirect(url_for('main.user'))
else:
flash('user is exist!!')
return redirect(url_for('main.user'))
return render_template('adduser.html', form=form)
def register_post():
"""registration post route that redirects to secret"""
form = AddUserForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
email = form.email.data
first_name = form.first_name.data
last_name = form.last_name.data
new_user = User.register(username, password, email, first_name,
last_name)
db.session.add(new_user)
db.session.commit()
return redirect("/login")
else:
return render_template("register.html")
def login_user():
"""Show login user form; handle logging in user."""
form = AddUserForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
user = User.authenticate(username, password)
if user:
flash(f"Welcome back {username}!", "success")
session["username"] = username
return redirect(f"/users/{username}")
else:
form.username.errors = ["Invalid Username/Password"]
return render_template("login.html", form=form)
def process_register_form():
form = AddUserForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
email = form.email.data
first_name = form.first_name.data
last_name = form.last_name.data
new_user = User.register(username, password, email, first_name,
last_name)
flash(f"Added {username}")
db.session.add(new_user)
db.session.commit()
return redirect("/")
else:
return render_template("add_user_form.html", form=form)
def signup():
"""sign up a new user, add to db, redirect home. if a user with that name already exists, flash message, show form again"""
form = AddUserForm()
if form.validate_on_submit():
try:
user = User.signup(
username = form.username.data,
password = form.password.data,
email = form.email.data,
)
db.session.commit()
except IntegrityError:
flash('sorry, this username is already taken!', 'danger')
return render_template('signup.html', form = form)
do_login(user)
return redirect('/')
else:
return render_template('signup.html', form = form)
def display_user_form():
""" Show and accept submission of user form. """
form = AddUserForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
email = form.email.data
first_name = form.first_name.data
last_name = form.last_name.data
user = User.register(username, password, email, first_name, last_name)
db.session.add(user)
db.session.commit()
session["user_username"] = user.username
return redirect(f"/users/{username}")
else:
return render_template("user_form.html", form=form)
def users():
error = None
form = AddUserForm(request.form)
role = 'admin' if form.adminorno.data else 'user'
if session['role'] == "admin":
if request.method == 'POST':
if form.validate_on_submit():
new_record = User(
datetime.datetime.utcnow(), form.name.data,
form.email.data,
bcrypt.generate_password_hash(form.password.data), role)
try:
db.session.add(new_record)
db.session.commit()
flash('Added new user!')
return redirect(url_for('users'))
except IntegrityError:
error = 'That username is already exist.'
db.session().rollback()
return redirect(url_for('users'))
return render_template('users.html', users=list_users(), form=form)
return redirect(url_for('main'))
def add_user():
if g.user.is_authenticated() and g.user.is_admin():
form = AddUserForm()
if request.method == 'POST' and form.validate_on_submit():
user_email = form.user_email.data
registered_user = User.query.filter_by(user_email = user_email).first()
if registered_user is not None:
flash("The email already exists" , "error")
return redirect(url_for('add_user'))
password = '******' if form.password.data == '' else form.password.data
user = User(form.name.data, form.user_email.data, password, form.role.data)
db.session.add(user)
db.session.commit()
flash("Added User : "******" with Email : " + form.user_email.data + ", password : "******", role : " + form.role.data)
if form.add_user.data:
return redirect(url_for('index'))
else:
return redirect(url_for('test'))
return render_template("addUser.html",
title = "Add User",
form = form)
flash("You are not authorised to view the page")
return redirect(url_for('index'))
def signup():
""" Handle user signup. """
form = AddUserForm()
if form.validate_on_submit():
try:
user = User.signup(username=form.username.data,
password=form.password.data,
image_url=form.image_url.data
or User.image_url.default.arg)
db.session.commit()
except IntegrityError:
flash("Username already taken!", 'text-danger text-center')
return render_template('signup.html', form=form)
login_user(user)
return redirect(f"/user/{user.id}")
else:
return render_template('signup.html', form=form)
def add_user():
form = AddUserForm()
if request.method == 'GET':
return render_template('add_user.html', form=form)
else:
if form.validate_on_submit():
username = request.form['username']
first_name = request.form['first_name']
last_name = request.form['last_name']
prog_lang = request.form['prog_lang']
experience_yr = request.form['experience_yr']
age = request.form['age']
hw1_hrs = request.form['hw1_hrs']
new_user = User(username=username,
first_name=first_name,
last_name=last_name,
prog_lang=prog_lang,
experience_yr=experience_yr,
age=age,
hw1_hrs=hw1_hrs)
db.session.add(new_user)
db.session.commit()
return redirect(url_for('index'))
def register_user():
"""Show register user form; handle adding user."""
form = AddUserForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
email = form.email.data
first_name = form.first_name.data
last_name = form.last_name.data
new_user = User.register(username, password, email, first_name,
last_name)
db.session.add(new_user)
db.session.commit()
session["username"] = username
flash(f"{username} is now registered!", "success")
return redirect(f"/users/{username}")
else:
return render_template("register.html", form=form)
def signup():
"""Sign up view."""
if g.user:
return redirect(url_for('.user_detail', username=g.user.username))
form = AddUserForm()
if form.validate_on_submit():
password = form.password.data
# Collect relevant form data items to a dictionary.
relevant_data = {
k: v
for k, v in form.data.items() if k in User.set_get()
}
try:
new_user = User.register(password=password, **relevant_data)
session['user_id'] = new_user.id
session.permanent = True
flash(f"Welcome {new_user.username}!", "success")
return H.next_page_logic(request)
except Exception as e:
db.session.rollback()
flash("Error Creating User", 'danger')
H.error_logging(e)
if request.method == 'POST':
flash("Please fix all form errors.", "warning")
# Create URL for login button that passes all URL data.
login_url = request.full_path.replace("/signup", "/login")
return render_template('user_views/signup.html',
form=form,
login_url=login_url)
def admin():
form = AddUserForm(prefix="form")
formUpdate = UpdateUserForm(prefix="formUpdate", idUser='******')
u = Users.query.order_by(Users.id).all()
formDelete = DeleteUserForm(prefix="formDelete")
if form.validate_on_submit() and form.submit.data:
a.createUser(session['author_id'], session['session_id'], form.login.data, form.email.data, form.password.data, form.admin.data)
elif request.method == 'POST' and form.validate() == False and not formUpdate.submit.data and not formDelete.submit.data:
flash("Error during the user creation!")
if formUpdate.validate_on_submit() and formUpdate.submit.data:
if formUpdate.idUser.data == '0':
formUpdate.idUser.data = session['author_id']
a.updatePassword(session['author_id'], session['session_id'], int(formUpdate.idUser.data), formUpdate.oldPassword.data, formUpdate.password.data)
if formDelete.validate_on_submit() and formDelete.submit.data:
a.getUserByName(session['author_id'], session['session_id'], formDelete.name.data)
time.sleep(1)
u = Users.query.filter_by(name = formDelete.name.data).all()
if not u:
flash("User not found or the server don't send the user information!")
else:
u = u[0]
print formDelete.password.data
a.delUser(session['author_id'], session['session_id'], u.id, formDelete.password.data)
return render_template('admin.html', form=form, formUpdate=formUpdate,u = u, formDelete=formDelete)
def register_user():
""" GET - Shows a form that registers a user.
POST - Creates new user in database
"""
form = AddUserForm()
if form.validate_on_submit():
if User.query.get(form.username.data):
flash("Username already exists. Please pick a new username.")
return render_template("register.html", form=form)
new_user = User.register({
"username": form.username.data,
"password": form.password.data,
"first_name": form.first_name.data,
"last_name": form.last_name.data,
"email": form.email.data,
})
db.session.add(new_user)
db.session.commit()
session["user_id"] = form.username.data
return redirect(f"/users/{new_user.username}")
else:
return render_template("register.html", form=form)
def adduser():
form = AddUserForm()
if form.validate_on_submit():
client = MongoClient(MONGODB_URI)
db = client.get_default_database()
users = db.users
newUser = {
"username": form.username.data,
"password": form.password.data
}
users.insert(newUser)
users = db.users
for user in users.find():
if user['username'] == form.username.data and user['password'] == form.password.data:
u = User.get(str(user['_id']))
login_user(u, remember=True)
return redirect('index')
flash("Logged in successfully.")
return redirect("/index")
return redirect('/login')
return render_template('adduser.html',
title = 'Add New User',
form = form)
def add_user_handler():
add_user_form = AddUserForm()
if add_user_form.validate_on_submit():
user_id = add_user_form.user_id.data
password = add_user_form.password.data
user_name = add_user_form.name.data
sex = add_user_form.sex.data
photo = add_user_form.photo.data
duty = add_user_form.duty.data
department = add_user_form.department.data
telephone = add_user_form.telephone.data
admin = add_user_form.admin.data
remarks = add_user_form.remarks.data
user = db.add_user(user_id, password, user_name, sex,
photo.read(), duty, department, telephone, admin, remarks)
if user is not None:
flash("添加用户成功!")
return redirect(url_for('dashboard_handler'))
else:
flash("添加用户失败!")
return render_template('add_user.html', form=add_user_form)
def register_page():
form = AddUserForm()
if 'user_id' in session:
username = session['user_id']
flash("You are already registered and signed in.", "warning")
return redirect(f"/users/{username}")
if form.validate_on_submit():
new_user = User.register_user(form=form)
db.session.add(new_user)
try:
db.session.commit()
except IntegrityError:
form.username.errors.append('Username already exists. Choose another username.')
return render_template('register.html')
session['user_id'] = new_user.username
flash("Account created successfully!", "success")
return redirect(f'/users/{new_user.username}')
return render_template('register.html', form=form)
def add_user(username):
form = AddUserForm()
#session_user = User.query.filter_by(username=session['username']).first()
session_user = User.query.filter(User.username == username).first()
if request.method == 'GET':
return render_template('add_user.html',
form=form,
title='Add User Information',
session_username=session_user.username)
else:
#session_user.first_name= 'first_name'
#session_user.last_name= 'last_name'
#db.session.commit()
#return redirect('/index')
#session_user = User.query.filter_by(username=session['username']).first()
if form.validate_on_submit():
first_name = request.form['first_name']
last_name = request.form['last_name']
state = request.form['state']
zipcode = request.form['zipcode']
age = request.form['age']
gender = request.form['gender']
drug_type = request.form['drug_type']
use_duration = request.form['use_duration']
med_insurer = request.form['med_insurer']
year_signed_up = request.form['year_signed_up']
years_from_first_litigation = request.form[
'years_from_first_litigation']
copay_or_coinsurance = request.form['copay_or_coinsurance']
personal_spending_per_year = request.form[
'personal_spending_per_year']
income = request.form['income']
if copay_or_coinsurance == 'copay':
total_spending = float(personal_spending_per_year) / .11
else:
total_spending = float(personal_spending_per_year) / .15
#return total_spending
session_user.first_name = first_name
session_user.last_name = last_name
session_user.state = state
session_user.zipcode = zipcode
session_user.age = age
session_user.gender = gender
session_user.drug_type = drug_type
session_user.use_duration = use_duration
session_user.med_insurer = med_insurer
session_user.year_signed_up = year_signed_up
session_user.years_from_first_litigation = years_from_first_litigation
session_user.copay_or_coinsurance = copay_or_coinsurance
session_user.personal_spending_per_year = personal_spending_per_year
session_user.total_spending = total_spending
session_user.income = income
if gender == "M":
gender_1hot = np.array([0, 1])
else:
gender_1hot = np.array([1, 0])
state_1hot = onehotState(state)
years_int = int(years_from_first_litigation)
years_1hot = onehotCategorical(years_int, 20)
print(state_1hot)
#test_pred = np.array([0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 36, 56361, 305.5437079])
#test_pred = np.hstack([gender_1hot, [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], years_1hot, [age, income, total_spending]])
test_pred = np.hstack([
gender_1hot, state_1hot, years_1hot,
[age, income, total_spending]
])
#prediction = model.predict(test_pred.reshape(1, -1))*473.5694972
prediction = model.predict(test_pred.reshape(1,
-1)) * total_spending
session_user.est_settle = float(np.squeeze(prediction.round(2)))
#settle = "$" + str(np.squeeze(prediction.round(2)))
db.session.commit()
#return redirect('/index')
return redirect(url_for('profile', username=session_user.username))
