def get(self):
db.create_all()
emp = Employee(lastName='Jones', firstName='Bob', extension='x5555')
db.session.add(emp)
emp = Employee(lastName='Parker', firstName='Alice', extension='x6666')
db.session.add(emp)
db.session.commit()
return {'hello': 'world'}
def temporary_employee():
from api.models import Employee
employee = Employee(name="Funcionario",
email="*****@*****.**",
department="TI")
employee.save()
return employee
def post(self):
args = self.get_args()
emp = Employee.query.filter(
Employee.employeeNumber == args.employeeNumber).first()
if not emp:
emp = Employee()
emp.firstName = args.firstName
emp.lastName = args.lastName
db.session.add(emp)
db.session.commit()
return self.all_employees()
def import_employees(request):
if request.method == 'POST':
Employee.objects.all().delete()
req_body = request.FILES['file'].read()
employees_dict = simplexml.loads(req_body)['employees']
for e in employees_dict:
e_is_intern = 1 if '1' in e['is_intern'] else 0
employee_tosave = Employee(full_name=e['full_name'],
characteristics=e['characteristics'],
is_intern=e_is_intern)
employee_tosave.save()
return HttpResponse(status=200)
return HttpResponse(status=400)
def _req_body_to_employee(request):
request_body_fields = request.body.decode('utf-8')
employee_fields = json.loads(request_body_fields)
employee = Employee(
full_name=employee_fields['full_name'].encode('utf-8'),
characteristics=employee_fields['characteristics'].encode('utf-8'),
is_intern=employee_fields['is_intern'])
return employee
def get_employees_for_import():
db_employees = visions.Viwpremployees("ID, FirstName, LastName, EmployeeSSN", status="Active").fetch_all_dict()
employees = []
for row in db_employees:
employee = VisionsImportEmployee(row["ID"], row["FirstName"], row["LastName"], row["EmployeeSSN"])
if Employee.should_import_employee(employee):
employees.append(employee)
return employees
def new_employee():
employee = Employee(
request.json["first_name"],
request.json["last_name"],
request.json["department"],
request.json["has_kids"],
)
db.session.add(employee)
db.session.commit()
return employee_schema.jsonify(employee)
def resolve_create_employee(obj, info, name, department, joined_Date):
try:
print("resolve_create_employee")
joined_Date = datetime.strptime(joined_Date, '%d-%m-%Y').date()
employee = Employee(name=name,
department=department,
joined_Date=joined_Date)
db.session.add(employee)
db.session.commit()
payload = {"success": True, "employee": employee.to_dict()}
except ValueError: # date format errors
payload = {
"success":
False,
"errors": [
f"Incorrect date format provided. Date should be in "
f"the format dd-mm-yyyy"
]
}
return payload
def post(self):
form = copy.deepcopy(request.form)
if 'name' not in form:
return {'status': 'failure', 'error': 'Name not provided.'}, 403
if 'phone' not in form:
return {'status': 'failure', 'error': 'Phone number not provided.'}, 403
if 'shift' not in form:
return {'status': 'failure', 'error': 'Shift not provided.'}, 403
elif not (form['shift'] == "1" or form['shift'] == "2"):
return {'status': 'failure', 'error': 'Shift value incorrect.'}, 403
if 'hub_id' not in form:
return {'status': 'failure', 'error': 'Status not provided.'}, 403
schema = self._schema()
schema['name'] = form['name']
schema['shift'] = form['shift']
schema['phone'] = json.loads(form['phone'])
hub_id = form['hub_id']
try:
employee = EmployeeModel(data=schema)
hub = MywashHub.query.filter(MywashHub.str_id == hub_id).first()
employee.hub_id = hub.id
if 'is_active' in form:
if form['is_active'] == "true":
employee.is_active = True
elif form['is_active'] == "false":
employee.is_active = False
pgdb.session.add(employee)
pgdb.session.commit()
return {'status': 'success', 'str_id': employee.str_id, 'emp_id': "%.5d" % employee.id}
except Exception, e:
print e
return {'status': 'failure', 'error': 'db error.'}, 500
def import_employees(request):
if request.method == 'POST':
req_body = request.FILES['file'].read()
employees_dict = simplexml.loads(req_body)['employees']
employees = []
for e in employees_dict:
e_is_intern = 1 if '1' in e['is_intern'] else 0
employees.append(Employee(
0, e['full_name'], e['characteristics'], e_is_intern
))
with connect() as conn:
dbemployees = Employees(conn)
dbemployees.importall(employees)
return HttpResponse(status=200)
return HttpResponse(status=400)
def create(self, validated_data):
profile = validated_data["profile"]
user = validated_data["user"]
if Employee.objects.all().filter(
user__username=user['username']).exists():
error('El nombre de usuario \"' + user['username'] +
'\" ya ha sido utilizado.')
assist_ed = validated_data['assist_ed']
employee = Employee().create(username=user['username'],
password=user['password'],
charge=validated_data['charge'],
**profile)
employee.save()
for i in assist_ed:
employee.set_assist_ed(i)
return employee
def _map_employee(employee_db_data):
return Employee(*employee_db_data)
def create_users():
user_1 = User.objects.create_user("testuser1", "*****@*****.**",
"test123")
user_2 = User.objects.create_user("testuser2", "*****@*****.**",
"test123")
user_3 = User.objects.create_user("testuser3", "*****@*****.**",
"test123")
user_4 = User.objects.create_user("testuser4", "*****@*****.**",
"test123")
user_5 = User.objects.create_user("testuser5", "*****@*****.**",
"test123")
user_6 = User.objects.create_user("testuser6", "*****@*****.**",
"test123")
team_1 = Team(name="Team 1", description="Team 1")
team_1.save()
team_2 = Team(name="Team 2", description="Team 2")
team_2.save()
team_3 = Team(name="Team 3", description="Team 3")
team_3.save()
emp_1 = Employee(user=user_1, team=team_1, name="Test 1")
emp_1.save()
emp_2 = Employee(user=user_2, team=team_1, name="Test 2")
emp_2.save()
emp_3 = Employee(user=user_3, team=team_2, name="Test 3")
emp_3.save()
emp_4 = Employee(user=user_4, team=team_2, name="Test 4")
emp_4.save()
emp_5 = Employee(user=user_5, team=team_3, name="Test 5")
emp_5.save()
emp_6 = Employee(user=user_6, team=team_3, name="Test 6")
emp_6.save()
def main(args):
#print args
with open(args) as f:
r = csv.reader(f)
for data in r:
print data[4]
print data[5]
print data[6]
print data[7]
print '#'*60
#break
u = User.objects.create_user(data[1], data[24], 'm$Y$'+data[1])
e = Employee()
e.user = u
e.employee_code = data[1]
e.name = data[2]
e.qualification = data[3]
e.year_of_passing = data[4]
e.doj = data[5] if data[5] else None
e.dor = data[6] if data[6] else None
e.designation = data[7]
e.client_name = data[8]
e.project_name = data[9]
e.father = data[10]
e.permanent_address = data[11]
e.temporary_address = data[12]
e.email_id = data[13]
e.pan_no = data[14]
e.passport_no = data[15]
e.dob = data[16]
e.blood_group = data[17]
e.emergency_contact = data[18]
e.primary_contact = data[19]
e.secondary_contact = data[20]
e.reference = data[21]
e.msys_reference = data[22]
e.contact_no = data[23]
e.msys_email = data[24]
e.acount_no = data[25]
e.save()
del(u)
del(e)
if not form.get('hub', False):
return {'status': 'failure', 'error': 'hub not provided.'}, 403
emp_data = {
'name': session['emp_data']['name'],
'phone': form['phone'],
'is_active': True,
'shift': '1',
}
try:
hub = MywashHub.query.filter(MywashHub.data.contains({'short': form['hub']})).first()
except Exception, e:
return {'status': 'failure', 'error': 'hub db error.'}, 500
employee = EmployeeModel(data=emp_data)
employee.login_creds = dict(session.get('emp_data', {}))
employee.hub_id = hub.id
employee.is_active = True
try:
pgdb.session.add(employee)
pgdb.session.commit()
session['employee_registered'] = True
except Exception, e:
return {'status': 'failure', 'error': 'emp db error.'}, 500
return jsonify({'status': 'success'})
def logout():
session.clear()
def handle(self, *args, **options):
# Admin
a = Employee().create(username="******",
password="******",
charge="secretary",
first_name="Administrador",
last_name="de Aesculapius")
a.user.is_superuser = True
a.user.is_staff = True
a.save()
a.user.save()
# Employees
s = Employee().create(username="******",
password="******",
charge="secretary",
first_name="Secretaria Marta",
last_name="Mocha",
email=None)
s1 = Employee().create(username="******",
password="******",
charge="secretary",
first_name="Secretaria Olga",
last_name="Mora",
email=None)
d = Employee().create(username="******",
password="******",
charge="doctor",
first_name="Medica Romina",
last_name="Valestrini",
email=None)
d1 = Employee().create(username="******",
password="******",
charge="doctor",
first_name="Medico Ricardo",
last_name="Morales",
email=None)
d5 = Employee().create(username="******",
password="******",
charge="doctor",
first_name="Medico Ricardao",
last_name="Moraleas",
email=None)
d5.save()
s.save()
s1.save()
d.save()
d1.save()
s.set_assist_ed(d)
d1.set_assist_ed(s1)
# Patients
p = Profile(first_name="Diego", last_name="Velinsky")
p1 = Profile(first_name="Maria", last_name="de la Fuente")
p2 = Profile(first_name="Pedro", last_name="Pitaria")
p.save()
p1.save()
p2.save()
# Visits
v = Visit(doctor=d,
patient=p,
detail="Tenia gripe y le di antibiotico")
v1 = Visit(doctor=d,
patient=p1,
detail="Tenia fiebre y lo deje en cuarentena")
v2 = Visit(doctor=d,
patient=p2,
detail="Le dolia la cabeza y le di bayaspirina")
v3 = Visit(doctor=d1,
patient=p,
detail="Seguia con gripe y le di mas antibiotico")
v4 = Visit(doctor=d1,
patient=p,
detail="La gripe no ceso y lo mande al hospital")
v5 = Visit(doctor=d1,
patient=p1,
detail="Luego de la cuarentena murio")
v.save()
v1.save()
v2.save()
v3.save()
v4.save()
v5.save()