def register(request):
userInfo = user()
if request.method == 'POST':
data = json.loads(request.body)
new_userName = data['username']
new_passWord = data['pwd']
new_sex = data['sex']
new_userAge = data['userage']
new_userTel = data['usertel']
new_email = data['email']
new_authorization = data['authorization']
try:
result = user.objects.get(username=new_userName)
res = {'data':new_userName,'message':'该用户名已存在，不得重复！','err_code':'201'}
return HttpResponse(json.dumps(res),content_type='application/json')
except:
userInfo.username = new_userName
userInfo.password = new_passWord
userInfo.sex = new_sex
userInfo.userage = new_userAge
userInfo.usertel = new_userTel
userInfo.email = new_email
userInfo.authorization = new_authorization
userInfo.save()
res = {'data': new_userName, 'message': '账号%s 注册成功'%new_userName,'err_code':'101'}
return HttpResponse(json.dumps(res),content_type='application/json')
def pharmacy_register():
form = RegistrationPharmForm()
if form.validate_on_submit():
#calculate id here
highest_id = db.session.query(pharm).order_by(
pharm.pc_id.desc()).first()
highest_id_num = int(highest_id.pc_id[2:])
id = "PC" + str(highest_id_num + 1)
phrm = pharm(pc_id=id,
pc_name=form.name.data,
pc_address=form.address.data,
pc_city=form.city.data,
pc_state=form.state.data,
pc_zipcode=form.zipcode.data)
usr = user(user_id=id,
username=form.username.data,
user_type="pharmacy")
usr.set_password(form.password.data)
db.session.add(phrm)
db.session.add(usr)
db.session.commit()
flash("You've successfully registered as a pharmacy")
return render_template('pharmacy_register.html',
title='Sign In',
form=form)
def post(self):
# 得到参数列表
args = parse.parse_args()
# 得到参数列表中的昵称
nickname = args.get('nickname')
# 判断用户是否已经存在
if models.user.query.get(nickname):
abort(400, message="{} existed".format(nickname))
# 创建用户
user = models.user()
# 将传入参数加入到user中
user.nickname = args.nickname
user.name = args.name
user.sex = args.sex
user.country = args.country
user.province = args.province
user.city = args.city
user.id = args.id
user.phone = args.phone
user.credit = 0
user.priority = 0
# 将user存入数据库
try:
db.session.add(user)
db.session.commit()
return {"message": "success"}
except Exception as e:
db.session.rollback()
abort(500)
def declaration(request):
usercondition = healthyCondition()
userInfo = user()
if request.method == 'POST':
data = json.loads(request.body)
applyUsername = data['username']
applyCondition = data['condition']
applyTemperature = data['temperature']
applyPosition = data['position']
try:
result = user.objects.get(username=applyUsername)
applyUserid = result.userid
usercondition.userid_id = applyUserid
usercondition.condition = applyCondition
usercondition.temperature = applyTemperature
usercondition.position = applyPosition
usercondition.save()
result2 = healthyCondition.objects.all().aggregate(Max('id'))
idddd = result2['id__max']
result3 = healthyCondition.objects.get(id=idddd)
healthyCondition.objects.filter(Q(date=result3.date)&~Q(pk=idddd)).delete()
res = {'data':applyUsername,'message':'健康信息申报成功','err_code':'150'}
return HttpResponse(json.dumps(res),content_type='application/json')
except:
res = {'data':applyUsername,'message':'该用户不存在，请先注册','err_code':'400'}
return HttpResponse(json.dumps(res), content_type='application/json')
def testDB():
from app.models import account, user, accountUser
import uuid
if not account.query.filter_by(name='sysAdmin').first():
acc = account(name='sysAdmin')
db.session.add(acc)
db.session.commit()
print('account {} added'.format(str(acc.name)))
if not user.query.filter_by(email='*****@*****.**').first():
usr = user(email='*****@*****.**')
db.session.add(usr)
db.session.commit()
print('user {} added'.format(str(usr.email)))
usr = user.query.filter_by(email='*****@*****.**').first()
acc = account.query.filter_by(name='sysAdmin').first()
if not accountUser.query.filter_by(accountID=acc.id, userID=usr.id).first():
accUsr = accountUser(accountID=acc.id,
userID=usr.id,
isSiteAdmin=True,
isAdmin=True,
isWriter=True,
uuid=str(uuid.uuid4()),
password='******',
loginAttempts = 0)
db.session.add(accUsr)
db.session.commit()
print('accountUser added')
def signout(sessionID):
user_class = user("", "", sessionID)
ret = user_class.signoutUser()
user_class.conn.close()
if ret < 0:
return create_error("internal error"), 500
return create_result("signout successful"), 200
def user_info(sessionID):
user_class = user("", "", sessionID)
ret, keys = user_class.getUserInfo()
user_class.conn.close()
if ret < 0:
return ((create_error("you must be logged in"), 401) if ret == -1 else (create_error("internal error"), 500))
return create_result(loads(keys)), 200
def registering():
thistime = int(time.time())
req = request.get_json()
account = req["account"]
password = req["password"]
users = models.user.query.all()
maxid = users[len(users) - 1].id
maxid = maxid + 1
for key in users:
if key.account == account:
return json.dumps({"msg": "accountexisting"})
user = models.user()
user.id = maxid
user.account = account
user.password = password
user.grade = ''
user.shortIntro = ''
user.head = ''
user.follow = 0
user.follower = 0
user.major = ''
user.time = thistime
user.sex = ''
user.likeCount = 0
user.starCount = 0
db.session.add(user)
db.session.commit()
return json.dumps({"msg": "registered"})
def edit_prof(request):
if request.method == "GET":
return render(request,'edit.html',{})
else:
username=request.session['username'];
obj=user.objects.get(name=username);
obj.delete();
Name=request.POST['name'];
Password=request.POST['password'];
Email=request.POST['email'];
Usertype=request.POST['usertype'];
obj1=models.user(name=Name,password=Password,email=Email,usertype=Usertype)
obj1.save()
context={}
try:
request.session['authority']=request.GET['authority']
except:
pass
context['authority']=request.session['authority']
context['name']=obj1.name;
context['usertype']=obj1.usertype;
context['email']=obj1.email;
if context['authority']=="Student":
return render(request,'student_profile.html',context);
else:
return render(request,'lib_profile.html',context);
def signup(request):
# send verification.(records in user table and verification table)
print("SIGNUP")
if request.POST is not None and request.POST[
'email'] is not None and request.POST['password'] is not None:
mail = request.POST['email']
password = request.POST['password']
try:
tempuser = user.objects.get(user_email=mail)
except user.DoesNotExist:
tempuser = None
if tempuser is not None:
messages.error(request, message="User already exists!")
return 'index'
hashed = hashlib.md5(password.encode())
password = hashed.hexdigest()
tempuser = user(user_email=mail, user_password=password)
tempuser.save()
otp = random.randint(100000, 999999)
verificationRecord = verification(user_id=tempuser.id, otp=otp)
verificationRecord.save()
send_email(
mail, "OTP", 'Here is your one time verification code:<h4>' +
str(otp) + "</h4>")
return 'verify'
return 'index'
def post(self, wechat):
# 判断用户是否存在
if models.user.query.get(wechat):
abort(400, message="{} existed".format(wechat))
# 创建用户
user = models.user()
user.wechat = wechat
args = parse.parse_args()
# 将传入参数加入到user中
user.name = args.name
user.phone = args.phone
user.grade = args.grade
user.college = args.college
user.major = args.major
user.classnum = args.classnum
user.studentnum = args.studentnum
user.plan_id = args.plan_id
# 将user存入数据库
try:
db.session.add(user)
db.session.commit()
return {"message": True}
except Exception as e:
db.session.rollback()
abort(500)
def index():
form = LoginForm()
if form.validate_on_submit():
email = user(email=form.email.data)
db.session.add(email)
db.session.commit()
flash('Subscribed')
return redirect(url_for('index'))
return render_template('index.html', form=form)
def register():
name = request.form.get('name')
phone = request.form.get('phone')
check1 = user.query.filter_by(phone=phone).first()
print(check1)
if check1:
return '已经注册过了'
else:
u = user(name=name, phone=phone)
db.session.add(u)
db.session.commit()
return '注册成功'
def userRegister():
global searchStatus
errStatus = False
err = ""
if request.method == "POST":
username = request.form.get("username")
password = request.form.get("password")
passwordConfirm = request.form.get("passwordConfirm")
phone = request.form.get("phone")
# 校验
validators = [
userRegisterForm.usernameValidator(username),
userRegisterForm.passwordValidator(password, passwordConfirm),
userRegisterForm.phoneValidator(phone)
]
# 去除重复值
validators = list(set(validators))
for val in validators:
if val is not None:
flash(val)
break
# 是否无错误信息
if len(validators) == 1 and validators[0] is None:
users = user(username, password, phone) # 创建ORM结构
try:
username = user.query.filter_by(
username=username).one() # 查询用户名
if username is not None:
searchStatus = False
except BaseException:
searchStatus = True
if searchStatus:
try:
db.session.add(users)
db.session.commit() # 提交到数据库进行持久化
appendStatus = True
except BaseException:
db.session.rollback()
appendStatus = False
if searchStatus and appendStatus:
err = "注册成功,即将跳转到登录界面"
errStatus = True
elif not searchStatus:
err = "该用户名已被使用"
else:
err = "注册失败"
flash(err)
# 重定向到登录页面
if errStatus:
return redirect(url_for("userBlueprint.userLogin"))
return render_template("./user/register.html", form=userRegisterForm())
def registerView():
if request.method == "POST":
username = request.form.get("username")
password = request.form.get("password")
password2 = request.form.get("password2")
error = registerValidate(username, password, password2).results()
if not error:
insertUser = user(username, password)
db.session.add(insertUser)
db.session.commit()
session["username"] = username
return redirect(url_for("indexBlueprint.index"))
return render_template("register.html", error=error)
return render_template("register.html")
def create_user():
name = None
form = CreateUser()
if form.validate():
name = form.user.data
form.user.data = ''
user = models.user(name=name)
db.session.add(user)
db.session.commit()
return redirect(url_for('home.index'))
return render_template('create_user.html', form=form)
def register():
form = RegisterForm()
if form.validate_on_submit():
if user.query.filter_by(username=form.username.data).first():
flash("邮箱已被注册")
return redirect(url_for("auth.register"))
myuser = user(email=form.email.data,
username=form.username.data,
password=form.password.data)
db.session.add(myuser)
db.session.commit()
sendmail("demo小站:新用户激活", myuser.email, "email/email.html",
myuser.generator_confirmed_token())
flash("已向您的注册邮箱发送激活邮件，请前往激活(有效期1小时)")
return redirect(url_for("main.index"))
return render_template("auth/register.html", form=form)
def login(request):
book_id = request.POST.get('x1')
full_name = request.POST.get('u1')
email_id = request.POST.get('u2')
phone_no = request.POST.get('u3')
password = request.POST.get('u4')
address = request.POST.get('u5')
try:
user_data = user(full_name=full_name,
email_id=email_id,
phone_no=phone_no,
password=password,
address=address)
user_data.save()
return render(request, 'login.html', {'book_id': book_id})
except IntegrityError:
return render(request, 'login.html', {'book_id': book_id})
def add(request):
compsny_all = company.objects.all()
department_all = department.objects.all()
if request.method == 'GET':
return render(request, 'add.html',{"compsny_all":compsny_all,"department_all":department_all})
else:
username = request.POST.get('username')
number = request.POST.get('number')
gender = request.POST.get('gender')
dep_id = request.POST.get('dep_id')
compsny_id = request.POST.get('company_id')
d = department.objects.filter(id = dep_id).first()
c = company.objects.filter(id = compsny_id).first()
user_e = user(username=username, number=number, gender=gender, dep_id=d, company_id=c)
user_e.save()
print(compsny_id)
return redirect('show')
def registrationpage():
if current_user.is_authenticated:
return redirect(url_for('homepage.homepage'))
form1 = LoginForm()
form2 = RegistrationForm()
if form2.validate_on_submit():
hashed_password = bcrypt.generate_password_hash(
form2.password.data).decode('utf-8')
print(form2.name.data)
type(form2.name.data)
u = user(form2.name.data, form2.email.data, form2.address.data,
form2.password.data)
db.session.add(u)
db.session.commit()
print(u)
return redirect(url_for('loginpage.loginpage'))
return render_template('register.html', form1=form1, form2=form2)
def testdb(request):
test1 = People(id=101,
name='吴进军',
sex='男',
year=6,
job='老板',
department='督导层')
test2 = salary(id=101,
name='lqy',
salary=100,
Bonus=0,
fakuan=50,
Overtime_pay=200)
test3 = user(user='******', password='******')
test1.save()
test2.save()
test3.save()
return HttpResponse("<p>数据添加成功!</p>")
def register(username, password, keys):
user_class = user(username, password, "")
ret = user_class.registerUser(keys)
if (ret < 0):
user_class.conn.close()
return ((create_error("account already exists"), 403) if ret == -1 else (create_error("internal error"), 500))
user_class.sessionID = token_urlsafe(32)
ret = user_class.submitSessionID()
max_tries = 10
try_count = 0
while ret == -1 and try_count < max_tries:
user_class.sessionID = token_urlsafe(32)
try_count += 1
ret = user_class.submitSessionID()
user_class.conn.close()
if try_count == max_tries or ret < 0:
return create_error("internal error"), 500
return dict(create_result("account created"), **{ "sessionID": user_class.sessionID }), 200
def signin(username, password):
user_class = user(username, password, "")
ret = user_class.checkInvalidCredentials()
if (ret < 0):
user_class.conn.close()
return ((create_error("login or password does not match"), 401) if ret == -1 else (create_error("internal error"), 500))
user_class.sessionID = token_urlsafe(32)
ret = user_class.submitSessionID()
max_tries = 10
try_count = 0
while ret == -1 and try_count < max_tries:
user_class.sessionID = token_urlsafe(32)
try_count += 1
ret = user_class.submitSessionID()
user_class.conn.close()
if try_count == max_tries or ret < 0:
return create_error("internal error"), 500
return dict(create_result("signin successful"), **{ "sessionID": user_class.sessionID }), 200
def register():
if current_user.is_authenticated:
return redirect(url_for('logged_index'))
form = SignUpForm()
if form.validate_on_submit():
usr = user(firstName=form.firstName.data,
lastName=form.lastName.data,
regNo=form.regNo.data)
usr.set_password(form.password.data)
database.session.add(usr)
database.session.commit()
return redirect(
url_for('picture_upload',
usr=pr.encoding_int(usr.id),
token=pr.token_key))
return render_template('register.html', form=form)
def plant_register():
form = RegistrationPlantForm()
if form.validate_on_submit():
#calculate id here
highest_id = db.session.query(pharm_plant).order_by(
pharm_plant.pp_id.desc()).first()
highest_id_num = int(highest_id.pp_id[2:])
id = "PP" + str(highest_id_num + 1)
plant = pharm_plant(pp_id=id,
pp_name=form.name.data,
pp_address=form.address.data,
pp_city=form.city.data,
pp_state=form.state.data,
pp_zipcode=form.zipcode.data)
usr = user(user_id=id, username=form.username.data, user_type="plant")
usr.set_password(form.password.data)
db.session.add(plant)
db.session.add(usr)
db.session.commit()
flash("You've successfully registered as a pharmacy plant")
return redirect('/admin')
return render_template('plant_register.html', title='Sign In', form=form)
def addUser(request):
reponse = {}
update = False
try:
user_id = request.POST.get('user_id')
if int(user_id) > -1:
update = True
user_dto = user.objects.filter(user_id=user_id).first()
user_role.objects.filter(user_id=user_id).delete()
else:
user_dto = user()
user_dto.account = request.POST.get('account')
if update == False:
user_dto.password = make_password(request.POST.get('password'))
user_dto.user_name = request.POST.get('user_name')
user_dto.phone = request.POST.get('phone', default="")
user_dto.email = request.POST.get('email', default="")
user_dto.save()
role_ids = request.POST.get('role', default=[]).split(',')
for i in role_ids:
role_dto = role.objects.filter(role_id=i).first()
user_role_dto = user_role()
user_role_dto.save()
user_role_dto.user_id.add(user_dto)
user_role_dto.role_id.add(role_dto)
user_role_dto.save()
except:
reponse['status'] = 300
reponse['message'] = "添加失败"
else:
reponse['status'] = 0
if update:
message = "修改成功"
else:
message = "添加成功"
reponse['message'] = message
return HttpResponse(
json.dumps(reponse, ensure_ascii=False, cls=CJsonEncoder))
def register(request):
if request.method=="GET":
return render(request,'register.html');
elif request.method=="POST":
Name=request.POST['name'];
if models.user.objects.filter(name=Name).exists():
return render(request,'register.html');
else:
Name=request.POST['name'];
Password=request.POST['password'];
Email=request.POST['email'];
try:
Usertype=request.POST['usertype']
except:
context={}
context['usertypeerror']="Please select one option..!!"
return render(request,'register.html',context);
obj1=models.user(name=Name,password=Password,email=Email,usertype=Usertype)
obj1.save()
return redirect("/login/")
def doctor_register():
form = RegistrationDocForm()
if form.validate_on_submit():
#calculate id here
highest_id = db.session.query(doctor).order_by(
doctor.doc_id.desc()).first()
highest_id_num = int(highest_id.doc_id[2:])
id = "DT" + str(highest_id_num + 1)
doc = doctor(doc_id=id,
doc_first_name=form.first_name.data,
doc_last_name=form.last_name.data,
doc_speciality=form.specialty.data,
doc_address=form.address.data,
doc_city=form.city.data,
doc_state=form.state.data,
doc_zipcode=form.zipcode.data)
usr = user(user_id=id, username=form.username.data, user_type="doctor")
usr.set_password(form.password.data)
db.session.add(doc)
db.session.add(usr)
db.session.commit()
flash("You've successfully registered as a doctor")
return render_template('doctor_register.html', title='Sign In', form=form)
def patient_register():
form = RegistrationPatientForm()
if form.validate_on_submit():
#calculate id here
highest_id = db.session.query(patient).order_by(
patient.pat_id.desc()).first()
highest_id_num = int(highest_id.pat_id[1:])
id = "P" + str(highest_id_num + 1)
pat = patient(pat_id=id,
doc_id=form.doc_id.data,
pat_first_name=form.first_name.data,
pat_last_name=form.last_name.data,
pat_gender=form.gender.data,
pat_ethnicity=form.ethnicity.data,
dob=form.dob.data)
usr = user(user_id=id,
username=form.username.data,
user_type="patient")
usr.set_password(form.password.data)
db.session.add(pat)
db.session.add(usr)
db.session.commit()
flash("You've successfully registered as a patient")
return render_template('patient_register.html', title='Sign In', form=form)
def add_user(email, name, db=db):
kwargs = {}
kwargs['email'] = email
kwargs['name'] = name
db.session.add(user(**kwargs))
db.session.commit()
