Add upgrade account chart and account codes from template

before/upgrade_account_chart.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python
+import sys
+import os
+
+dbname = sys.argv[1]
+config_file = sys.argv[2]
+from trytond.config import config as CONFIG
+CONFIG.update_etc(config_file)
+
+from trytond.transaction import Transaction
+from trytond.pool import Pool
+import trytond.tools as tools
+import logging
+
+Pool.start()
+pool = Pool(dbname)
+pool.init()
+
+context = {}
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+ch = logging.StreamHandler(sys.stdout)
+ch.setLevel(logging.DEBUG)
+formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+ch.setFormatter(formatter)
+logger.addHandler(ch)
+
+with Transaction().start(dbname, 1, context=context):
+    Company = pool.get('company.company')
+    ModelData = pool.get('ir.model.data')
+    Configuration = pool.get('account.configuration')
+    Account = pool.get('account.account')
+    AccountTemplate = pool.get('account.account.template')
+
+    cursor = Transaction().cursor
+    UpdateChart = pool.get('account.update_chart', type='wizard')
+
+    for company in Company.search([]):
+        with Transaction().set_context(company=company.id):
+            template = AccountTemplate(ModelData.get_id('account_es', 'pgc_0'))
+            account, = Account.search([('template', '=', template)], limit=1)
+            config = Configuration(1)
+
+            session_id, _, _ = UpdateChart.create()
+            update_chart = UpdateChart(session_id)
+            update_chart.start.account = account
+            update_chart.start.account_code_digits = config.default_account_code_digits
+            logger.info('%s: Upgrading Account Chart' % (company.rec_name))
+            update_chart.transition_update()
+            logger.info('%s: End Account Chart' % (company.rec_name))
+
+    Transaction().cursor.commit()
+
+    logger.info('Done')

before/upgrade_account_codes_from_template.py

@@ -0,0 +1,73 @@
+#!/usr/bin/env python
+import sys
+import os
+
+dbname = sys.argv[1]
+config_file = sys.argv[2]
+from trytond.config import config as CONFIG
+CONFIG.update_etc(config_file)
+
+from trytond.transaction import Transaction
+from trytond.pool import Pool
+import trytond.tools as tools
+import logging
+
+Pool.start()
+pool = Pool(dbname)
+pool.init()
+
+context = {}
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+ch = logging.StreamHandler(sys.stdout)
+ch.setLevel(logging.DEBUG)
+formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+ch.setFormatter(formatter)
+logger.addHandler(ch)
+
+with Transaction().start(dbname, 1, context=context):
+    Company = pool.get('company.company')
+    Configuration = pool.get('account.configuration')
+    Account = pool.get('account.account')
+    Template = pool.get('account.account.template')
+
+    cursor = Transaction().cursor
+
+    for company in Company.search([]):
+        with Transaction().set_context(company=company.id):
+            config = Configuration(1)
+
+            to_write = []
+            with Transaction().set_context(active_test=False):
+                for template in Template.search([('kind', '!=', 'view'), ('code', '!=', None)]):
+                    digits = config.default_account_code_digits
+                    digits = int(digits - len(template.code))
+                    code = template.code
+
+                    if '%' in template.code:
+                        new_code = code.replace('%', '0' * (digits + 1))
+                    else:
+                        new_code = code + '0' * digits
+
+                    accounts = Account.search([
+                        ('code', '=', new_code),
+                        ('template', '=', None),
+                        ('kind', '=', template.kind),
+                        ], limit=1)
+
+                    if accounts:
+                        account, = accounts
+                        to_write.extend(([account], {
+                            'template': template,
+                            }))
+                        logger.info('%s: %s : ID %s' % (company.rec_name, new_code, account.id))
+
+            logger.info('%s: Upgrading Account from Template' % (company.rec_name))
+
+            if to_write:
+                Account.write(*to_write)
+
+    Transaction().cursor.commit()
+
+    logger.info('Done')