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)