From a95a6842f9fa02c016e935d12f21c8e43cc7af80 Mon Sep 17 00:00:00 2001
From: Mauricio Baeza <elmau@correolibre.net>
Date: Sat, 2 Oct 2021 22:23:51 -0500
Subject: [PATCH] Save warehouse in product

---
 source/app/models/main.py | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/source/app/models/main.py b/source/app/models/main.py
index c3a6258..a6cb2e6 100644
--- a/source/app/models/main.py
+++ b/source/app/models/main.py
@@ -5280,6 +5280,7 @@ class Facturas(BaseModel):
                 receptor['NumRegIdTrib'] = invoice.cliente.id_fiscal
 
         conceptos = []
+
         rows = FacturasDetalle.select().where(FacturasDetalle.factura==invoice)
         for row in rows:
             if is_global:
@@ -5581,12 +5582,6 @@ class Facturas(BaseModel):
             )
             q.execute()
 
-            # ~ obj.xml = result['xml']
-            # ~ obj.uuid = result['uuid']
-            # ~ obj.fecha_timbrado = result['date']
-            # ~ obj.estatus = 'Timbrada'
-            # ~ obj.error = ''
-            # ~ obj.save()
             row = {'uuid': cfdi_uuid, 'estatus': 'Timbrada'}
             if enviar_correo:
                 cls._send(cls, id, rfc)
@@ -5892,6 +5887,9 @@ class Facturas(BaseModel):
                 if warehouse is None:
                     continue
 
+                p.warehouse = warehouse
+                p.save()
+
                 fields = (
                     WareHouseProduct.warehouse==warehouse,
                     WareHouseProduct.product==p.producto,
@@ -6463,6 +6461,7 @@ class FacturasDetalle(BaseModel):
     cuenta_predial = TextField(default='')
     empaques = DecimalField(default=0.0, max_digits=14, decimal_places=4,
         auto_round=True)
+    warehouse = ForeignKeyField(Almacenes, null=True)
 
     class Meta:
         order_by = ('factura',)
@@ -10296,6 +10295,12 @@ def _migrate_tables(rfc=''):
         warehouse = ForeignKeyField(Almacenes, null=True, to_field=Almacenes.id)
         migrations.append(migrator.add_column(table, 'warehouse', warehouse))
 
+    table = 'facturasdetalle'
+    columns = [c.name for c in database_proxy.get_columns(table)]
+    if not 'warehouse_id' in columns:
+        warehouse = ForeignKeyField(Almacenes, null=True, to_field=Almacenes.id)
+        migrations.append(migrator.add_column(table, 'warehouse_id', warehouse))
+
     if migrations:
         with database_proxy.atomic() as txn:
             migrate(*migrations)