odoo-2-flectra-converter/migrator/migrator.py

# -*- coding: utf-8 -*-
# !/usr/bin/python
import ast
import logging
import os
import re
import shutil
from os.path import join as oj
from shutil import copytree

from jinja2 import FileSystemLoader, Environment

_logger = logging.getLogger(__name__)


class RepoMigrator(object):

    def __init__(self, name, src_path, dst_path, company, contributor):
        self._name = name
        self.src_path = src_path
        self.dst_path = dst_path
        self._company = company
        self._contributor = contributor
        template_loader = FileSystemLoader(searchpath=oj(os.path.dirname(__file__), 'templates'))
        self._template_env = Environment(loader=template_loader)
        self._migrated_modules = False

    def migrate(self):
        if not os.path.isdir(self.src_path):
            return

        _logger.info('Starting migration of repository %s', self._name)
        self._migrated_modules = self._migrate_modules()

        # copy requirements.txt if exists
        self._has_requirements = False
        if os.path.exists(oj(self.src_path, 'requirements.txt')):
            shutil.copy(oj(self.src_path, 'requirements.txt'), oj(self.dst_path, 'requirements.txt'))
            self._has_requirements = True

        return self._migrated_modules


    def _migrate_modules(self):
        migrated_modules = {}
        for file_name in os.listdir(self.src_path):
            if file_name[:1] == '.':
                continue
            src_path = os.path.join(os.path.abspath(self.src_path), file_name)
            if os.path.isdir(src_path) and file_name not in ['setup', 'docs', 'doc', 'client_tools']:
                dst_path = os.path.join(os.path.abspath(self.dst_path), file_name)
                if os.path.exists(dst_path):
                    shutil.rmtree(dst_path)
                module = ModuleMigrator(src_path, dst_path, '2.0', self._company, self._contributor).migrate()
                migrated_modules = {**migrated_modules, **module}
        return migrated_modules

    def cleanup(self):
        shutil.rmtree(self.src_path)
        shutil.rmtree(self.dst_path)


class ModuleMigrator(object):
    _readme_intro = u''

    _replacements = {
        'odoo': 'flectra',
        'Odoo': 'Flectra',
        'ODOO': 'FLECTRA',
        '8069': '7073',
        'Part of Flectra.': 'Part of Odoo, Flectra.',
        'openerp': 'flectra',
        'Openerp': 'Flectra',
        'OpenERP': 'Flectra',
        'OpenErp': 'Flectra',
        'OPENERP': 'FLECTRA',
        'Part of Flectra.': 'Part of Openerp, Flectra.',
        'Part of Flectra.': 'Part of OpenERP, Flectra.',
    }

    _xml_replacements = {
        'odoo': 'flectra',
        'Odoo': 'Flectra',
        'ODOO': 'FLECTRA',
        'Part of Odoo.': 'Part of Odoo, Flectra.',
        'openerp': 'flectra',
        'Openerp': 'Flectra',
        'OpenERP': 'Flectra',
        'OpenErp': 'Flectra',
        'OPENERP': 'FLECTRA',
        'Part of Flectra.': 'Part of Openerp, Flectra.',
        'Part of Flectra.': 'Part of OpenERP, Flectra.',
    }

    _init_replacements = {
        'Odoo.': 'Odoo, Flectra.',
        'odoo': 'flectra',
        'openerp': 'flectra',
        'Openerp': 'Flectra',
        'OpenERP': 'Flectra',
        'OpenErp': 'Flectra',
    }

    _manifest_replacements = {
        'Odoo.': 'Odoo, Flectra.',
        'odoo': 'flectra',
        'openerp': 'flectra',
        'Openerp': 'Flectra',
        'OpenERP': 'Flectra',
        'OpenErp': 'Flectra',
    }

    _readme_regex_patterns = [
        '(.. image:: https://odoo-community.org.*?)^(Bug)',
        '(Bug Tracker.*)^(Credits)',
        '(Maintainer.*)',
        '(.. \|badge.*\|)',
    ]

    _obsolete_line_patterns = [
        '-*- coding: utf-8 -*-',
    ]
    _copyright_line_patterns = [
        'copyright',
        'Copyright',
        '©',
    ]

    _ingnore_dir = [
        'cla',
        'doc',
    ]

    _delete_dir = [
        'readme',
    ]

    _ingnore_files = [
        'LICENSE',
        'COPYRIGHT',
        'README.md',
        'CONTRIBUTING.md',
        'Makefile',
        'MANIFEST.in'
    ]

    _ingnore_py_words = [
        'OpenERPSession'
    ]

    _ignore_xml_words = [
        'provider_openerp'
    ]

    _website_replacements = {
        'https://www.odoo.com': 'https://flectrahq.com',
        'www.odoo.com': 'https://flectrahq.com',
        'https://www.openerp.com': 'https://flectrahq.com',
        'www.opernerp.com': 'https://flectrahq.com',
    }

    _replace_email = {
        'info@odoo.com': 'info@flectrahq.com',
        'info@openerp.com': 'info@flectrahq.com',
    }
    _copyright_list = []

    def __init__(self, src, destination, flectra_release, company, contributor):
        self._src = src
        self._destination = destination
        self._module_name = os.path.basename(destination)
        self._repo_name = os.path.basename(os.path.dirname(self._destination))
        self._release = flectra_release.split('.')[:2]
        self._company = company
        self._contributor = contributor
        self._copyright_list = []
        self._manifest = False

        self._contributor_part = u'''Contributors
        ------------

        * %s''' % self._contributor

    def migrate(self):
        _logger.info('Starting migration of module %s', self._module_name)
        copytree(self._src, self._destination, symlinks=True, ignore_dangling_symlinks=True)
        for base, odoo_dirs, odoo_files in os.walk(self._destination, topdown=True):
            odoo_files = [f for f in odoo_files if not f[0] == '.']
            odoo_dirs[:] = [d for d in odoo_dirs if not d[0] == '.']
            self._rename_files(base, odoo_files)
            self._rename_dir(base, odoo_dirs)
        self._render_templates()
        return {self._module_name: self._manifest}

    def _render_templates(self):
        # Module icon
        desc_path = oj(self._destination, 'static', 'description')
        os.makedirs(desc_path, exist_ok=True)
        if not os.path.exists(oj(desc_path, 'icon.png')) and \
                not os.path.exists(oj(desc_path, 'icon.jpg')):
            shutil.copy(oj(os.path.dirname(__file__), 'templates', 'icon.jpg'), desc_path)

    def _rename_files(self, base, items):
        for name in items:
            if name in self._ingnore_files:
                continue

            if name == '__openerp__.py':
                os.rename(oj(base, name), oj(base, '__manifest__.py'))
                name = '__manifest__.py'
            if name == '__init__.py':
                self._init_files(base)
                self._remove_obsolete_lines(base, name)
            elif name == '__manifest__.py':
                self._manifest_files(base)
                self._remove_obsolete_lines(base, name)
            elif name.lower() == 'readme.rst':
                self._readme_files(base, name)
            else:
                sp_name = name.split('.')
                if len(sp_name) >= 2:
                    if sp_name[-1] in ['xml', 'csv', 'json', 'html']:
                        self._remove_obsolete_lines(base, name)
                        self._xml_csv_json_files(base, name)
                    elif sp_name[-1] in ['py', 'css', 'less', 'js', 'yml']:
                        self._python_files(base, name)
                        self._remove_obsolete_lines(base, name)
            try:
                for old, new in self._replacements.items():
                    if name != (name.replace(old, new)):
                        os.rename(oj(base, name), oj(base, name.replace(old, new)))
            except OSError:
                pass

    def _rename_dir(self, base, items):
        for folder in items:
            if folder in self._delete_dir:
                shutil.rmtree(oj(base, folder))
                continue
            if folder in self._ingnore_dir:
                continue
            for new_base, odoo_dirs, odoo_files in os.walk(oj(base, folder), topdown=True):
                if odoo_files:
                    self._rename_files(new_base, odoo_files)
                if odoo_dirs:
                    self._rename_dir(new_base, odoo_dirs)
            if 'odoo' in folder:
                os.rename(oj(base, folder), oj(base, folder.replace('odoo', 'flectra')))

    def _init_files(self, base):
        infile = self._open_read(base, '__init__.py')
        out = self._open_write(base, '__init__.py')
        for old, new in self._init_replacements.items():
            infile = infile.replace(old, new)
        out.write(infile)
        out.close()

    def _manifest_files(self, base):
        temp = {**self._replace_email, **self._website_replacements}
        infile = self._open_read(base, '__manifest__.py')
        out = self._open_write(base, '__manifest__.py')
        for old, new in temp.items():
            infile = infile.replace(old, new)
        out.write(infile)
        out.close()
        self._content_replacements(base, '__manifest__.py', self._manifest_replacements)
        infile = self._open_read(base, '__manifest__.py')
        out = self._open_write(base, '__manifest__.py')
        manifest = ast.literal_eval(infile)
        manifest['author'] += ', %s' % self._company
        version_parts = manifest['version'].split('.')
        if len(version_parts) <= 3:
            version_parts = ['0', '0'] + version_parts
        if len(version_parts) < 5:
            version_parts += ['0'] * (5 - len(version_parts))
        version_parts[0] = self._release[0]
        version_parts[1] = self._release[1]

        manifest['version'] = '.'.join(version_parts)
        manifest['website'] = 'https://gitea.szsolutions.ch/flectra/%s' % self._repo_name
        self._manifest = manifest
        m = re.search('''["']version["']:[ '"]*([0-9a-zA-Z.]+)["']''', infile)
        if m and m.group(1):
            infile = infile.replace(m.group(1), manifest['version'])
        m = re.search('''["']website["']:[ '"]*(.*?)["']''', infile)
        if m and m.group(1):
            infile = infile.replace(m.group(1), manifest['website'])
        m = re.search('''["']author["']:[\n '"]*(.*?)["']''', infile, re.MULTILINE | re.DOTALL)
        if m and m.group(1):
            infile = infile.replace(m.group(1), m.group(1) + ', %s' % self._company)
        out.write(infile)
        out.close()

    def _xml_csv_json_files(self, base, name):
        infile = self._open_read(base, name)
        out = self._open_write(base, name)
        for old, new in self._replace_email.items():
            infile = infile.replace(old, new)
        for old, new in self._xml_replacements.items():
            must_replace = True
            for ignore_word in self._ignore_xml_words:
                if ignore_word in old:
                    must_replace = False
                    break
            if must_replace:
                infile = infile.replace(old, new)
        out.write(infile)
        out.close()

    def _python_files(self, base, name):
        infile = self._open_read(base, name)
        out = self._open_write(base, name)
        for old, new in self._replace_email.items():
            infile = infile.replace(old, new)
        out.write(infile)
        out.close()
        self._content_replacements(base, name, self._replacements)

    def _content_replacements(self, base, name, replace_dict):
        infile = self._open_read_lines(base, name)
        lines = []
        if infile:
            for line in infile:
                words = line.split(' ')
                line_parts = []
                for word in words:
                    if word.startswith('info@') or word.startswith("'info@") or word.startswith('"info@'):
                        line_parts.append(word)
                        continue
                    for old, new in replace_dict.items():
                        must_replace = True
                        for ignore_word in self._ingnore_py_words:
                            if ignore_word in word:
                                must_replace = False
                        if must_replace:
                            word = word.replace(old, new)
                    line_parts.append(word)
                lines.append(line_parts)

        with open(oj(self._destination, 'temp'), 'a') as temp_file:
            for line in lines:
                for word in line:
                    word = word if word.endswith('\n') else word + ' ' if word else ' '
                    temp_file.write(word)
            os.rename(oj(self._destination, 'temp'), oj(base, name))

    def _open_read_lines(self, base, name):
        return open(oj(base, name), 'r').readlines()

    def _open_write(self, base, filename):
        return open(oj(base, filename), 'w')

    def _open_read(self, base, filename):
        return open(oj(base, filename), 'r').read()

    def _remove_obsolete_lines(self, base, name):
        infile = self._open_read_lines(base, name)
        lines = []
        if not infile:
            return

        temp = self._obsolete_line_patterns + self._copyright_line_patterns
        for line in infile:
            remove_line = False
            for pattern in temp:
                if pattern in line:
                    if pattern in self._copyright_line_patterns:
                        cr_line = line.replace('#', '').strip()
                        self._copyright_list.append(cr_line)
                    else:
                        remove_line = True
            if not remove_line:
                lines.append(line)

        with open(oj(self._destination, 'temp'), 'a') as temp_file:
            for line in lines:
                temp_file.write(line)
            os.rename(oj(self._destination, 'temp'), oj(base, name))

    def _readme_files(self, base, name):
        infile = self._open_read(base, name)
        out = self._open_write(base, name)
        infile = self._readme_intro + infile

        for pattern in self._readme_regex_patterns:
            infile = self._regex_replace(infile, pattern)
        infile = self._regex_replace(infile,
                                     '(Contributors\n[-|~]*[\n|\n\n])',
                                     self._contributor_part)
        infile = infile.rstrip('\n')
        out.write(infile)
        out.close()

    def _regex_replace(self, source, pattern, replace=''):
        m = re.search(pattern, source, re.MULTILINE | re.DOTALL)
        if not m or not m.group(1):
            return source
        return source.replace(m.group(1), replace)