feat: Se crea reporte de extracto de ventas

report_close_statement.py

@@ -0,0 +1,99 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+from trytond.model import ModelSQL, ModelView, fields
+from trytond.wizard import Wizard, StateView, Button, StateTransition, StateReport
+from trytond.transaction import Transaction
+from trytond.pool import Pool
+from trytond.report import Report
+from datetime import date
+
+__all__ = ['ReportCloseStatementStart', 'PrintReportCloseStatement', 'CashRegister'] 
+
+class ReportCloseStatementStart(ModelView):
+    "Vista inicial de reporte de extracto"
+    __name__ = 'sale.print_cash_register.start'
+
+    date = fields.Date('Date', required=True)
+    shop = fields.Many2One('sale.shop', 'Shop', required=True)
+
+    @classmethod
+    def default_date(cls):
+        return date.today()
+
+    @classmethod
+    def default_shop(cls):
+        context = Transaction().context
+        shop = context['shop']
+
+        if shop:
+            return shop
+
+
+class PrintReportCloseStatement(Wizard):
+    "Asistente para imprimir extracto"
+    __name__ = 'sale.print_cash_register'
+
+    start = StateView(
+        'sale.print_cash_register.start',
+        'sale_fast_food.print_cash_register_start_view_form',[
+            Button("Cancel", 'end', 'tryton-cancel'),
+            Button("Print Cash Register",
+             'print_cash_register', 'tryton-ok', default=True)])
+
+    print_cash_register = StateReport('sale.cash_register')
+
+    def do_print_cash_register(self, action):
+        data = {
+            'date': self.start.date,
+            'shop': self.start.shop.id
+        }
+
+        return action, data
+
+    def transition_print_cash_register(self):
+        return 'end'
+    
+class CashRegister(Report):
+    "Extracto"
+    __name__ = 'sale.cash_register'
+
+    def get_number_pizzas_sold(lines):
+        pizzas = 0
+        for line in lines:                
+            if line.bought_pizza:
+                pizzas += line.quantity
+        return pizzas
+
+    @classmethod
+    def get_context(cls, records, header, data):
+        report_context = super(
+            CashRegister, cls).get_context(records, header, data)
+
+        pool = Pool()
+        Sale = pool.get('sale.sale')
+        today_sales_domain = [('state', 'in', ['processing', 'done']),
+                              ('sale_date', '=', data['date']),
+                              ('shop', '=', data['shop'])]
+        today_sales = Sale.search(today_sales_domain, 
+                                  order=[('id', 'ASC')])
+
+        if not today_sales:
+            return
+
+        untaxed_amount = 0
+        tax_amount = 0
+        total_amount = 0
+        number_pizzas_sold = 0
+
+        for sale in today_sales:
+            untaxed_amount += sale.untaxed_amount
+            tax_amount += sale.tax_amount
+            number_pizzas_sold += cls.get_number_pizzas_sold(sale.lines)
+            total_amount+=sale.total_amount
+
+        report_context['untaxed_amount'] = untaxed_amount
+        report_context['tax_amount'] = tax_amount
+        report_context['total_amount'] = total_amount
+        report_context['number_pizzas_sold'] = number_pizzas_sold
+
+        return report_context