def setUp(self): """Setup the test driver and create test users.""" self.driver = webdriver.Chrome() self.driver.get(self.get_server_url()) db.session.commit() db.drop_all() db.create_all() self.admin = Employee(username=test_admin_username, email=test_admin_email, password=test_admin_password, is_admin=True) self.employee = Employee(username=test_employee1_username, first_name=test_employee1_first_name, last_name=test_employee1_last_name, email=test_employee1_email, password=test_employee1_password) self.department = Department(name=test_department1_name, description=test_department1_description) self.role = Role(name=test_role1_name, description=test_role1_description) db.session.add(self.admin) db.session.add(self.employee) db.session.add(self.department) db.session.add(self.role) db.session.commit()
def register(): title = "Register Organization" formRegister = RegisterForm() if formRegister.validate_on_submit(): organization = Organization( name=formRegister.name_company.data, email=formRegister.email.data, typeofbusiness=formRegister.type_company.data, address=formRegister.address.data, phone_number=formRegister.business_phone_number.data) db.session.add(organization) db.session.commit() employee = Employee( fname=formRegister.manager_namef.data, lname=formRegister.manager_namel.data, email=formRegister.email.data, phone_number=formRegister.manager_phone_number.data, manager=True, organization_id=organization.id, firsttimelogin=False, question1=formRegister.question1.data, answer1=formRegister.answer1.data, question2=formRegister.question2.data, answer2=formRegister.answer2.data, ) employee.set_password(formRegister.enter_password.data) db.session.add(employee) db.session.commit() flash('New Organization has been added') return redirect(url_for('login')) return render_template("register.html", title=title, formRegister=formRegister)
def test_01_department_pages(self): #Test Registering a user testuser = Employee(first_name="testfirst", last_name="testlast", email="*****@*****.**", username="******", password="******", is_admin=False) testadmin = Employee(first_name="adminfirst", last_name="adminsecond", email="*****@*****.**", username="******", password="******", is_admin=True) db.session.add(testuser) db.session.add(testadmin) #Test going to departments with non-admin user target_url = url_for('admin.list_departments') redirect_url = url_for('auth.login') response = self.client.get(target_url) self.assertEqual(302, response.status_code, "###Test depertment pages: without logging in") #self.assertRedirects(response, redirect_url) #self.assertTrue(b"302 Error" in response.data) #TODO assert False, "Tests are not finished" self.login("testuser", "testpassword") response = self.client.get(url_for('admin.list_departments'), follow_redirects=True) #rv=list_departments() print("Response:", response.data)
def employee_create(request): if request.method == 'GET': employee_form = EmployeeForm() if request.method == 'POST': try: first_name = request.POST.get('first_name') email = request.POST.get('first_name') password = request.POST.get('password') print('first_name: {}, email: {}, password: {}'.format( first_name, email, password)) user = User.objects.create_user(username=first_name, first_name=first_name, email=email, password=password) employee = Employee( user=user, salary=request.POST.get('salary'), role=Role.objects.get(pk=request.POST.get('role'))) employee.save() messages.success(request, '{}'.format('Usuario guardado exitosamente')) return redirect('employee_list') except Exception as e: print(e) return render(request, 'app/employee_create.html', {'employee_form': employee_form})
def index(): emp = Employee().query.get(0) template = 'index.html' def get_subords(i): subords = Employee().query.filter(Employee.manager_id == i).all() return subords if request.method == 'GET': if 'emp_id' in request.args: try: emp_id = int(request.args['emp_id']) except ValueError: pass emp = Employee().query.get(emp_id) template = 'subords.html' if request.method=='POST': '''drag'n'drop changing manager directly in org-tree''' employee_id = int(request.form['set_manager_id[employee]']) new_manager = int(request.form['set_manager_id[manager]']) if employee_id != new_manager: empl = Employee().query.get(employee_id) empl.manager_id = new_manager db.session.add(empl) db.session.commit() return render_template( template, e=emp, Employee=Employee, get_subords=get_subords )
def register(): """ View function for the registration page. :return: HTTPResponse in the form of register.html template """ #If user is already logged in then redirect to homepage if current_user.is_authenticated: return redirect(url_for('index')) form = RegistrationForm() if form.validate_on_submit(): #Create a new employee instance and add it to database employee = Employee(username=form.username.data, email=form.email.data, name=form.name.data, contact=form.contact.data, address=form.address.data, manager=form.manager.data) employee.set_password(form.password.data) db.session.add(employee) db.session.commit() flash('Congratulations! You are now a registered user') return redirect(url_for('login')) return render_template('register.html', title='Register', form=form)
def test_company_membership(self): # create companies c1 = Company(company_name='test', email='*****@*****.**') c2 = Company(company_name='test2', email='*****@*****.**') # create employees with memberships e1 = Employee(email='*****@*****.**', first_name='thomas', last_name='mcdonnell', is_admin=True, member_of=c1) e2 = Employee(email='*****@*****.**', first_name='tom', last_name='jones', member_of=c2) # commit to database db.session.add(c1) db.session.add(c2) db.session.add(e1) db.session.add(e2) db.session.commit() # test memberships self.assertEqual(c1.employees.count(), 1) self.assertEqual(c2.employees.count(), 1) # test memberships self.assertEqual(c1.employees.all(), [e1]) self.assertNotEqual(c1.employees.all(), [e2]) self.assertEqual(c2.employees.all(), [e2]) self.assertNotEqual(c2.employees.all(), [e1])
def init_db(): db.drop_all() db.create_all() Role.insert_roles() Employee.create_admin() dept_fake() employee_fake() cost_fake()
def init_db(): db.drop_all() db.create_all() Role.insert_roles() Employee.create_admin() dept_fake() label_fake() employee_fake() cost_fake()
def test_employee_model_verifyPasswordIfTrue(self): # create test non-admin user employee = Employee(username="******", email="*****@*****.**", first_name="firstname", last_name="lastname", password="******") self.assertTrue(employee.verify_password("test2016"))
def setUp(self): db.create_all() admin = Employee(username="******", password="******", is_admin=True) employee = Employee(username="******", password="******") db.session.add(admin) db.session.add(employee) db.session.commit()
def setUp(self): db.create_all() admin = Employee(username="******", email="*****@*****.**", password="******", is_admin=True) employee = Employee(username="******", email="*****@*****.**", password="******") db.session.add_all([admin, employee]) db.session.commit()
def setUp(self): """Will be called before every test""" db.create_all() # create test admin user admin = Employee(username="******", password="******", is_admin=True) # create test non-admin user employee = Employee(username="******", password="******") # save users to database db.session.add(admin) db.session.add(employee) db.session.commit()
def setUp(self): db.create_all() #create test admin user admin = Employee(username="******", password="******", is_admin=True) #create a test non-admin user employee = Employee(username="******", password="******") db.session.add(admin) db.session.add(employee) db.session.commit()
def register(): if current_user.is_authenticated: return redirect(url_for('index')) form = RegistrationForm() if form.validate_on_submit(): user = Employee(ename=form.name.data, emp_id=form.emp_id.data) user.set_password(form.password.data) db.session.add(user) db.session.commit() flash('Congratulations, you are now a registered user!') return redirect(url_for('login')) return render_template('Login/register.html', title='Register', form=form)
def setUp(self): """ Call this before every test :return: """ db.create_all() # create test admin user admin = Employee(username='******', password='******', is_admin=True) employee = Employee(username="******", password="******") db.session.add(admin) db.session.add(employee) db.session.commit()
def create_employee(): data = request.get_json() or {} if 'name' not in data: return bad_request('name is required') if Employee.query.filter_by(name=data['name']).first(): return bad_request('use a different name') item = Employee() item.from_dict(data) db.session.add(item) db.session.commit() response = jsonify(item.to_dict()) response.status_code = 201 return response
def setUp(self): """Set up the testing environment.""" db.session.commit() db.drop_all() db.create_all() admin = Employee(username='******', password='******', is_admin=True) employee = Employee(username='******', password='******') db.session.add(admin) db.session.add(employee) db.session.commit()
def setUp(self): # Will be called before every test db.create_all() # create test admin user admin = Employee(username='******', password='******', is_admin=True) # create test non-admin user employee = Employee(username='******', password='******') # save users to database db.session.add(admin) db.session.add(employee) db.session.commit()
def setUp(self): # creating the database and the table db.create_all() # create test admin user admin = Employee(username='******', password='******', is_admin=True) # create non-admin user employee = Employee(username='******', password='******') # save the users to the database db.session.add(admin) db.session.add(employee) db.session.commit()
def add_employee(data): try: new_employee = Employee(employee_username=data['employee_username'], employee_email=data['employee_email'], employee_status=data['employee_status']) new_employee._set_password = data['employee_password'] db_session.add(new_employee) db_session.commit() db_session.refresh(new_employee) flash( 'Employee {} added successfully .'.format( data['employee_username']), 'success') except exc.SQLAlchemyError as e: db_session.rollback() flash('there was an error committing your data.', 'danger')
def register(): if current_user.is_authenticated: return redirect('/index') form = RegistrationForm() if form.validate_on_submit(): employee = Employee(employee_id=form.userid.data, employee_name=form.name.data, phone=form.phone.data, password=form.password.data) employee.set_password(form.password.data) db.session.add(employee) db.session.commit() flash('Registration successful.') return redirect('/login') return render_template('register.html', title='Register', form=form)
def setUp(self): """ Will be called before every test """ db.create_all() # create test admin user admin = Employee(username='******', password='******', is_admin=True) # create test non-admin user employee = Employee(username='******', password='******') # save users to database db.session.add(admin) db.session.add(employee) db.session.commit()
def mock_employee(): employee = Employee(first_name='John', last_name='Doe') db.session.add(employee) db.session.commit() yield employee db.session.delete(employee) db.session.commit()
def register(): if current_user.is_authenticated: return redirect(url_for('menu')) form = RegisterForm() if form.validate_on_submit(): work_days = "" if form.work_monday.data: work_days = 'monday,' if form.work_tuesday.data: work_days = 'tuesday,' if form.work_wednesday.data: work_days = 'wednesday,' if form.work_thursday.data: work_days = 'thursday,' if form.work_friday.data: work_days = 'friday,' if form.work_saturday.data: work_days = 'saturday,' if form.work_sunday.data: work_days = 'sunday,' employee = Employee(id=form.bartender_id.data, Wages=6, Shift=work_days, Name=form.first_name.data + " " + form.middle_name.data + " " + form.last_name.data) db.session.add(employee) db.session.commit() flash('You are now registered!') return redirect(url_for('login')) return render_template('newuser.html', title='register', form=form)
def employees_id_put(id,body,jwt = None): try: if not Employee.check_valid({'id':id,'login_user':body.get('login_user')}): raise Exception('登录用户(%s)已被使用,请重新指定!'% body.get('login_user')) if type(id)!='int': id=int(id) Employee.check_dateformat(body) Employee.query.filter(Employee.id == id).update(body) except Exception as e: db.session.rollback() return {"error":str(e)}, 422, {"content-type": "chatset=utf8"} data = Employee.query.filter_by(id=id).first_or_404() # data.cids = ',' # for auth in data.company_auths: # data.cids += str(auth.companyid) + ',' return data.to_json(), 201, {"content-type": "chatset=utf8"}
def create_employees(count=10, manager=False): for _ in range(count): email = fk.email() password = fk.password() confirmed = True firstname = fk.first_name() lastname = fk.last_name() salary = random.randint(MIN_SALARY, MAX_SALARY) user = User(email=email, password=password, confirmed=True, is_manager=manager) db.session.add(user) try: db.session.commit() except IntegrityError: db.session.rollback() employee = Employee(user=user, firstname=firstname, lastname=lastname, salary=salary) db.session.add(employee) try: db.session.commit() except IntegrityError: db.session.rollback()
def employees(department_id, vacancy_id): vacancy = Vacancy.query.filter_by(id=vacancy_id).first() if not vacancy: abort(404) form = EmployeeForm(obj=request.form) if form.validate_on_submit(): employee = Employee( name=form.name.data, surname=form.surname.data, position_id=form.position_id.data, email=form.email.data, phone_number=form.phone_number.data, birth_date=form.birth_date.data, department_id=form.department_id.data, vacancy=vacancy ) if form.director.data: update_director(department_id, employee, True) db.session.add(employee) vacancy.is_open = False vacancy.employee = employee vacancy.closing_date = form.start_date.data db.session.commit() update_position(employee, form.position_id.data, form.department_id.data, form.start_date.data) db.session.commit() flash_errors(form) return redirect(url_for('department', department_id=department_id))
def Register(request): name = request.POST.get("name") age = request.POST.get("age") cno = request.POST.get("cno") email = request.POST.get("email") image = request.FILES["file"] password = str(name[0]) + str(name[-1]) + str(len(cno)) if Employee.objects.filter(email=email): return render(request, "index.html", {"message": "Email already registred"}) else: Employee(name=name, age=age, contactno=cno, email=email, password=password, image=image).save() #for mail sending send_mail( "New Account Password:"******"hello mr/miss: " + name + "Your new account pasword is autogenerated \n the new password is " + password, settings.EMAIL_HOST_USER, [email]) #For message sending to mobile number accountSid = "ACaa3347d26965e518ac905c4b7199095e" authToken = "48d134d5473e93b558aabde8d4df8a18" twilioClient = Client(accountSid, authToken) myTwilioNumber = "+12512201736" destCellPhone = "+918639142835" myMessage = twilioClient.messages.create( body= "Thank you for register,Your account is created\n your temporary password is ---" + password, from_=myTwilioNumber, to=destCellPhone) return render(request, "index.html", {"message": "successs"})
def post(self): json_data = request.get_json() if not json_data: response = {'error': 'No Input Data Provided'} return jsonify(response), 400 employee = Employee.query.filter( Employee.email == json_data['email']).first() if employee: return {'message': 'Employee with that email already exists'} employee = Employee( date_of_first_appointment=json_data['date_of_first_appointment'], email=json_data['email'], first_name=json_data['first_name'], last_name=json_data['last_name'], password=json_data['password'], date_of_birth=json_data['date_of_birth'], next_of_kin=json_data['next_of_kin'], educational_qualification=json_data['educational_qualification'], address=json_data['address'], phone_number=json_data['phone_number'], marital_status=json_data['marital_status']) db.session.add(employee) db.session.commit() result = employee_schema.dump(employee) return {'status': 'success', 'data': result}, 201
def setup(): print('setup()') print('drop_all') db.drop_all() print('create_all') db.create_all() emp = Employee(name="Bobby", age=24, date=datetime.now()) db.session.add(emp) emp = Employee(name="Sue", age=43, date=datetime.now()) db.session.add(emp) print('commit') db.session.commit()
def employees(): current_company = Company.query.filter(Company.user_id == current_user.id).first() # send subsidiary item subsidiaries = current_company.subsidiaries # send new employee form newempform = NewEmployeeForm(request.form) sub_choices = [ (c.id, c.name) for c in Subsidiary.query.\ filter(Subsidiary.company_id == current_company.id).\ order_by(Subsidiary.name).all()] newempform.subsidiary_id.choices = sub_choices if request.method == "POST": # validate form if newempform.validate(): new_emp = Employee(newempform.email.data, newempform.password.data) if new_emp: new_emp.name = newempform.name.data new_emp.lastname = newempform.lastname.data if new_emp.second_name is not None: new_emp.second_name = newempform.second_name.data new_emp.lastname = newempform.lastname.data if new_emp.second_lastname is not None: new_emp.second_lastname = newempform.second_lastname.data new_emp.rfc = newempform.rfc.data new_emp.job_name = newempform.job_name.data new_emp.subsidiary_id = newempform.subsidiary_id.data # add new employee db.session.add(new_emp) try: # commit new employee to database db.session.commit() except IntegrityError: db.session.rollback() mess = "Either Email or the RFC exist in the database" return jsonify({"error":mess}), 400 else: abort(404) else: return jsonify(newempform.errors), 400 return render_template('admin/employee.html', newempform=newempform, current_company=current_company, subsidiaries=subsidiaries)
def form_valid(self, form): employee = Employee.get_by_id(self.kwargs['pk']) form.instance.person = employee return super(CalendarCreate, self).form_valid(form)