59 lines
2.1 KiB
Python
Executable File
59 lines
2.1 KiB
Python
Executable File
# 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 decimal import Decimal
|
|
|
|
from trytond.pool import Pool, PoolMeta
|
|
from trytond.transaction import Transaction
|
|
|
|
|
|
class Party(metaclass=PoolMeta):
|
|
__name__ = 'party.party'
|
|
|
|
@classmethod
|
|
def get_credit_amount(cls, parties, name):
|
|
pool = Pool()
|
|
Currency = pool.get('currency.currency')
|
|
Sale = pool.get('sale.sale')
|
|
Uom = pool.get('product.uom')
|
|
User = pool.get('res.user')
|
|
|
|
amounts = super(Party, cls).get_credit_amount(parties, name)
|
|
|
|
user = User(Transaction().user)
|
|
if not user.company:
|
|
return amounts
|
|
currency = user.company.currency
|
|
|
|
sales = Sale.search([
|
|
('company', '=', user.company.id),
|
|
('party', 'in', [p.id for p in parties]),
|
|
('state', 'in', ['confirmed', 'processing']),
|
|
])
|
|
for sale in sales:
|
|
amount = 0
|
|
for line in sale.lines:
|
|
quantity = line.credit_limit_quantity
|
|
if quantity is None:
|
|
continue
|
|
for invoice_line in line.invoice_lines:
|
|
if invoice_line.type != 'line':
|
|
continue
|
|
invoice = invoice_line.invoice
|
|
if invoice and invoice.move:
|
|
quantity -= Uom.compute_qty(
|
|
invoice_line.unit, invoice_line.quantity,
|
|
line.unit, round=False)
|
|
if quantity > 0:
|
|
amount += Currency.compute(
|
|
sale.currency,
|
|
Decimal(str(quantity)) * line.unit_price, currency,
|
|
round=False)
|
|
amounts[sale.party.id] = currency.round(
|
|
amounts[sale.party.id] + amount)
|
|
return amounts
|
|
|
|
@classmethod
|
|
def _credit_limit_to_lock(cls):
|
|
return super(Party, cls)._credit_limit_to_lock() + [
|
|
'sale.sale', 'sale.line']
|