first commit

modules/account/tests/tools.py

@@ -0,0 +1,151 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+import datetime
+
+from dateutil.relativedelta import relativedelta
+
+from proteus import Model, Wizard
+from proteus.config import get_config
+from trytond.modules.company.tests.tools import get_company
+
+__all__ = ['create_fiscalyear', 'create_chart', 'get_accounts',
+    'create_tax', 'create_tax_code']
+
+
+def create_fiscalyear(company=None, today=None, config=None):
+    "Create a fiscal year for the company on today or range"
+    FiscalYear = Model.get('account.fiscalyear', config=config)
+    Sequence = Model.get('ir.sequence.strict', config=config)
+    SequenceType = Model.get('ir.sequence.type', config=config)
+
+    if not company:
+        company = get_company(config=config)
+
+    if not today:
+        today = datetime.date.today()
+    if isinstance(today, datetime.date):
+        start_date = end_date = today
+    else:
+        start_date, end_date = today
+        today = start_date + (end_date - start_date) / 2
+    start_date = min(start_date, today - relativedelta(months=6, day=1))
+    end_date = max(end_date, today + relativedelta(months=5, day=31))
+    assert start_date <= end_date
+
+    fiscalyear = FiscalYear(name=str(today.year))
+    fiscalyear.start_date = start_date
+    fiscalyear.end_date = end_date
+    fiscalyear.company = company
+
+    sequence_type, = SequenceType.find(
+        [('name', '=', "Account Move")], limit=1)
+    move_sequence = Sequence(
+        name=str(today.year),
+        sequence_type=sequence_type,
+        company=company)
+    move_sequence.save()
+    fiscalyear.move_sequence = move_sequence
+    return fiscalyear
+
+
+def create_chart(
+        company=None, chart='account.account_template_root_en', config=None):
+    "Create chart of accounts"
+    if config is None:
+        config = get_config()
+    AccountTemplate = Model.get('account.account.template', config=config)
+    ModelData = Model.get('ir.model.data', config=config)
+
+    if not company:
+        company = get_company(config=config)
+
+    chart_id = ModelData.get_id(chart, config.context)
+
+    account_template = AccountTemplate(chart_id)
+
+    create_chart = Wizard('account.create_chart', config=config)
+    create_chart.execute('account')
+    create_chart.form.account_template = account_template
+    create_chart.form.company = company
+    create_chart.execute('create_account')
+
+    accounts = get_accounts(company, config=config)
+
+    if accounts['receivable'].party_required:
+        create_chart.form.account_receivable = accounts['receivable']
+    if accounts['payable'].party_required:
+        create_chart.form.account_payable = accounts['payable']
+    if hasattr(create_chart.form, 'category_account_expense'):
+        create_chart.form.category_account_expense = accounts['expense']
+    if hasattr(create_chart.form, 'category_account_revenue'):
+        create_chart.form.category_account_revenue = accounts['revenue']
+    create_chart.execute('create_properties')
+    return create_chart
+
+
+def get_accounts(company=None, config=None):
+    "Return accounts per kind"
+    Account = Model.get('account.account', config=config)
+
+    if not company:
+        company = get_company(config=config)
+
+    accounts = {}
+    for type in ['receivable', 'payable', 'revenue', 'expense']:
+        try:
+            accounts[type], = Account.find([
+                    ('type.%s' % type, '=', True),
+                    ('company', '=', company.id),
+                    ('closed', '!=', True),
+                    ], limit=1)
+        except ValueError:
+            pass
+    try:
+        accounts['cash'], = Account.find([
+                ('company', '=', company.id),
+                ('code', '=', '1.1.1'),
+                ], limit=1)
+    except ValueError:
+        pass
+    try:
+        accounts['tax'], = Account.find([
+                ('company', '=', company.id),
+                ('code', '=', '6.3.6'),
+                ], limit=1)
+    except ValueError:
+        pass
+    return accounts
+
+
+def create_tax(rate, company=None, config=None):
+    "Create a tax of rate"
+    Tax = Model.get('account.tax', config=config)
+
+    if not company:
+        company = get_company(config=config)
+
+    accounts = get_accounts(company, config=config)
+
+    tax = Tax()
+    tax.name = 'Tax %s' % rate
+    tax.description = tax.name
+    tax.type = 'percentage'
+    tax.rate = rate
+    tax.invoice_account = accounts['tax']
+    tax.credit_note_account = accounts['tax']
+    return tax
+
+
+def create_tax_code(
+        tax, amount='tax', type='invoice', operator='+', config=None):
+    "Create a tax code for the tax"
+    TaxCode = Model.get('account.tax.code', config=config)
+
+    tax_code = TaxCode(name="Tax Code %s" % tax.name)
+    tax_code.company = tax.company
+    line = tax_code.lines.new()
+    line.operator = '+'
+    line.tax = tax
+    line.amount = amount
+    line.type = type
+    return tax_code