import sqlite3
from employee import Employee
conn = sqlite3.connect(':memory:') #for testing
c = conn.cursor()
c.execute("""CREATE TABLE employees(
first text,
last text,
pay integer
)""") # doc string
# c.execute("INSERT INTO employees VALUES ('Mary', 'Schafer', 70000)")
#
# conn.commit()
def insert_emp(emp):
with conn: # automaticlly commit, etc.
c.execute("INSERT INTO employees VALUES (:first, :last, :pay)",{'first': emp.first, 'last': emp.last, 'pay': emp.pay})
def get_emps_by_name(lastname):
c.execute("SELECT * FROM employees WHERE last = :last", {'last': 'Doe'})
return c.fetchall()
def update_pay(emp, pay):
with conn:
c.execute("""UPDATE employees SET pay = :pay
WHERE first = :first AND last = :last""",
{'first': emp.first, 'last': emp.last, 'pay': pay})
def remove_emp(emp):
with conn:
c.execute("DELETE from employees WHERE first = :first AND last = :last",
{'first': emp.first, 'last': emp.last})
emp_1 = Employee('John', 'Doe',80000)
emp_2 = Employee('Jane', 'Doe',90000)
insert_emp(emp_1)
insert_emp(emp_2)
emps = get_emps_by_name('Doe')
print(emps)
update_pay(emp_2, 95000)
remove_emp(emp_1)
emps = get_emps_by_name('Doe')
print(emps)
conn.close()