pantry-track/application/poe/poe_processes.py
2025-07-02 18:20:52 -05:00

103 lines
4.6 KiB
Python

from application import postsqldb
from application.poe import poe_database
import datetime
import psycopg2
import config
def postTransaction(site_name, user_id, data: dict, conn=None):
#dict_keys(['item_id', 'logistics_info_id', 'barcode', 'item_name', 'transaction_type',
# 'quantity', 'description', 'cost', 'vendor', 'expires', 'location_id'])
def quantityFactory(quantity_on_hand:float, quantity:float, transaction_type:str):
if transaction_type == "Adjust In":
quantity_on_hand += quantity
return quantity_on_hand
if transaction_type == "Adjust Out":
quantity_on_hand -= quantity
return quantity_on_hand
raise Exception("The transaction type is wrong!")
self_conn = False
if not conn:
database_config = config.config()
conn = psycopg2.connect(**database_config)
conn.autocommit = False
self_conn = True
transaction_time = datetime.datetime.now()
cost_layer = postsqldb.CostLayerPayload(
aquisition_date=transaction_time,
quantity=float(data['quantity']),
cost=float(data['cost']),
currency_type="USD",
vendor=int(data['vendor']),
expires=data['expires']
)
transaction = postsqldb.TransactionPayload(
timestamp=transaction_time,
logistics_info_id=int(data['logistics_info_id']),
barcode=data['barcode'],
name=data['item_name'],
transaction_type=data['transaction_type'],
quantity=float(data['quantity']),
description=data['description'],
user_id=user_id,
)
#location = database.selectItemLocationsTuple(conn, site_name, payload=(data['item_id'], data['location_id']), convert=True)
location = poe_database.selectItemLocationsTuple(site_name, payload=(data['item_id'], data['location_id']), conn=conn)
cost_layers: list = location['cost_layers']
if data['transaction_type'] == "Adjust In":
cost_layer = poe_database.insertCostLayersTuple(site_name, cost_layer.payload(), conn=conn)
#cost_layer = database.insertCostLayersTuple(conn, site_name, cost_layer.payload(), convert=True)
cost_layers.append(cost_layer['id'])
if data['transaction_type'] == "Adjust Out":
if float(location['quantity_on_hand']) < float(data['quantity']):
return {"error":True, "message":f"The quantity on hand in the chosen location is not enough to satisfy your transaction!"}
#cost_layers = database.selectCostLayersTuple(conn, site_name, (location['id'], ), convert=True)
cost_layers = poe_database.selectCostLayersTuple(site_name, payload=(location['id'], ))
new_cost_layers = []
qty = float(data['quantity'])
for layer in cost_layers:
if qty == 0.0:
new_cost_layers.append(layer['id'])
elif qty >= float(layer['quantity']):
qty -= float(layer['quantity'])
layer['quantity'] = 0.0
else:
layer['quantity'] -= qty
new_cost_layers.append(layer['id'])
poe_database.updateCostLayersTuple(site_name, {'id': layer['id'], 'update': {'quantity': layer['quantity']}}, conn=conn)
#database.__updateTuple(conn, site_name, f"{site_name}_cost_layers", {'id': layer['id'], 'update': {'quantity': layer['quantity']}})
qty = 0.0
if layer['quantity'] == 0.0:
poe_database.deleteCostLayersTuple(site_name, (layer['id'],), conn=conn)
#database.deleteCostLayersTuple(conn, site_name, (layer['id'], ))
cost_layers = new_cost_layers
quantity_on_hand = quantityFactory(float(location['quantity_on_hand']), data['quantity'], data['transaction_type'])
updated_item_location_payload = (cost_layers, quantity_on_hand, data['item_id'], data['location_id'])
poe_database.updateItemLocation(site_name, updated_item_location_payload, conn=conn)
#database.updateItemLocation(conn, site_name, updated_item_location_payload)
site_location = poe_database.selectLocationsTuple(site_name, (location['location_id'], ), conn=conn)
#site_location = database.__selectTuple(conn, site_name, f"{site_name}_locations", (location['location_id'], ), convert=True)
transaction.data = {'location': site_location['uuid']}
poe_database.insertTransactionsTuple(site_name, transaction.payload(), conn=conn)
#database.insertTransactionsTuple(conn, site_name, transaction.payload())
if self_conn:
conn.rollback()
conn.close()
return {"error": False, "message":f"Transaction Successful!"}