[ADD]l10n_ar_stock: massive COt

Author: jcadhoc

l10n_ar_stock/models/stock_picking.py

@@ -87,7 +87,7 @@ def get_arba_file_data(
 
         company = self.mapped("company_id")
         if len(company) > 1:
-            raise UserError(self.env._("Los remitos seleccionados deben pertenecer a la misma " "compañía"))
+            raise UserError(self.env._("Los remitos seleccionados deben pertenecer a la misma compañía"))
         cuit = company.partner_id.ensure_vat()
         cuit_carrier = carrier_partner.ensure_vat()
 
@@ -106,7 +106,7 @@ def get_arba_file_data(
         nro_secuencial = self.env["ir.sequence"].with_company(company).next_by_code("arba.cot.file")
         if not nro_secuencial:
             raise UserError(
-                self.env._('No sequence found for COT files (code = "arba.cot.file") on ' 'company "%s', company.name)
+                self.env._('No sequence found for COT files (code = "arba.cot.file") on company "%s', company.name)
             )
 
         filename = "TB_%s_%s%s_%s_%s.txt" % (
@@ -136,7 +136,7 @@ def get_arba_file_data(
 
             if not source_partner.state_id.code or not dest_partner.state_id.code:
                 raise UserError(
-                    self.env._("Las provincias de origen y destino son obligatorias y " "deben tener un código válido")
+                    self.env._("Las provincias de origen y destino son obligatorias y deben tener un código válido")
                 )
 
             if not rec.document_type_id:
@@ -147,7 +147,7 @@ def get_arba_file_data(
             if not CODIGO_DGI or not letter:
                 raise UserError(
                     self.env._(
-                        "Document type has no validator, code or letter configured" " (Id: %s", rec.document_type_id.id
+                        "Document type has no validator, code or letter configured (Id: %s", rec.document_type_id.id
                     )
                 )
 
@@ -294,7 +294,7 @@ def get_arba_file_data(
                     if not uom_arba_with_code:
                         raise UserError(
                             self.env._(
-                                'No arba code for uom "%(uom)s" (Id: %(id)s) or any uom in ' 'category "%(category)s"',
+                                'No arba code for uom "%(uom)s" (Id: %(id)s) or any uom in category "%(category)s"',
                                 uom=line.product_uom.name,
                                 id=line.product_uom.id,
                                 category=line.product_uom.category_id.name,
@@ -352,80 +352,89 @@ def do_pyafipws_presentar_remito(
         prod_no_term_dev,
         importe,
     ):
-        self.ensure_one()
-
-        COT = self.company_id.arba_cot_connect()
+        # self.ensure_one()
+        for rec in self:
+            COT = rec.company_id.arba_cot_connect()
 
-        if not carrier_partner:
-            raise UserError(
-                'Debe vincular una "Empresa transportista" a la forma de envío'
-                " seleccionada o elegir otra forma de envío"
+            if not carrier_partner:
+                raise UserError(
+                    'Debe vincular una "Empresa transportista" a la forma de envío'
+                    " seleccionada o elegir otra forma de envío"
+                )
+            content, filename = rec.get_arba_file_data(
+                datetime_out,
+                tipo_recorrido,
+                carrier_partner,
+                patente_vehiculo,
+                patente_acoplado,
+                prod_no_term_dev,
+                importe,
             )
-        content, filename = self.get_arba_file_data(
-            datetime_out, tipo_recorrido, carrier_partner, patente_vehiculo, patente_acoplado, prod_no_term_dev, importe
-        )
 
-        filename = "/tmp/%s" % filename
-        file = open(filename, "w")
-        file.write(content)
-        file.close()
-        _logger.info('Presentando COT con archivo "%s"' % filename)
-        COT.PresentarRemito(filename, testing="")
-        os.remove(filename)
-
-        if COT.TipoError:
-            msg = self.env._(
-                "Error al presentar remito:\n"
-                "* Tipo Error: %(tipo)s\n"
-                "* Codigo Error: %(cod)s\n"
-                "* Mensaje Error: %(msj)s",
-                tipo=COT.TipoError,
-                cod=COT.CodigoError,
-                msj=COT.MensajeError,
+            filename = "/tmp/%s" % filename
+            file = open(filename, "w")
+            file.write(content)
+            file.close()
+            _logger.info('Presentando COT con archivo "%s"' % filename)
+            COT.PresentarRemito(filename, testing="")
+            os.remove(filename)
+
+            if COT.TipoError:
+                msg = rec.env._(
+                    "Error al presentar remito:\n"
+                    "* Tipo Error: %(tipo)s\n"
+                    "* Codigo Error: %(cod)s\n"
+                    "* Mensaje Error: %(msj)s",
+                    tipo=COT.TipoError,
+                    cod=COT.CodigoError,
+                    msj=COT.MensajeError,
+                )
+                _logger.warning(msg)
+                raise UserError(msg)
+            elif COT.Excepcion:
+                msg = rec.env._("Error al presentar remito:\n* %s", COT.Excepcion)
+                _logger.warning(msg)
+                raise UserError(msg)
+
+            errors = []
+            while COT.LeerErrorValidacion():
+                errors.append(
+                    ("* MensajeError: %s\n* TipoError: %s\n* CodigoError: %s\n")
+                    % (COT.MensajeError, COT.TipoError, COT.CodigoError)
+                )
+
+            if errors:
+                raise UserError(rec.env._("Error al presentar remito:\n%s", "\n".join(errors)))
+
+            attachments = [(filename, content)]
+            body = """
+    <p>
+        Resultado solicitud COT:
+        <ul>
+            <li>Número Comprobante: %s</li>
+            <li>Codigo Integridad: %s</li>
+            <li>Procesado: %s</li>
+            <li>Número Único: %s</li>
+            <li>COT: %s</li>
+        </ul>
+    </p>
+    """ % (COT.NumeroComprobante, COT.CodigoIntegridad, COT.Procesado, COT.NumeroUnico, COT.COT)
+
+            rec.write(
+                {
+                    "cot_numero_unico": COT.NumeroComprobante,
+                    "cot_numero_comprobante": COT.NumeroUnico,
+                    "cot": COT.COT,
+                }
             )
-            _logger.warning(msg)
-            raise UserError(msg)
-        elif COT.Excepcion:
-            msg = self.env._("Error al presentar remito:\n* %s", COT.Excepcion)
-            _logger.warning(msg)
-            raise UserError(msg)
-
-        errors = []
-        while COT.LeerErrorValidacion():
-            errors.append(
-                ("* MensajeError: %s\n" "* TipoError: %s\n" "* CodigoError: %s\n")
-                % (COT.MensajeError, COT.TipoError, COT.CodigoError)
+            rec.message_post(
+                body=body,
+                subject=rec.env._("Remito Electrónico Solicitado"),
+                attachments=attachments,
+                body_is_html=True,
             )
 
-        if errors:
-            raise UserError(self.env._("Error al presentar remito:\n%s", "\n".join(errors)))
-
-        attachments = [(filename, content)]
-        body = """
-<p>
-    Resultado solicitud COT:
-    <ul>
-        <li>Número Comprobante: %s</li>
-        <li>Codigo Integridad: %s</li>
-        <li>Procesado: %s</li>
-        <li>Número Único: %s</li>
-        <li>COT: %s</li>
-    </ul>
-</p>
-""" % (COT.NumeroComprobante, COT.CodigoIntegridad, COT.Procesado, COT.NumeroUnico, COT.COT)
-
-        self.write(
-            {
-                "cot_numero_unico": COT.NumeroComprobante,
-                "cot_numero_comprobante": COT.NumeroUnico,
-                "cot": COT.COT,
-            }
-        )
-        self.message_post(
-            body=body, subject=self.env._("Remito Electrónico Solicitado"), attachments=attachments, body_is_html=True
-        )
-
-        return True
+            return True
 
     def _action_done(self):
         res = super()._action_done()

l10n_ar_stock/views/stock_picking_views.xml

@@ -8,7 +8,7 @@
             <button name="do_print_picking" position="after">
                 <button name="%(action_arba_cot_wizard)d" string="Obtener COT (BETA)"
                     help="Obtener Código de Operaciones de Traslado (ARBA / Santa Fe)"
-                    type="action" invisible="not book_id or picking_type_code == 'incoming' or state != 'done'"
+                    type="action" invisible="not book_id or picking_type_code == 'incoming' or state != 'done' or cot"
                     context="{'default_importe': declared_value}"
                     groups="l10n_ar_stock.arba_cot_enabled"/>
             </button>
@@ -25,4 +25,14 @@
         </field>
     </record>
 
+    <record id="stock_picking_action_arba_cot_wizard_batch" model="ir.actions.act_window">
+        <field name="name">Obtener COT para múltiples remitos</field>
+        <field name="res_model">arba.cot.wizard</field>
+        <field name="view_mode">form</field>
+        <field name="target">new</field>
+        <field name="context">{'active_ids': active_ids, 'active_model': 'stock.picking'}</field>
+        <field name="binding_model_id" ref="stock.model_stock_picking"/>
+        <field name="binding_view_types">list</field>
+    </record>
+
 </odoo>

l10n_ar_stock/wizards/arba_cot_wizard.py

@@ -13,7 +13,7 @@ class ArbaCotWizard(models.TransientModel):
     _description = "arba.cot.wizard"
 
     datetime_out = fields.Datetime(
-        required=True, help="Fecha de salida. No debe ser inferior a ayer ni superior a " "dentro de 30 días."
+        required=True, help="Fecha de salida. No debe ser inferior a ayer ni superior a dentro de 30 días."
     )
     tipo_recorrido = fields.Selection(
         [("U", "Urbano"), ("R", "Rural"), ("M", "Mixto")],
@@ -27,7 +27,7 @@ class ArbaCotWizard(models.TransientModel):
     )
 
     patente_vehiculo = fields.Char(
-        help="Requerido si CUIT Transportista = CUIT Compañía\n" "3 letras y 3 numeros o 2 letras, 3 números y 2 letras"
+        help="Requerido si CUIT Transportista = CUIT Compañía\n3 letras y 3 numeros o 2 letras, 3 números y 2 letras"
     )
     patente_acoplado = fields.Char(help="3 letras y 3 numeros o 2 letras, 3 números y 2 letras")
     prod_no_term_dev = fields.Selection(
@@ -71,15 +71,29 @@ def _constrain_check_format_patente(self):
 
     def confirm(self):
         self.ensure_one()
-        if self._context.get("active_model") != "stock.picking":
-            return True
-        pickings = self.env["stock.picking"].browse(self._context.get("active_ids"))
-        pickings.do_pyafipws_presentar_remito(
-            fields.Date.from_string(self.datetime_out),
-            self.tipo_recorrido,
-            self.partner_id,
-            self.patente_vehiculo,
-            self.patente_acoplado,
-            self.prod_no_term_dev,
-            self.importe,
-        )
+        ctx = self._context or {}
+        pickings = self.env["stock.picking"]
+        # Soporta acción desde batch o desde remitos individuales
+        if ctx.get("active_model") == "stock.picking.batch":
+            batch_ids = ctx.get("active_ids", [])
+            batches = self.env["stock.picking.batch"].browse(batch_ids)
+            pickings = batches.mapped("picking_ids")
+        elif ctx.get("active_model") == "stock.picking":
+            picking_ids = ctx.get("active_ids", [])
+            pickings = self.env["stock.picking"].browse(picking_ids)
+        else:
+            picking_ids = ctx.get("active_ids", [])
+            pickings = self.env["stock.picking"].browse(picking_ids)
+
+        for pick in pickings:
+            pick.do_pyafipws_presentar_remito(
+                fields.Date.from_string(self.datetime_out),
+                self.tipo_recorrido,
+                self.partner_id,
+                self.patente_vehiculo,
+                self.patente_acoplado,
+                self.prod_no_term_dev,
+                self.importe,
+            )
+
+        return True