def staff_profile():
firstname = request.json['firstname']
lastname = request.json['lastname']
sex = request.json['sex']
phone = request.json['phone']
role = request.json['staff']
staff = Staff(staff_id=current_user.id,
firstname=firstname,
lastname=lastname,
sex=sex,
phone_no=phone)
staff.save()
if (role == "HOD"):
department_code = request.json['department_code']
department = Department.query.filter_by(code=department_code).first()
department.staff_id = current_user.id
db.session.commit()
elif (role == "Adviser"):
department_code = request.json['department_code']
adviser = Adviser(staff_id=current_user.id,
department_code=department_code,
level=4)
adviser.save()
elif (role == "Dean"):
faculty = request.json['faculty']
faculty = Faculty.query.filter_by(name=faculty).first()
faculty.staff_id = current_user.id
db.session.commit()
data = {'message': 'success'}
return jsonify(data)
class StaffTestCase(TestCase):
def setUp(self):
Base.metadata.create_all(engine)
self.staff = Staff(name="Some Name")
self.staff.save()
def tearDown(self):
Base.metadata.drop_all(engine)
def test_class_has_required_attributes(self):
self.assertTrue(hasattr(Staff, 'id'))
self.assertTrue(hasattr(Staff, 'name'))
self.assertTrue(hasattr(Staff, '__table__'))
def test_class_attributes_mapped_to_right_db_types(self):
self.assertEqual(type(self.staff.id), int)
self.assertEqual(type(self.staff.name), str)
def test_class_has_mapped_to_right_table(self):
self.assertEqual(getattr(Staff, '__table__').name, 'staff')
def test_create(self):
current_count = Staff.all().count()
Staff.create("New Person")
self.assertEqual(Staff.all().count(), current_count + 1)
def save(self, request):
user = super(SignupStaffForm, self).save(request)
staff = Staff()
staff.account_core = CustomUser.objects.filter(
email=self.cleaned_data['email'])[0]
staff.name = self.cleaned_data['name']
staff.furigana = self.cleaned_data['furigana']
staff.position = self.cleaned_data['position']
staff.address = self.cleaned_data['address']
staff.tel = self.cleaned_data['tel']
staff.save()
return user
def post(self):
auth_t = auth_token(request)
if auth_t:
resp = User.decode_auth_token(auth_t)
u = User.query.filter_by(vid=resp).first()
req = request.get_json()
if u.super():
st = Staff.query.filter_by(vid=req.get('vid'),
team=req.get('team')).first()
if st is not None:
st.level = req.get('level')
db.session.commit()
responseObject = {
'status': 'success',
'message': 'Upgraded Staff Level'
}
return jsonify(responseObject)
else:
st = Staff(vid=req.get('vid'),
team=req.get('team'),
level=req.get('level'))
db.session.add(st)
db.session.commit()
responseObject = {
'status': 'success',
'message': 'Upgraded to Staff'
}
return jsonify(responseObject)
def register():
form = RegisterForm(
MultiDict({
**request.form.to_dict(),
**request.files.to_dict()
}))
if form.validate():
staff = Staff(number=form.number.data,
name=form.name.data,
branch=form.branch.data,
duty=form.duty.data)
img = cv2.imdecode(np.fromstring(form.img.data.read(), np.uint8),
cv2.IMREAD_COLOR)
emd = bytes(json.dumps(face_model.get_embedding(img).tolist()),
encoding='utf-8')
try:
db.session.add(staff)
db.session.commit()
except Exception:
db.session.rollback()
return make_response(jsonify({'msg': '写入数据库失败: add staff'}), 400)
sid = Staff.query.filter_by(number=form.number.data).first().id
embedding = Embedding(sid=sid, emd=emd)
try:
db.session.add(embedding)
db.session.commit()
except Exception:
db.session.rollback()
return make_response(jsonify({'msg': '写入数据库失败: add embedding'}),
400)
face_model.update_facebank()
return make_response(jsonify({'msg': '注册成功'}))
else:
return make_response(jsonify({'msg': '参数验证失败'}), 400)
def register():
if current_user.is_authenticated:
return redirect(url_for('dashboard', username=current_user.username))
form = StaffRegister()
ran = str(random.randint(100000, 900000))
#j= json.dumps(ran)
if form.validate_on_submit():
#form.emp_no.data=ran
date1 = str(form.dob.data)
pas = str(generate_password_hash(form.confirm.data))
staff = Staff(emp_no=ran,
dob=date1,
name=form.name.data,
username=form.username.data,
gender=form.gender.data,
email=form.email.data,
phone=form.telephone.data,
adress=form.address.data,
position=form.position.data,
password_harsh=pas)
#staff.set_password(form.confirm.data)
db.session.add(staff)
db.session.commit()
return redirect(url_for('login'))
return render_template('staff_register.html',
title='E-Restaurant | Register',
form=form,
ran=ran)
def engineering_add():
form = ItemForm()
if form.validate_on_submit():
data = form.data
ensure = Item.query.filter_by(item_id=data["item_id"]).count()
if ensure == 1:
flash("工程号已存在！", "err")
return redirect(url_for('main.engineering_add'))
file_url = secure_filename(form.bridge.data.filename)
if not os.path.exists(app.config["UP_DIR"]):
os.makedirs(app.config["UP_DIR"])
os.chmod(app.config["UP_DIR"], "rw")
bridge = change_filename(file_url)
form.bridge.data.save(app.config["UP_DIR"] + bridge)
item = Item(item_id=data["item_id"],
item_name=data["name"],
item_loc=data["location"],
item_manager=data["manager"],
item_starttime=data["time"],
bridge=bridge)
db.session.add(item)
db.session.commit()
staff = Staff(item_id=data["item_id"],
staff_name=data["staff_name"],
staff_phone=data["staff_rel"])
db.session.add(staff)
db.session.commit()
flash("添加工程成功!", "ok")
return redirect(url_for('main.engineering_add'))
return render_template("engineering_add.html", form=form)
def save(self, request):
user = super(SignupStaffForm, self).save(request)
staff = Staff()
staff.account_core = CustomUser.objects.filter(email=self.cleaned_data['email'])[0]
staff.first_name = self.cleaned_data['first_name']
staff.last_name = self.cleaned_data['last_name']
staff.tel = self.cleaned_data['tel']
staff.save()
return user
def test_get_people(self):
office = Office.create("office1")
staff = Staff.create("New Staff")
fellow = Fellow.create("New Fellow")
staff.add_to_room(office)
fellow.add_to_room(office)
self.assertIn(fellow, office.get_people().all())
self.assertIn(staff, office.get_people().all())
def addStaff():
form = StaffForm()
if request.method == 'POST' and form.validate_on_submit():
s = Staff(request.form['title'], request.form['fname'],
request.form['lname'], request.form['position'])
db.session.add(s)
db.session.commit()
flash('Added Staff Member successfully', 'success')
return redirect(url_for('staffMembers'))
return render_template('staff_form.html', form=form)
def populate_staff(): #This will populate the db with the staff. data = [(u'Joe', u'Bloggs',1998,10,10,u'07400232054',u'12 This Street',u'DE23 6L1',u'QQ123456A'), (u'Jane', u'Doe', 1997,11,11,u'07400232055',u'16 Another Place',u'DE23 6L2',u'QQ123456B'), (u'Ralph', u'Billings',1990,12,12,u'07400232056',u'12 Lovely Place',u'DE23 6L3',u'QQ123456D'), (u'Name', u'McNameface',1992,9,9,u'07400232057',u'36 Great Road',u'DE23 6L4',u'QQ123456C')] for d in data: staff = Staff(staff_f_name=d[0], staff_s_name=d[1],staff_dob=datetime.date(d[2],d[3],d[4]) ,staff_mobile=d[5],staff_address=d[6],staff_postcode=d[7],staff_ni=d[8]) db.session.add(staff) db.session.commit()
def staff_add_post():
password = request.form.get('add-password')
staff = Staff(staffid=request.form.get('add-staffid'),
staffname=request.form.get('add-staffname'),
phone=request.form.get('add-phone'),
idcard=request.form.get('add-idcard'),
salary=request.form.get('add-salary'),
job=request.form.get('add-job'),
age=request.form.get('add-age'),
password=request.form.get('add-password'),
gender=request.form.get('add-gender'),
role_id='4')
db.session.add(staff)
db.session.commit()
return redirect(url_for('staff.staff_message'))
def add_staff():
form = AddStaffForm()
if form.validate_on_submit():
staff = Staff(staff_id=form.staff_id.data,
name=form.name.data,
ip=form.ip.data,
mac=form.mac.data,
method=form.method.data)
try:
db.session.add(staff)
db.session.commit()
flash('Congratulations, you have added the staff!')
except exc.IntegrityError:
db.session.rollback()
flash('Could not add the staff, repeated values!')
return redirect(url_for('add_staff'))
return render_template('add.html', title='Add Staff', form=form)
def register_staff(self, model: RegisterStaffDto):
staff = Staff()
staff.department = model.department
staff.role = model.role
staff.date_of_employment = model.date_of_employment
user = User.objects.create_user(username=model.username,
email=model.email,
password=model.password)
user.first_name = model.first_name
user.last_name = model.last_name
user.email = model.email
user.save()
staff.user = user
staffs = Group.objects.get(name__exact='staffs')
user.groups.add(staffs)
staff.save()
def stfSignup():
form = StaffRegisterForm()
if form.validate_on_submit():
data = form.data
if data["password"] != data["passwordtwo"]:
flash("密码不一致")
return redirect(url_for("home.stfSignup"))
staff = Staff.query.filter_by(username=data["username"]).count()
if staff == 1:
flash("The username already exists!")
return redirect(url_for("home.stfSignup"))
staff = Staff(
username=data["username"],
password=generate_password_hash(data["password"]),
airline_name=data["airline_name"]
)
db.session.add(staff)
db.session.commit()
flash("Registration success！")
return redirect(url_for("home.login"))
return render_template("home/staffsignup.html",form=form)
def setUp(self):
Base.metadata.create_all(engine)
self.staff = Staff(name="Some Name")
self.staff.save()
def createStaff(request):
"""
add a staff
"""
if request.method=="POST":
myDict = dict(request.POST.iterlists())
userName=myDict['UserName'][0].strip()
passWord=myDict['Password'][0].strip()
passWord2=myDict['Password2'][0].strip()
realName=myDict['RealName'][0].strip()
content=myDict['Remark'][0].strip()
gender=int(myDict['Gender'][0].strip())
groupList=request.POST.getlist('Group')
calendarDispList=request.POST.getlist('setCalendarDisp')
if(userName==''):# info error
messages.error(request, '用户名不能为空')
return False,myDict
if(len(userName)>50):# info error
messages.error(request, '用户名不能大于50个字符')
return False,myDict
if(passWord==''):# info error
messages.error(request, '密码不能为空')
return False,myDict
if(realName==''):# info error
messages.error(request, '姓名不能为空')
return False,myDict
if(len(realName)>50):# info error
messages.error(request, '姓名不能大于50个字符')
return
if(passWord!=passWord2):# info error
messages.error(request, '两次密码输入不一致')
return False,myDict
if(User.objects.filter(username=userName).count()!=0):#same user
messages.error(request, '用户名已存在')
return False,myDict
if(Staff.objects.filter(name=realName).count()!=0):#same realName
messages.error(request, '姓名已存在')
return False,myDict
if(not myDict.get('Role')):
messages.error(request, '权限不能为空')
return False,myDict
roleID=myDict['Role'][0].strip()
md5Password=hashlib.md5(passWord).hexdigest()[8:-8]
user=User(username=userName,password=md5Password)
#user.set_password(user.password)#hash password
#password = make_password(passWord,'md5')
user.save()
role =Role.objects.get(id=roleID)
mS=Staff(user=user,name=realName,role=role,gender=gender,content=content)
mS.save()
if(groupList!=''):
for groupItem in groupList:
g=Group.objects.get(id=groupItem)
mS.group.add(g)
if(calendarDispList!=''):
for staffID in calendarDispList:
staff=Staff.objects.get(id=int(staffID))
mS.visibleStaff.add(staff)
mS.visibleStaff.add(mS)
mS.save()
cache.clear()
messages.success(request, '添加成功')
return True,''
def test_create(self):
current_count = Staff.all().count()
Staff.create("New Person")
self.assertEqual(Staff.all().count(), current_count + 1)
