def fire_employee(employee):
try:
session.query(Employee).filter_by(
employee_id=employee.employee_id).delete()
session.commit()
except:
session.rollback()
def store_new_first_name(customer, new_value):
try:
customer.contactFirstName = new_value
# ....
session.commit()
except:
session.rollback()
def delete_product(product):
success = False
try:
session.delete(product)
session.commit()
success = True
except:
session.rollback()
finally:
return success
def delete_customer(customer):
success = False
try:
session.query(Customer).filter(
Customer.customer_id == customer[0].customer_id).delete()
session.commit()
success = True
except:
session.rollback()
finally:
return success
def add_product(product: SparePart):
success = False
try:
session.add(product)
session.commit()
success = True
except:
session.rollback()
finally:
return success
"""
def update_product(product: SparePart, attribute_name, new_value):
success = False
try:
for product_attribute, value in vars(product).items():
if product_attribute == attribute_name:
product.__setattr__(product_attribute, new_value)
session.commit()
success = True
except:
session.rollback()
finally:
return success
def create(**kwargs):
# TODO: Use default in model
kwargs['date_created'] = dt.datetime.utcnow()
try:
order = Order(**kwargs) # TODO: Check if order could be created
except Exception:
return None
try:
session.add(order)
except Exception as e:
print(repr(e))
return None
try:
session.commit()
except Exception as e:
print(repr(e))
session.rollback()
return None
return order
def create_order(new_order):
try:
session.add(new_order)
session.commit()
except:
session.rollback()
def update_stock(old_stock, quantity):
try:
old_stock.stock -= quantity
session.commit()
except:
session.rollback()
