[12.0][ADD] Add script for update structure agreement

Author: newtratip

z_migration/controller/connection.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+"""
+For only connection. You can change config file name from keyword
+"Config file name" in comment line, ex: migration.conf.
+In config file has hostname, port, database, login, password, user_id and
+protocal.
+Prepare config file
+1. migration_remote.conf (For server production).
+2. migration.conf (For server localhost).
+"""
+import openerplib
+import ConfigParser
+import os
+
+# Config file name
+conf = 'migration.conf'
+
+
+def get_connection(config_file):
+    file_dir = os.path.dirname(os.path.realpath(__file__))
+    file_path = '%s/%s' % (file_dir, config_file)
+    config = ConfigParser.ConfigParser()
+    config.readfp(open(file_path))
+    hostname = config.get('server', 'hostname')
+    port = int(config.get('server', 'port'))
+    database = config.get('server', 'database')
+    login = config.get('server', 'login')
+    password = config.get('server', 'password')
+    user_id = int(config.get('server', 'user_id'))
+    protocol = config.get('server', 'protocol')
+    connection = openerplib.get_connection(
+        hostname=hostname, port=port, database=database, login=login,
+        password=password, protocol=protocol, user_id=user_id)
+    return connection
+
+
+connection = get_connection(conf)
+connection.check_login()

z_migration/controller/log.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+"""
+Create log files
+# Reference source code
+https://gist.githubusercontent.com/mcdeid/ccd7c00e9ae9a80e45297abb7b06ea1d/raw/8f8423e066228779b8129e8bca836dcde8bdcece/log.py
+"""
+import logging
+import os
+import time
+
+
+def setup_custom_logger(filename):
+    module_dir = \
+        os.path.dirname(os.path.realpath(__file__)).split('/controller')[0]
+    log_path = '%s/logs/%s_%s.log' \
+        % (module_dir, filename, time.strftime('%Y_%m_%d_%H_%M_%S'))
+    formatter = logging.Formatter(
+        fmt='%(asctime)s %(levelname)s %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S')
+    sthandler = logging.StreamHandler()
+    sthandler.setFormatter(formatter)
+    flhandler = logging.FileHandler(log_path)
+    flhandler.setFormatter(formatter)
+    logger = logging.getLogger('root')
+    logger.setLevel(logging.INFO)
+    if not logger.handlers:
+        logger.addHandler(sthandler)
+        logger.addHandler(flhandler)
+    else:
+        logger.info('Handlers already added to logger.')
+    logger.info('Log created at level: ' + str(logger.getEffectiveLevel()))
+    return logger

z_migration/controller/migration.conf

@@ -0,0 +1,8 @@
+[server]
+hostname = localhost
+port = 8069
+database = ACM_TEST2
+login = admin
+password = admin
+user_id = 1
+protocol = xmlrpc

z_migration/logs/empty.log

@@ -20,3 +20,50 @@ def fields_view_get(self, view_id=None, view_type='form',
         root.set('delete', 'true')
         res['arch'] = etree.tostring(root)
         return res
+
+    @api.multi
+    def mock_unlink_recital_structure(self):
+        self.mapped('recital_ids').unlink()
+        return True
+
+    @api.multi
+    def mock_unlink_section_structure(self):
+        self.mapped('sections_ids').unlink()
+        return True
+
+    @api.multi
+    def mock_unlink_clause_structure(self):
+        self.mapped('clauses_ids').unlink()
+        return True
+
+    @api.multi
+    def mock_unlink_appendix_structure(self):
+        self.mapped('appendix_ids').unlink()
+        return True
+
+    @api.multi
+    def mock_copy_recital_structure(self):
+        self.ensure_one()
+        for rec in self.template_id.recital_ids:
+            recital = rec.copy()
+            recital.agreement_id = self.id
+        return True
+
+    @api.multi
+    def mock_copy_section_structure(self):
+        self.ensure_one()
+        for rec in self.template_id.sections_ids:
+            section = rec.copy()
+            section.agreement_id = self.id
+            section.mapped('clauses_ids').write({
+                'agreement_id': self.id,
+            })
+        return True
+
+    @api.multi
+    def mock_copy_appendix_structure(self):
+        self.ensure_one()
+        for rec in self.template_id.appendix_ids:
+            appendix = rec.copy()
+            appendix.agreement_id = self.id
+        return True

z_migration/models/agreement.py

@@ -0,0 +1,55 @@
+"""
+This method will create structure (recitals, sections, clauses, appendices)
+in agreement
+"""
+import sys
+import os
+try:
+    agreement_path = os.path.dirname(os.path.realpath(__file__))
+    script_path = os.path.dirname(agreement_path)
+    migration_path = os.path.dirname(script_path)
+    controller_path = '%s/controller' % migration_path
+    sys.path.insert(0, controller_path)
+    from connection import connection
+    import log
+except Exception:
+    pass
+
+# Model
+Agreement = connection.get_model('agreement')
+
+# Domain
+dom = [('is_template', '=', False)]
+
+# Search Agreement
+agreements = Agreement.search_read(dom)
+
+log_agreement_ids = [[], []]
+logger = log.setup_custom_logger('update_structure')
+logger.info('Start process')
+logger.info('Total agreement: %s' % len(agreements))
+for agreement in agreements:
+    try:
+        # Make sure no have line in recital, section, clause, appendix
+        Agreement.mock_unlink_recital_structure([agreement['id']])
+        Agreement.mock_unlink_section_structure([agreement['id']])
+        Agreement.mock_unlink_clause_structure([agreement['id']])
+        Agreement.mock_unlink_appendix_structure([agreement['id']])
+        # Update structure
+        Agreement.mock_copy_recital_structure([agreement['id']])
+        Agreement.mock_copy_section_structure([agreement['id']])
+        Agreement.mock_copy_appendix_structure([agreement['id']])
+        # Write log
+        log_agreement_ids[0].append(agreement['id'])
+        logger.info('Pass ID: %s' % agreement['id'])
+    except Exception as ex:
+        log_agreement_ids[1].append(agreement['id'])
+        logger.error('Fail ID: %s (reason: %s)' % (agreement['id'], ex))
+summary = 'Summary: pass %s%s and fail %s%s' \
+          % (len(log_agreement_ids[0]),
+             log_agreement_ids[0] and ' %s' % str(tuple(log_agreement_ids[0]))
+             or '', len(log_agreement_ids[1]),
+             log_agreement_ids[1] and ' %s' % str(tuple(log_agreement_ids[1]))
+             or '')
+logger.info(summary)
+logger.info('End process')