trytonpsk-sale_pos/statement.py

# This file is part of the sale_pos module for Tryton.
# The COPYRIGHT file at the top level of this repository contains the full
# copyright notices and license terms.
from datetime import date, datetime
from decimal import Decimal
from dateutil import tz

from trytond.model import fields, ModelView, ModelSQL, Workflow
from trytond.pool import Pool, PoolMeta
from trytond.transaction import Transaction
from trytond.wizard import Button, StateTransition, StateView, Wizard
from trytond.pyson import Eval
from trytond.i18n import gettext
from .exceptions import StatementValidationError
from trytond.exceptions import UserError

_ZERO = Decimal("0.0")

MONEY = [100000, 50000, 20000, 10000, 5000, 2000, 1000, 500, 200, 100, 50]


class Journal(metaclass=PoolMeta):
    __name__ = 'account.statement.journal'
    # IS THIS NECCESARY?
    devices = fields.One2Many('sale.device', 'journal', 'Devices')


class Statement(metaclass=PoolMeta):
    __name__ = 'account.statement'
    users = fields.Function(fields.One2Many(
        'res.user', None, 'Users'),
        'get_users', setter='set_users', searcher='search_users')
    turn = fields.Integer('Turn', states={'readonly': True})
    count_money = fields.One2Many(
        'sale_pos.money_count', 'statement', 'Count Money')
    sale_device = fields.Many2One('sale.device', 'Device', required=False)
    total_money = fields.Function(
        fields.Numeric("Total Money"), 'get_total_money')
    amount_difference = fields.Function(fields.Numeric("Amount Difference"),
            'get_amount_difference')
    expenses_daily = fields.One2Many('sale_pos.expenses_daily', 'statement',
            'Expenses Daily')
    additional_income = fields.One2Many('sale_pos.additional_income', 'statement',
            'Additional Income')
    balance_next_shift = fields.Numeric('balance next shift', digits=(10, 2))
    salesman = fields.Many2One('company.employee', 'Salesman',  states={
        'readonly': Eval('state').in_(['validated', 'cancel', 'posted']),
    })

    @classmethod
    def __setup__(cls):
        super(Statement, cls).__setup__()
        cls._transitions |= set((
            ('posted', 'draft'),
        ))
        cls._buttons.update({
            'force_draft': {
                'invisible': Eval('state') != 'posted',
                'depends': ['state'],
            },
        })

    @classmethod
    def cron_autopost(cls):
        # Add automatic post for validate statement
        statements = cls.search([
            ('state', '=', 'validated'),
        ], order=[('date', 'ASC')])
        if statements:
            cls.post([statements[0]])

    def get_total_money(self, name):
        res = 0
        for line in self.count_money:
            res = res + line.amount
        if res:
            return res

    def get_amount_difference(self, name):
        res = 0
        amount = sum(li.amount for li in self.lines)
        if self.total_money is not None:
            res = self.total_money - amount
            return res

    @classmethod
    def get_users(cls, statements, names):
        return {'users':
                {s.id: [
                    u.id
                    for j in s.journal
                    for d in j.devices
                    for u in d.users
                ]} for s in statements}

    @classmethod
    def search_users(cls, name, clause):
        pool = Pool()
        Journal = pool.get('account.statement.journal')
        Device = pool.get('sale.device')
        DeviceJournal = pool.get('sale.device.account.statement.journal')
        User = pool.get('res.user')

        statement = cls.__table__()
        journal = Journal.__table__()
        device = Device.__table__()
        device_journal = DeviceJournal.__table__()
        user = User.__table__()

        query = statement.join(
                journal, condition=statement.journal == journal.id).join(
                device_journal,
                    condition=journal.id == device_journal.journal).join(
                device, condition=device_journal.device == device.id).join(
                user, condition=device.id == user.sale_device).select(
                statement.id,
                where=user.id == clause[2])
        return [('id', 'in', query)]

    # @classmethod
    # def trigger_create(cls, records):
    #     super(Statement, cls).trigger_create(records)
    #     pool = Pool()
    #     MoneyCount = pool.get('sale_pos.money_count')
    #     BillMoney = pool.get('sale_pos.bill_money')
    #     bills = BillMoney.search([])
    #     money_to_create = []
    #     for statement in records:
    #         for bill in bills:
    #             money_to_create.append({
    #                 'statement': statement.id,
    #                 'bill': bill.value,
    #                 'quantity': 0,
    #             })
    #     MoneyCount.create(money_to_create)

    @classmethod
    @ModelView.button
    @Workflow.transition('draft')
    def force_draft(cls, statements):
        pass

    @classmethod
    def create_move(cls, statements):
        '''Create move for the statements and try to reconcile the lines.
        Returns the list of move, statement and lines
        '''
        pool = Pool()
        Line = pool.get('account.statement.line')
        Move = pool.get('account.move')
        Invoice = pool.get('account.invoice')

        to_write = []
        moves = []
        for statement in statements:
            for line in statement.lines:
                if line.move:
                    continue
                if line.invoice and line.invoice.state in (
                        'draft', 'validated'):
                    if not line.invoice.invoice_date:
                        for sale in line.invoice.sales:
                            line.invoice.write(
                                [line.invoice], {
                                    'invoice_date': sale.sale_date}
                            )
                    Invoice.post([line.invoice])
                move = line._get_move2()
                Move.post([move])
                to_write.append([line])
                to_write.append({'move': move.id})

                moves.append(move)
                for mline in move.lines:
                    if mline.account == line.account:
                        line.reconcile([(mline, line)])
        if to_write:
            Line.write(*to_write)
        return moves

    @classmethod
    @ModelView.button
    @Workflow.transition('validated')
    def validate_statement(cls, statements):
        for statement in statements:
            getattr(statement, 'validate_%s' % statement.validation)()

        cls.create_move(statements)
        cls.write(statements, {
            'state': 'validated',
        })

    @classmethod
    def delete(cls, statements):
        for statement in statements:
            if statement.lines:
                raise StatementValidationError(
                    gettext('sale_pos.msg_cant_delete_statement', s=statement.rec_name))
            else:
                super(Statement, cls).delete(statements)


class StatementLine(metaclass=PoolMeta):
    __name__ = 'account.statement.line'
    extra_amount = fields.Numeric('Extra', digits=(16, 2))
    net_amount = fields.Function(fields.Numeric('Net Amount', digits=(16, 2)),
        'get_net_amount')

    def get_net_amount(self, name=None):
        res = 0
        if self.extra_amount and self.amount:
            res = self.extra_amount + self.amount
        return res

    @staticmethod
    def default_extra_amount():
        return 0
    # All this was moved to Sale Shop
    # def get_move_line2(self, values):
    #     'Return counterpart Move Line for the amount'
    #     move_id = values['move_id']
    #     journal_account = values.get('journal_account', False)
    #     account = values.get('account')
    #     amount = values.get('amount')
    #     credit = _ZERO
    #     debit = _ZERO
    #     if journal_account:
    #         if amount >= 0:
    #             debit = amount
    #         else:
    #             credit = abs(amount)
    #     else:
    #         if amount >= 0:
    #             credit = amount
    #         else:
    #             debit = abs(amount)
    #
    #     if not account:
    #         raise StatementValidationError(
    #             gettext('sale_pos.msg_account_statement_journal', s=self.journal.rec_name))
    #     res = {
    #         'description': self.description,
    #         'debit': debit,
    #         'credit': credit,
    #         'account': account.id,
    #         'party': self.party.id if account.party_required else None,
    #         'move': move_id,
    #     }
    #     return res
    #
    # def _get_move2(self):
    #     # Return Move for the grouping key
    #     pool = Pool()
    #     Move = pool.get('account.move')
    #     MoveLine = pool.get('account.move.line')
    #     Period = pool.get('account.period')
    #     Journal = pool.get('account.statement.journal')
    #
    #     company_id = self.statement.company.id
    #     period_id = Period.find(company_id, date=self.date)
    #     _move = {
    #         'period': period_id,
    #         'journal': self.statement.journal.journal.id,
    #         'date': self.date,
    #         'origin': str(self.statement),
    #         'company': company_id,
    #         'state': 'draft',
    #     }
    #
    #     move, = Move.create([_move])
    #
    #     move_line1 = self.get_move_line2({
    #         'move_id': move.id,
    #         'account': self.account,
    #         'amount': self.amount
    #     })
    #     extra_amount = 0
    #     if self.extra_amount:
    #         extra_amount = self.extra_amount
    #
    #     move_line2 = self.get_move_line2({
    #         'account': self.statement.journal.account,
    #         'move_id': move.id,
    #         'amount': self.amount + extra_amount,
    #         'journal_account': True,
    #     })
    #     if self.statement.journal.require_party and self.statement.journal.party:
    #         party = self.statement.journal.party.id
    #         move_line2['party'] = party
    #     to_create = [move_line1, move_line2]
    #
    #     journals = Journal.search([
    #         ('kind', '=', 'cash')
    #     ])
    #     journal_ids = [j.id for j in journals]
    #
    #     # Just add extra amount to move line when payment is different to cash
    #     if self.extra_amount and self.extra_amount > 0 and \
    #             self.statement.journal.id not in journal_ids:
    #         if journals:
    #             journal = journals[0]
    #             move_line3 = self.get_move_line2({
    #                 'account': journal.account,
    #                 'move_id': move.id,
    #                 'amount': self.extra_amount
    #             })
    #             to_create.append(move_line3)
    #     MoveLine.create(to_create)
    #     return move


class AdditionalIncome(ModelSQL, ModelView):
    " Additional Incomes "
    __name__ = 'sale_pos.additional_income'

    statement = fields.Many2One('account.statement', 'Statement')
    description = fields.Char('Description')
    party = fields.Many2One('party.party', 'Party')
    amount = fields.Numeric('Amount', digits=(16, 2), states={
        'required': True})


class OpenStatementStart(ModelView):
    'Open Statement'
    __name__ = 'open.statement.start'


class OpenStatementDone(ModelView):
    'Open Statement'
    __name__ = 'open.statement.done'
    result = fields.Text('Result', readonly=True, translate=True)


class OpenStatement(Wizard):
    'Open Statement'
    __name__ = 'open.statement'
    start = StateView(
        'open.statement.start',
        'sale_pos.open_statement_start', [
            Button('Cancel', 'end', 'tryton-cancel'),
            Button('Ok', 'create_', 'tryton-ok', default=True),
        ])
    create_ = StateTransition()
    done = StateView(
        'open.statement.done',
        'sale_pos.open_statement_done', [
            Button('Done', 'end', 'tryton-ok', default=True),
        ])

    @classmethod
    def __setup__(cls):
        super(OpenStatement, cls).__setup__()

    def default_done(self, fields):
        return {
            'result': self.result,
        }

    def _check_previous(self, sale_device, today):
        pool = Pool()
        Statement = pool.get('account.statement')
        res = Statement.search_read([
            ('sale_device', '=', sale_device.id),
            ('state', '=', 'draft'),
            ], fields_names=['id'])
        return res

    def transition_create_(self):
        pool = Pool()
        User = pool.get('res.user')
        Statement = pool.get('account.statement')
        CountMoney = pool.get('sale_pos.money_count')
        Sale = pool.get('sale.sale')
        Invoice = pool.get('account.invoice')
        Date = pool.get('ir.date')
        Company = pool.get('company.company')
        company = Company(Transaction().context.get('company'))
        today = Date.today()
        user = Transaction().user
        user = User(user)
        device = user.sale_device
        time_zone = company.timezone
        now_ = datetime.now(tz.gettz(time_zone))
        today = date(now_.year, now_.month, now_.day)
        previous = self._check_previous(device, today)
        print("res............", previous)
        if previous:
            raise UserError(
                gettext('sale_pos.msg_previous_statement_open'))
            return 'end'
        if device:
            journals = [j.id for j in device.journals]
            statements = Statement.search([
                    ('journal', 'in', journals),
                    ('sale_device', '=', device.id),
                ], order=[
                ('date', 'ASC'),
            ])
            journals_of_draft_statements = [
                s.journal for s in statements if s.state == 'draft'
            ]

            vlist = []
            result = ''

            for journal in device.journals:
                statements_today = Statement.search([
                    ('journal', '=', journal.id),
                    ('date', '=', today),
                    ('sale_device', '=', device.id),
                ])
                turn = len(statements_today) + 1
                if journal not in journals_of_draft_statements:
                    values = {
                        'name': '%s - %s' % (device.rec_name, journal.rec_name),
                        'journal': journal.id,
                        'company': user.company.id,
                        'start_balance': journal.default_start_balance or Decimal('0.0'),
                        'end_balance': Decimal('0.0'),
                        'turn': turn,
                        'sale_device': device.id,
                    }
                    vlist.append(values)
                    # result += 'sale_shop.msg_open_statement %(s)' % journal.rec_name
                    result += f'Estado de Cuenta {journal.rec_name} abierto.\n \n'
                else:
                    result += f'El Estado de Cuenta {journal.rec_name} ya estaba abierto.\n \n'

            created_sts = Statement.create(vlist)
            for st in created_sts:
                if st.journal.kind == 'cash':
                    turn = st.turn
                    sales = Sale.search([
                        ('sale_device', '=', device.id),
                        ('sale_date', '=', today),
                        ('payment_term.payment_type', '=', '1'),
                        ('payments', '=', None)
                    ])
                    invoices = [s.invoices[0] for s in sales if s.invoices]
                    write_ = {'turn': turn}
                    Sale.write(sales, write_)
                    Invoice.write(invoices, write_)
                    to_create = []
                    for m in MONEY:
                        to_create.append({
                            'bill': m,
                            'quantity': 0,
                            'statement': st.id,
                        })
                    CountMoney.create(to_create)
            self.result = result
        else:
            self.result = f'El usuario {user.rec_name} no tiene asginado un terminal.\n \n'
        return 'done'


class CloseStatementStart(ModelView):
    'Close Statement'
    __name__ = 'close.statement.start'


class CloseStatementDone(ModelView):
    'Close Statement'
    __name__ = 'close.statement.done'
    result = fields.Text('Result', readonly=True)


class CloseStatement(Wizard):
    'Close Statement'
    __name__ = 'close.statement'
    start = StateView(
        'close.statement.start',
        'sale_pos.close_statement_start', [
            Button('Cancel', 'end', 'tryton-cancel'),
            Button('Ok', 'validate', 'tryton-ok', default=True),
        ])
    validate = StateTransition()
    done = StateView(
        'close.statement.done',
        'sale_pos.close_statement_done', [
            Button('Done', 'end', 'tryton-ok', default=True),
        ])

    @classmethod
    def __setup__(cls):
        super(CloseStatement, cls).__setup__()

    def default_done(self, fields):
        return {
            'result': self.result,
        }

    def transition_validate(self):
        pool = Pool()
        User = pool.get('res.user')
        Statement = pool.get('account.statement')

        user = Transaction().user
        user = User(user)
        device = user.sale_device
        if device:
            journals = [j.id for j in device.journals]
            draft_statements = {
                s.journal: s for s in Statement.search([
                        ('journal', 'in', journals),
                        ('sale_device', '=', device.id),
                    ], order=[
                    ('create_date', 'ASC'),
                ])}

            result = ''
            statements = []
            for journal in device.journals:
                statement = draft_statements.get(journal)
                if statement and statement.state == 'draft':
                    end_balance = statement.start_balance
                    for line in statement.lines:
                        end_balance += line.amount
                    statement.end_balance = end_balance
                    statement.save()
                    statements.append(statement)
                    result += f'Estado de Cuenta {journal.rec_name} cerrado.\n \n'
                elif statement:
                    result += f'El Estado de Cuenta {journal.rec_name} ya estaba cerrado.\n \n'
                else:
                    result += f'El Estado de Cuenta {journal.rec_name} no encontrado.\n \n'
            if statements:
                Statement.validate_statement(statements)
            self.result = result
        else:
            self.result = f'El usuario {user.rec_name} no tiene asginado un terminal.\n \n'
        return 'done'


class MoneyCount(ModelSQL, ModelView):
    'Money Count'
    __name__ = 'sale_pos.money_count'
    statement = fields.Many2One('account.statement', 'Statement',
                                required=True)
    bill = fields.Integer('Bill', required=True)
    quantity = fields.Integer('Quantity')
    amount = fields.Function(fields.Numeric('Amount', digits=(16, 2)),
                             'get_amount')

    @staticmethod
    def default_quantity():
        return 0

    def get_amount(self, name=None):
        res = Decimal(0)
        if self.quantity and self.bill:
            res = Decimal(self.quantity * self.bill)
        return res

    @classmethod
    def __setup__(cls):
        super(MoneyCount, cls).__setup__()
        cls._order.insert(0, ('bill', 'DESC'))


class ExpensesDaily(ModelSQL, ModelView):
    'Expenses Daily'
    __name__ = 'sale_pos.expenses_daily'
    _rec_name = 'invoice_number'
    statement = fields.Many2One(
        'account.statement', 'Statement', required=True)
    invoice_number = fields.Char('Invoice Number', required=True)
    description = fields.Char('Description', required=True)
    amount = fields.Numeric('Amount', digits=(16, 2), required=True)
    party = fields.Many2One('party.party', 'Party')
    reference = fields.Char('Reference')


class BillMoney(ModelSQL, ModelView):
    'Bill Money'
    __name__ = 'sale_pos.bill_money'
    value = fields.Integer('Value', required=True)

    @classmethod
    def __setup__(cls):
        super(BillMoney, cls).__setup__()