Re-process USF invoices safely

CORE-POS · Oct 31, 2019 · 904acc2 · 904acc2
1 parent a953605
commit 904acc2
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/fannie/modules/plugins2.0/USFoodsInvoiceGrabber/USFTask.php b/fannie/modules/plugins2.0/USFoodsInvoiceGrabber/USFTask.php
@@ -36,6 +36,7 @@ public function run()
             }
 
             $this->connection->startTransaction();
+            $first = true;
             while (!feof($fp)) {
                 $data = fgetcsv($fp);
 
@@ -88,11 +89,11 @@ public function run()
                 $unitPrice = round($unitPrice, 3);
 
                 $upc = $this->updateCatalog($vendorID, $sku, $item, $brand, $size, $units, $unitPrice, $doUpdates);
-                $orderID = $this->findPO($store, $invoice, $vendorID, $orderDate);
+                $orderID = $this->findPO($store, $invoice, $vendorID, $orderDate, $first);
                 if (!$orderID) continue;
                 $this->updatePO($orderID, $shipDate, $ordered, $shipped, $sku, $upc, $item, $brand, $size, $units, $unitPrice, $fullPrice);
                 $orders[$orderID] = $invoice;
-                $count++;
+                $first = false;
             }
             $this->connection->commitTransaction();
         }
@@ -159,7 +160,7 @@ private function updatePO($orderID, $shipDate, $ordered, $shipped, $sku, $upc, $
     }
 
     private $poP = false;
-    private function findPO($store, $invoice, $vendor, $date)
+    private function findPO($store, $invoice, $vendor, $date, $first)
     {
         if (!$this->poP) {
             $this->poP = $this->connection->prepare("SELECT orderID FROM PurchaseOrder WHERE storeID=? AND vendorID=? AND vendorInvoiceID=?");
@@ -176,6 +177,11 @@ private function findPO($store, $invoice, $vendor, $date)
             $model->vendorInvoiceID($invoice);
             $orderID = $model->save();
 
+        } elseif ($first) {
+            // zero out old amounts to avoid doubling them on re-import
+            $prep = $this->connection->prepare("UPDATE PurchaseOrderItems SET quantity=0, receivedQty=0, receivedTotalCost=0
+                WHERE orderID=?");
+            $this->connection->execute($prep, array($orderID));
         }
 
         return $orderID;