changes

fusion_accounting/services/tools/hst_management.py

@@ -0,0 +1,171 @@
+import logging
+
+_logger = logging.getLogger(__name__)
+
+
+def calculate_hst_balance(env, params):
+    date_from = params.get('date_from')
+    date_to = params.get('date_to')
+    base_domain = [
+        ('parent_state', '=', 'posted'),
+        ('company_id', '=', env.company.id),
+    ]
+    if date_from:
+        base_domain.append(('date', '>=', date_from))
+    if date_to:
+        base_domain.append(('date', '<=', date_to))
+
+    collected_accounts = env['account.account'].search([
+        ('code', '=like', '2005%'), ('company_id', '=', env.company.id),
+    ])
+    itc_accounts = env['account.account'].search([
+        ('code', '=like', '2006%'), ('company_id', '=', env.company.id),
+    ])
+
+    collected_lines = env['account.move.line'].search(
+        base_domain + [('account_id', 'in', collected_accounts.ids)]
+    )
+    itc_lines = env['account.move.line'].search(
+        base_domain + [('account_id', 'in', itc_accounts.ids)]
+    )
+
+    hst_collected = abs(sum(l.balance for l in collected_lines))
+    itcs = abs(sum(l.balance for l in itc_lines))
+
+    return {
+        'hst_collected': hst_collected,
+        'input_tax_credits': itcs,
+        'net_hst': hst_collected - itcs,
+        'status': 'owing' if (hst_collected - itcs) > 0 else 'refund',
+        'period': f'{date_from or "all"} to {date_to or "now"}',
+    }
+
+
+def get_tax_report(env, params):
+    report_ref = params.get('report_ref', 'account.generic_tax_report')
+    try:
+        report = env.ref(report_ref)
+    except Exception:
+        return {'error': f'Report not found: {report_ref}'}
+    options = report.get_options({
+        'date': {
+            'date_from': params.get('date_from', ''),
+            'date_to': params.get('date_to', ''),
+        }
+    })
+    lines = report._get_lines(options)
+    return {
+        'report_name': report.name,
+        'lines': [{
+            'name': l.get('name', ''),
+            'columns': [c.get('no_format', c.get('name', '')) for c in l.get('columns', [])],
+        } for l in lines[:50]],
+    }
+
+
+def find_missing_tax_invoices(env, params):
+    date_from = params.get('date_from')
+    date_to = params.get('date_to')
+    domain = [
+        ('move_type', 'in', ('out_invoice', 'out_refund')),
+        ('state', '=', 'posted'),
+        ('company_id', '=', env.company.id),
+    ]
+    if date_from:
+        domain.append(('date', '>=', date_from))
+    if date_to:
+        domain.append(('date', '<=', date_to))
+
+    invoices = env['account.move'].search(domain)
+    missing = invoices.filtered(
+        lambda inv: not any(line.tax_ids for line in inv.invoice_line_ids)
+    )
+    return {
+        'total_invoices': len(invoices),
+        'missing_tax_count': len(missing),
+        'invoices': [{
+            'id': inv.id,
+            'name': inv.name,
+            'partner': inv.partner_id.name if inv.partner_id else '',
+            'amount_total': inv.amount_total,
+            'date': str(inv.date),
+        } for inv in missing[:30]],
+    }
+
+
+def find_missing_itc_bills(env, params):
+    date_from = params.get('date_from')
+    date_to = params.get('date_to')
+    domain = [
+        ('move_type', 'in', ('in_invoice', 'in_refund')),
+        ('state', '=', 'posted'),
+        ('company_id', '=', env.company.id),
+    ]
+    if date_from:
+        domain.append(('date', '>=', date_from))
+    if date_to:
+        domain.append(('date', '<=', date_to))
+
+    bills = env['account.move'].search(domain)
+    missing = bills.filtered(
+        lambda b: not any(line.tax_ids for line in b.invoice_line_ids)
+    )
+    return {
+        'total_bills': len(bills),
+        'missing_itc_count': len(missing),
+        'bills': [{
+            'id': b.id,
+            'name': b.name,
+            'partner': b.partner_id.name if b.partner_id else '',
+            'amount_total': b.amount_total,
+            'date': str(b.date),
+        } for b in missing[:30]],
+    }
+
+
+def get_tax_return_status(env, params):
+    returns = env['account.return'].search([
+        ('company_id', '=', env.company.id),
+    ], order='date_start desc', limit=10)
+    return {
+        'returns': [{
+            'id': r.id,
+            'name': r.display_name,
+            'date_start': str(r.date_start) if hasattr(r, 'date_start') else '',
+            'date_end': str(r.date_end) if hasattr(r, 'date_end') else '',
+            'state': r.state if hasattr(r, 'state') else '',
+        } for r in returns],
+    }
+
+
+def generate_tax_return(env, params):
+    try:
+        env['account.return']._generate_or_refresh_all_returns(
+            company=env.company
+        )
+        return {'status': 'generated', 'message': 'Tax returns refreshed successfully.'}
+    except Exception as e:
+        return {'error': str(e)}
+
+
+def validate_tax_return(env, params):
+    return_id = int(params['return_id'])
+    tax_return = env['account.return'].browse(return_id)
+    if not tax_return.exists():
+        return {'error': 'Tax return not found'}
+    try:
+        tax_return.action_validate()
+        return {'status': 'validated', 'return_id': return_id}
+    except Exception as e:
+        return {'error': str(e)}
+
+
+TOOLS = {
+    'calculate_hst_balance': calculate_hst_balance,
+    'get_tax_report': get_tax_report,
+    'find_missing_tax_invoices': find_missing_tax_invoices,
+    'find_missing_itc_bills': find_missing_itc_bills,
+    'get_tax_return_status': get_tax_return_status,
+    'generate_tax_return': generate_tax_return,
+    'validate_tax_return': validate_tax_return,
+}