Archived
94 lines
2.8 KiB
Python
94 lines
2.8 KiB
Python
import sqlite3
|
|
import json
|
|
|
|
def get_all_employees():
|
|
# Open a connection to the database
|
|
with sqlite3.connect("./kennel.sqlite3") as conn:
|
|
|
|
# Just use these. It's a Black Box.
|
|
conn.row_factory = sqlite3.Row
|
|
db_cursor = conn.cursor()
|
|
|
|
# Write the SQL query to get the information you want
|
|
db_cursor.execute("""
|
|
SELECT
|
|
e.id,
|
|
e.name,
|
|
e.address,
|
|
e.email,
|
|
e.password,
|
|
FROM employee e
|
|
""")
|
|
|
|
# Initialize an empty list to hold all employee representations
|
|
employees = []
|
|
|
|
# Convert rows of data into a Python list
|
|
dataset = db_cursor.fetchall()
|
|
|
|
# Iterate list of data returned from database
|
|
for row in dataset:
|
|
|
|
# Create an employee instance from the current row.
|
|
# Note that the database fields are specified in
|
|
# exact order of the parameters defined in the
|
|
# Animal class above.
|
|
employee = Employee(row['id'], row['name'], row['breed'],
|
|
row['status'], row['location_id'],
|
|
row['employee_id'])
|
|
|
|
employees.append(employee.__dict__)
|
|
|
|
# Use `json` package to properly serialize list as JSON
|
|
return json.dumps(employees)
|
|
|
|
def get_single_employee():
|
|
requested_employee = None
|
|
|
|
|
|
for employee in EMPLOYEES:
|
|
if EMPLOYEES["id"] == id:
|
|
requested_EMPLOYEE = employee
|
|
|
|
return requested_employee
|
|
|
|
def create_employee(employee):
|
|
# Get the id value of the last location in the list
|
|
max_id = EMPLOYEES[-1]["id"]
|
|
|
|
# Add 1 to whatever that number is
|
|
new_id = max_id + 1
|
|
|
|
# Add an `id` property to the location dictionary
|
|
employee["id"] = new_id
|
|
|
|
# Add the location dictionary to the list
|
|
EMPLOYEES.append(employee)
|
|
|
|
# Return the dictionary with `id` property added
|
|
return employee
|
|
def delete_employee(id):
|
|
# Initial -1 value for employee index, in case one isn't found
|
|
employee_index = -1
|
|
|
|
# Iterate the EMPLOYEES list, but use enumerate() so that you
|
|
# can access the index value of each item
|
|
for index, employee in enumerate(EMPLOYEES):
|
|
if employee["id"] == id:
|
|
# Found the employee. Store the current index.
|
|
employee_index = index
|
|
|
|
# If the employee was found, use pop(int) to remove it from list
|
|
if employee_index >= 0:
|
|
EMPLOYEES.pop(employee_index)
|
|
|
|
|
|
def update_employee(id, new_employee):
|
|
# Iterate the EMPLOYEES list, but use enumerate() so that
|
|
# you can access the index value of each item.
|
|
for index, employee in enumerate(EMPLOYEES):
|
|
if employee["id"] == id:
|
|
# Found the employee. Update the value.
|
|
EMPLOYEES[index] = new_employee
|
|
break
|