def on_save(self):
first_name = self.first_name_entry.get()
last_name = self.last_name_entry.get()
subdivision_index = self.subdivision_name_select.curselection()
subdivision_name = self.subdivision_name_select.get(subdivision_index)
subdivision = Subdivision.get_entity(name=subdivision_name)
if self.validate_form_data(first_name, last_name, subdivision):
new_employee = Employee.create(first_name, last_name, subdivision)
self.master.add_table_data(new_employee)
self.destroy()
else:
self.show_error()
def get_employee():
"""Gets employee name for purpose of creating time entry.
Asks user for employee name, determines whether name already exists,
returns employee name as a string.
"""
employee_name = input("Employee Name \n" +
"Please enter employee's name: ").strip()
try:
employee = Employee.create(name=employee_name)
except IntegrityError:
employee = Employee.get(Employee.name == employee_name)
return employee
def post_employee():
req = Req().request
req.add_argument('employee')
reqargs = req.parse_args()
resp = Resp().response
if not reqargs['employee']:
resp['meta']['code'] = 403
resp['meta']['description'] = 'incorrect request'
return jsonify(resp), 403
employee = Employee()
if not employee.create(
json.loads(reqargs['employee'].replace("'", '"'))):
resp['meta']['code'] = 404
resp['meta'][
'description'] = 'column not found/failed/login already exists'
return jsonify(resp), 404
employee.id_registered_by = g.employee.id_employee
try:
g.db_session.add(employee)
g.db_session.commit()
except sqlalchemy.exc.IntegrityError as ex:
log.warning('resources.Employees.post_employee(): ' + str(ex))
if 'UNIQUE constraint failed: employees.login' in str(ex):
resp['meta']['code'] = 403
resp['meta']['description'] = 'login already exist'
return jsonify(resp), 403
else:
resp['meta']['code'] = 403
resp['meta']['description'] = str(ex)
return jsonify(resp), 403
resp['meta']['description'] = 'success'
resp['content']['id_employee'] = employee.id_employee
return jsonify(resp), 200
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from models.base import Base
from util.config import app_config
from models.employee import Employee
from models.client import Client
from models.season_ticket import SeasonTicket
engine = create_engine(app_config.db_path, echo=False)
Session = sessionmaker(bind=engine)
if __name__ == "__main__":
Base.metadata.create_all(engine)
session = Session()
employee = Employee()
employee.create(
dict(login='******',
password='******',
role='admin',
status='active',
name='aaa',
surname='bbb',
patronymic='ccc'))
employee.id_registered_by = 1
session.add(employee)
session.commit()
session.close()
from peewee import *
from models.employee import Employee
from models.time_entry import Time_Entry
db = SqliteDatabase('time_entries.db')
def initialize():
db.connect()
db.create_tables([Employee, Time_Entry], safe=True)
if __name__ == '__main__':
initialize()
ee1 = Employee.create(name="Brian Peterson")
Time_Entry.create(name=ee1.name,
title="test title",
time_spent=50,
notes="These are the notes")
def test_get_employee(self):
with patch('builtins.input', side_effect=["Test Employee"]):
Employee.create(name="Test Employee")
expected_name = "Test Employee"
result = get_employee()
self.assertEqual(expected_name, result.name)
