From 5c857d8a9e04b9d6b994ef32338be03270747ad3 Mon Sep 17 00:00:00 2001
From: sinergia <alejandro.ayala@onecluster.org>
Date: Wed, 3 Aug 2022 13:14:15 -0500
Subject: [PATCH] changes de state and propietary of equipment

---
 __init__.py                     |   1 +
 configuration_equipment.xml     |   6 +
 equipment.py                    |  17 ++-
 equipment.xml                   |   5 +-
 sale.py                         | 230 ++++++++++++++++++--------------
 subscription.py                 |  28 +++-
 view/optical_equipment_form.xml |   3 +-
 7 files changed, 174 insertions(+), 116 deletions(-)

diff --git a/__init__.py b/__init__.py
index c2a5519..ef75e64 100644
--- a/__init__.py
+++ b/__init__.py
@@ -11,6 +11,7 @@ def register():
         product.Product,
         purchase.Purchase,
         purchase.Line,
+        sale.Sale,
         sale.SaleLine,
         sale.CreateSubscriptionStart,
         equipment.OpticalEquipment,
diff --git a/configuration_equipment.xml b/configuration_equipment.xml
index ac2a452..49ca580 100644
--- a/configuration_equipment.xml
+++ b/configuration_equipment.xml
@@ -18,7 +18,13 @@
 	    <field name="act_window" ref="act_optical_equipment_configuration_form"/>
 	</record>
 	<menuitem
+            name="Configuration"
             parent="menu_equipment"
+            sequence="0"
+            id="menu_equipment_configuration"
+            icon="tryton-settings"/>
+	<menuitem
+            parent="menu_equipment_configuration"
             action="act_optical_equipment_configuration_form"
             sequence="10"
             id="menu_optical_equipment_configuration"
diff --git a/equipment.py b/equipment.py
index 17f8726..07475e9 100644
--- a/equipment.py
+++ b/equipment.py
@@ -108,8 +108,8 @@ class OpticalEquipment(Workflow, ModelSQL, ModelView):
             ('uncontrated', 'contrated'),
             })
         cls._buttons.update({
-            'draft': {
-                'invisible': Eval('state') == 'draft'},
+            # 'draft': {
+            #     'invisible': Eval('state') == 'draft'},
             'registred': {
                 'invisible': Eval('state').in_(['registred', 'contrated'])}}
         )
@@ -204,20 +204,19 @@ class OpticalEquipment(Workflow, ModelSQL, ModelView):
             self.health_register=None
             self.software_version=None
 
-    @classmethod
-    @ModelView.button
-    @Workflow.transition('draft')
-    def draft(cls, equipments):
-        pass
+    # @classmethod
+    # @ModelView.button
+    # @Workflow.transition('draft')
+    # def draft(cls, equipments):
+    #     pass
 
     @classmethod
     @ModelView.button
     @Workflow.transition('registred')
     def registred(cls, equipments):
-        #raise UserError(str(equipments[0].serial))
         for i in equipments:
             if i.serial == None:
-                raise UserError(str(i.serial))
+                raise UserError(str("El Equipo no cuenta con un Serial"))
             else:
                 cls.set_code(equipments)
             
diff --git a/equipment.xml b/equipment.xml
index 57547ff..de77bb7 100644
--- a/equipment.xml
+++ b/equipment.xml
@@ -83,14 +83,17 @@
             <field name="domain"></field>
             <field name="act_window" ref="act_optical_equipment_form"/>
         </record>
+	<!--
 	<record model="ir.model.button" id="draft_equipment_button">
 	    <field name="name">draft</field>
 	    <field name="string">Draft</field>
 	    <field name="model" search="[('model', '=', 'optical_equipment.equipment')]"/>
-	</record>
+	    </record>
+	 -->
 	<record model="ir.model.button" id="registred_equipment_button">
 	    <field name="name">registred</field>
 	    <field name="string">Registred</field>
+	    <field name="confirm">Are you sure you want to registred these equipments?</field>
 	    <field name="model" search="[('model', '=', 'optical_equipment.equipment')]"/>
 	</record>
 	<menuitem parent="menu_equipment"
diff --git a/sale.py b/sale.py
index db1b6c2..5d88e4b 100644
--- a/sale.py
+++ b/sale.py
@@ -5,119 +5,55 @@ from trytond.pyson import Eval, Bool, If
 from decimal import Decimal
 from trytond.modules.product import price_digits
 from trytond.transaction import Transaction
+from trytond.model import Workflow
+from trytond.modules.company.model import (
+    employee_field, set_employee, reset_employee)
+
 from trytond.exceptions import UserError
 
 from trytond.wizard import (
     Button, StateAction, StateTransition, StateView, Wizard)
 
+class Sale(metaclass=PoolMeta):
+    'Sale'
+    __name__ = 'sale.sale'
 
-class CreateSubscriptionStart(ModelView):
-    'Create Subscription Start'
-    __name__ = 'sale.create.subscription.start'
+    @classmethod
+    @ModelView.button
+    @Workflow.transition('confirmed')
+    @set_employee('confirmed_by')
+    def confirm(cls, sales):
+        pool = Pool()
+        Configuration = pool.get('sale.configuration')
+        cls.set_sale_date(sales)
+        cls.store_cache(sales)
+        config = Configuration(1)
 
-    start_date = fields.Date("Start Date", required=True)
-    end_date = fields.Date("End Date", required=True)
-    invoice_recurrence = fields.Many2One('sale.subscription.recurrence.rule.set',
-                                         "Invoice Recurrence",required=True)
-    invoice_start_date = fields.Date("Invoice Start Date",
-                                     help='Billing start date')
-    service = fields.Many2One('sale.subscription.service', "Service")
-    quantity = fields.Float("Quantity", digits='unit')
-    unit_price = Monetary("Unit Price", currency='currency',
-                          digits=price_digits,
-                          )
+        pool = Pool()
+        Equipments = pool.get('optical_equipment.equipment')
+        
+        for sale in sales:
+            for line in sale.lines:
+                if line.equipment:
+                    equipment=line.equipment
+                    equipment.propietary=sale.party.id
+                    equipment.propietary_address=sale.shipment_address.id
+                    equipment.state="uncontrated"
+                    equipment.save()                
+        
+        with Transaction().set_context(
+                queue_name='sale',
+                queue_scheduled_at=config.sale_process_after):
+            cls.__queue__.process(sales)
     
-    @classmethod
-    def default_start_date(cls):
-        pool = Pool()
-        Date = pool.get('ir.date')
-        return Date.today()
 
-    @classmethod
-    @fields.depends(methods=['default_start_date'])
-    def default_invoice_start_date(self):
-        invoice_start_date = self.default_start_date()
-
-        return invoice_start_date
-        
-class CreateSubscription(Wizard):
-    'Create Subscription'
-    __name__ = 'sale.create.subscription'
-
-    start = StateView('sale.create.subscription.start',
-                      'optical_equipment.create_subscription_view_form',[
-                          Button('Cancel', 'end', 'tryton-cancel'),
-                          Button('Create', 'create_subscription', 'tryton-ok', default=True),
-                      ])
-    create_subscription = StateAction('sale_subscription.act_subscription_form')
-
-    done_ = StateView('sale.create.subscription.start',
-                      'optical_equipment.create_subscription_view_form',[
-                          Button('Done', 'end', 'tryton-cancel'),
-                      ])
-    @property
-    def _subscription_start(self):
-        return dict(start_date=self.start.start_date,
-                    end_date=self.start.end_date,
-                    invoice_recurrence=self.start.invoice_recurrence,
-                    invoice_start_date=self.start.invoice_start_date,
-                    service=self.start.service,
-                    quantity=self.start.quantity,
-                    unit_price=self.start.unit_price)
-
-    def _equipments_to_subscription(self):
-        sale = self.records[0]
-        equipments_to_subscription = []
-        for line in sale.lines:
-            if line.product_equipment:
-                equipments_to_subscription.append(line.equipment)
-
-        return equipments_to_subscription
-                
-    def do_create_subscription(self, action):
-        pool = Pool()
-        Subscription = pool.get('sale.subscription')
-        SubscriptionLine = pool.get('sale.subscription.line')
-
-        sale = self.records[0]  
-        a = self._subscription_start
-
-        equipments_to_subscription=self._equipments_to_subscription()
-        subscription_lines = [SubscriptionLine(
-            start_date=a['start_date'],
-            end_date=a['end_date'],
-            consumption_recurrence=a['invoice_recurrence'],
-            service=a['service'],
-            unit=a['service'].product.default_uom,
-            quantity=a['quantity'],
-            unit_price=a['unit_price']
-        )]
-
-        subscription = Subscription(
-            start_date=a['start_date'],
-            end_date=a['end_date'],
-            invoice_recurrence=a['invoice_recurrence'],
-            invoice_start_date=a['invoice_start_date'],
-            party=sale.party.id,
-            contact=sale.contact.id if sale.contact else None,
-            invoice_party=sale.invoice_party.id if sale.invoice_party else None,
-            invoice_address=sale.invoice_address.id,
-            payment_term=sale.payment_term.id if sale.payment_term else None,
-            lines=subscription_lines,
-            equipments=equipments_to_subscription,
-        )
-        subscription.save()
-
-        return 'done_'
-        
 class SaleLine(metaclass=PoolMeta):
     'SaleLine'
     __name__ = 'sale.line'
 
     product_equipment = fields.Boolean("Product Equipment")
     equipment = fields.Many2One('optical_equipment.equipment', "Equipment",
-                                domain=[('state', '=', 'registred'),
-                                        (Eval('product'), '=', Eval('_parent_sale.line.product'))],
+                                domain=[('state', '=', 'registred')],
                                 states={'invisible': If(~Eval('product_equipment'), True)},)
     address_equipment = fields.Many2One('party.address', "Direccion")
     unit_digits = fields.Function(fields.Integer('Unit Digits'),
@@ -218,3 +154,101 @@ class SaleLine(metaclass=PoolMeta):
             ('//page[@id="equipment"]', 'states', {
                     'invisible': ~Eval('product_equipment', True),
                     })]
+
+
+class CreateSubscriptionStart(ModelView):
+    'Create Subscription Start'
+    __name__ = 'sale.create.subscription.start'
+
+    start_date = fields.Date("Start Date", required=True)
+    end_date = fields.Date("End Date", required=True)
+    invoice_recurrence = fields.Many2One('sale.subscription.recurrence.rule.set',
+                                         "Invoice Recurrence",required=True)
+    invoice_start_date = fields.Date("Invoice Start Date",
+                                     help='Billing start date')
+    service = fields.Many2One('sale.subscription.service', "Service")
+    quantity = fields.Float("Quantity", digits='unit')
+    unit_price = Monetary("Unit Price", currency='currency',
+                          digits=price_digits,
+                          )
+    
+    @classmethod
+    def default_start_date(cls):
+        pool = Pool()
+        Date = pool.get('ir.date')
+        return Date.today()
+
+    @classmethod
+    @fields.depends(methods=['default_start_date'])
+    def default_invoice_start_date(self):
+        invoice_start_date = self.default_start_date()
+
+        return invoice_start_date
+        
+class CreateSubscription(Wizard):
+    'Create Subscription'
+    __name__ = 'sale.create.subscription'
+
+    start = StateView('sale.create.subscription.start',
+                      'optical_equipment.create_subscription_view_form',[
+                          Button('Cancel', 'end', 'tryton-cancel'),
+                          Button('Create', 'create_subscription', 'tryton-ok', default=True),
+                      ])
+    create_subscription = StateAction('sale_subscription.act_subscription_form')
+
+    done_ = StateView('sale.create.subscription.start',
+                      'optical_equipment.create_subscription_view_form',[
+                          Button('Done', 'end', 'tryton-cancel'),
+                      ])
+    @property
+    def _subscription_start(self):
+        return dict(start_date=self.start.start_date,
+                    end_date=self.start.end_date,
+                    invoice_recurrence=self.start.invoice_recurrence,
+                    invoice_start_date=self.start.invoice_start_date,
+                    service=self.start.service,
+                    quantity=self.start.quantity,
+                    unit_price=self.start.unit_price)
+
+    def _equipments_to_subscription(self):
+        sale = self.records[0]
+        equipments_to_subscription = []
+        for line in sale.lines:
+            if line.product_equipment:
+                equipments_to_subscription.append(line.equipment)
+
+        return equipments_to_subscription
+                
+    def do_create_subscription(self, action):
+        pool = Pool()
+        Subscription = pool.get('sale.subscription')
+        SubscriptionLine = pool.get('sale.subscription.line')
+
+        sale = self.records[0]  
+        a = self._subscription_start
+
+        equipments_to_subscription=self._equipments_to_subscription()
+        subscription_lines = [SubscriptionLine(
+            start_date=a['start_date'],
+            end_date=a['end_date'],
+            consumption_recurrence=a['invoice_recurrence'],
+            service=a['service'],
+            unit=a['service'].product.default_uom,
+            quantity=a['quantity'],
+            unit_price=a['unit_price']
+        )]
+
+        subscription = Subscription(
+            start_date=a['start_date'],
+            end_date=a['end_date'],
+            invoice_recurrence=a['invoice_recurrence'],
+            invoice_start_date=a['invoice_start_date'],
+            party=sale.party.id,
+            contact=sale.contact.id if sale.contact else None,
+            invoice_party=sale.invoice_party.id if sale.invoice_party else None,
+            invoice_address=sale.invoice_address.id,
+            payment_term=sale.payment_term.id if sale.payment_term else None,
+            lines=subscription_lines,
+            equipments=equipments_to_subscription,
+        )
+        subscription.save()
diff --git a/subscription.py b/subscription.py
index 1b3c5cb..d296792 100644
--- a/subscription.py
+++ b/subscription.py
@@ -10,27 +10,41 @@ class Subscription(metaclass=PoolMeta):
 
     equipments = fields.Many2Many('sale.subscription-optical_equipment.equipment', 'subscription',
                                   'equipment', "Equipments")
+
     @classmethod
     @ModelView.button
     @Workflow.transition('running')
     @set_employee('run_by')
     def run(cls, subscriptions):
         pool = Pool()
-        Equipment = pool.get('optical_equipment.equipment')
         Line = pool.get('sale.subscription.line')
         lines = []
-        subscription = subscriptions[0]
+
+        pool = Pool()
+        Equipments = pool.get('optical_equipment.equipment')
+
+        for subscription in subscriptions:
+            if not subscription.next_invoice_date:
+                subscription.next_invoice_date = (
+                    subscription.compute_next_invoice_date())
+            for line in subscription.lines:
+                if (line.next_consumption_date is None
+                        and not line.consumed_until):
+                    line.next_consumption_date = (
+                        line.compute_next_consumption_date())
+            lines.extend(subscription.lines)
+            for equipment in subscription.equipments:
+                equipment.state = "contrated"
+                equipment.save()
+        Line.save(lines)
+        cls.save(subscriptions)
         
 class SubscriptionEquipment(ModelSQL):
     'Optical Equipment - Subscription'
     __name__ = 'sale.subscription-optical_equipment.equipment'
     
     subscription = fields.Many2One('sale.subscription', 'Subscription', select=True)
-    equipment = fields.Many2One('optical_equipment.equipment', 'Equipment', select=True,
-                                domain=[('state', '=', 'registred'),
-                                        ('state', '=', 'uncontrated')],
-                                depends=['state'])
-
+    equipment = fields.Many2One('optical_equipment.equipment', 'Equipment', select=True)
     
 class Line(metaclass=PoolMeta):
     __name__ = 'sale.subscriptioin.line'
diff --git a/view/optical_equipment_form.xml b/view/optical_equipment_form.xml
index d8fd50d..c8f5959 100644
--- a/view/optical_equipment_form.xml
+++ b/view/optical_equipment_form.xml
@@ -57,8 +57,9 @@
 	    <field name="maintenance_history"/>
 	</page>
     </notebook>
+    
     <group col="2" colspan="2" id="button">
-	<button name="draft"/>
+	<!--<button name="draft"/> -->
 	<button name="registred"/>
     </group>
     <label name="state"/>