def toggle_order_shipped():
""" toggle Shipped Date, to trigger balance adjustment """
""" also test join.
session.query(Customer).join(Invoice).filter(Invoice.amount == 8500).all()
"""
pre_cust = session.query(models.Customer).filter(models.Customer.Id == "ALFKI").one()
session.expunge(pre_cust)
print("")
test_order = session.query(models.Order).filter(models.Order.Id == 11011).join(models.Employee).one()
if test_order.ShippedDate is None or test_order.ShippedDate == "":
# with restored db, cust[ALFKI] has bal 960 & 3 unpaid orders, Order[11011) is 960, unshipped
test_order.ShippedDate = str(datetime.now())
print(prt("Shipping order - ShippedDate: ['' -> " + test_order.ShippedDate + "]" +
" for customer balance: " + str(pre_cust.Balance) +
", with UnpaidOrderCount: " + str(pre_cust.UnpaidOrderCount)))
else:
test_order.ShippedDate = None
print(prt("Returning order - ShippedDate: [ -> None]"))
insp = inspect(test_order)
# session.commit()
print("")
post_cust = session.query(models.Customer).filter(models.Customer.Id == "ALFKI").one()
logic_row = LogicRow(row=post_cust, old_row=pre_cust, ins_upd_dlt="*", nest_level=0, a_session=session, row_sets=None)
if abs(post_cust.Balance - pre_cust.Balance) == 960:
logic_row.log("Correct adjusted Customer Result")
assert True
else:
row_prt(post_cust, "\nERROR - incorrect adjusted Customer Result")
print("\n--> probable cause: Order customer update not written")
row_prt(pre_cust, "\npre_alfki")
assert False
def customer_flush_dirty(a_row, a_session: session):
"""
Called from listeners.py on before_flush
"""
old_row = get_old_row(a_row)
row_prt(a_row, "\ncustomer_flush_dirty")
customer_update(a_row, old_row, a_session)
def order_detail_flush_delete(a_row, a_session: session):
order = a_row.OrderHeader
old_order = ObjectView(
row2dict(order)) # hmm... key ShippedDate vs. "ShippedDate"
order.AmountTotal -= a_row.Amount
order_update(order, old_order, a_session)
row_prt(order,
"order_detail_flush_delete adjusted to: " + str(order.AmountTotal))
def order_detail_flush_new(a_row: models.OrderDetail, a_session: session):
"""
OrderDetail before_flush, new rows
compute amount, adjust Order.AmountTotal
.. which adjusts Customer.balance)
"""
# no "old" in inserts... old_row = get_old_row(a_row)
row_prt(a_row, "\norder_detail_flush_new") # readable log: curr/old values
# nice try.. product = row.Product
product = a_session.query(models.Product).\
filter(models.Product.Id == a_row.ProductId).one()
a_row.UnitPrice = product.UnitPrice
a_row.Amount = a_row.Quantity * a_row.UnitPrice
order = a_row.OrderHeader
"""
2 issues make this a little more complicated than expected:
1. can't just alter AmountTotal - does not trigger Order's before_flush
2. can't just call Order's before_flush - old values not available
"""
old_order = ObjectView(row2dict(order))
order.AmountTotal += a_row.Amount
order_update(order, old_order, a_session)
row_prt(order,
"order_detail_flush_new adjusted to: " + str(order.AmountTotal))
def order_detail_flush_dirty(a_row: models.OrderDetail, a_session: session):
old_row = get_old_row(a_row) # type: models.OrderDetail
if a_row.OrderId == old_row.OrderId:
if a_row.ProductId != old_row.ProductId:
product = a_session.query(models.Product). \
filter(models.Product.Id == a_row.ProductId).one()
a_row.UnitPrice = product.UnitPrice
a_row.Amount = a_row.UnitPrice * a_row.Quantity
if a_row.Amount != old_row.Amount:
order = a_row.OrderHeader
order.AmountTotal += a_row.Amount - old_row.Amount
old_order = ObjectView(row2dict(order))
order_update(order, old_order, a_session)
row_prt(
order, "order_detail_flush_dirty adjusted to: " +
str(order.AmountTotal))
else: # moved item to different order
order = a_row.OrderHeader # reduce the old one
order.AmountTotal -= old_row.Amount
old_order = ObjectView(row2dict(order))
order_update(order, old_order, a_session)
row_prt(
order,
"order_detail_flush_dirty adjusted to: " + str(order.AmountTotal))
if a_row.ProductId != old_row.ProductId:
product = a_session.query(models.Product). \
filter(models.Product.Id == old_row.ProductId).one()
a_row.UnitPrice = product.UnitPrice
a_row.Amount = a_row.UnitPrice * a_row.Quantity
order = a_session.query(models.Order). \
filter(models.Order.Id == a_row.OrderId).one()
old_order = ObjectView(row2dict(order))
order.AmountTotal += a_row.Amount
order_update(order, old_order, a_session)
row_prt(
order,
"order_detail_flush_dirty adjusted to: " + str(order.AmountTotal))
UnitPrice=18,
Discount=0)
new_order.OrderDetailList.append(new_item1)
new_item2 = models.OrderDetail(ProductId=2,
Amount=0,
Quantity=2,
UnitPrice=18,
Discount=0)
new_order.OrderDetailList.append(new_item2)
session.commit()
post_cust = session.query(
models.Customer).filter(models.Customer.Id == "ALFKI").one()
print("\nadd_order, update completed\n\n")
row_prt(new_order, "\nnew Order Result") # $18 + $38 = $56
if new_order.AmountTotal != 56:
print("==> ERROR - unexpected AmountTotal: " + str(new_order.AmountTotal) +
"... expected 56")
row_prt(new_item1, "\nnew Order Detail 1 Result") # 1 Chai @ $18
row_prt(new_item2, "\nnew Order Detail 2 Result") # 2 Chang @ $19 = $38
logic_row = LogicRow(row=post_cust,
old_row=pre_cust,
ins_upd_dlt="*",
nest_level=0,
a_session=session,
row_sets=None)
if post_cust.Balance == pre_cust.Balance + 56:
logic_row.log("Correct adjusted Customer Result")
assert True
def customer_commit_dirty(a_row, a_session: session):
old_row = get_old_row(a_row)
row_prt(a_row, "order_commit_dirty")
test_order.RequiredDate = str(datetime.now())
print(
prt("Shipping order - RequiredDate: ['' -> " +
test_order.RequiredDate + "]"))
else:
test_order.RequiredDate = None
print(prt("Returning order - RequiredDate: [ -> None]"))
insp = inspect(test_order)
session.commit()
print("")
post_cust = session.query(
models.Customer).filter(models.Customer.Id == "ALFKI").one()
logic_row = LogicRow(row=pre_cust,
old_row=post_cust,
ins_upd_dlt="*",
nest_level=0,
a_session=session,
row_sets=None)
if abs(post_cust.Balance - pre_cust.Balance) == 0:
logic_row.log("Correct non-adjusted Customer Result")
assert True
else:
row_prt(post_cust, "\nERROR - incorrect adjusted Customer Result")
print("\n--> probable cause: Order customer update not written")
row_prt(pre_cust, "\npre_alfki")
assert False
print("\nupd_order_shipped, ran to completion")
print("")
post_alfki = session.query(
models.Customer).filter(models.Customer.Id == "ALFKI").one()
logic_row = LogicRow(row=post_alfki,
old_row=pre_alfki,
ins_upd_dlt="*",
nest_level=0,
a_session=session,
row_sets=None)
if abs(post_alfki.Balance - pre_alfki.Balance) == 960:
logic_row.log("Correct non-adjusted Customer Result")
assert True
else:
row_prt(post_alfki, "\nERROR - incorrect adjusted Customer Result")
print("\n--> probable cause: Order customer update not written")
row_prt(pre_alfki, "\npre_alfki")
assert False
post_anatr = session.query(
models.Customer).filter(models.Customer.Id == "ANATR").one()
logic_row = LogicRow(row=post_anatr,
old_row=pre_anatr,
ins_upd_dlt="*",
nest_level=0,
a_session=session,
row_sets=None)
if abs(post_anatr.Balance - pre_anatr.Balance) == 960:
logic_row.log("Correct non-adjusted Customer Result")
def order_detail_commit_dirty(a_row, a_session: session):
old_row = get_old_row(a_row)
row_prt(a_row, "\norder_detail_commit_dirty")
