#!./.venv/bin/python import argparse # import glob import ast import logging import os import sys from pathlib import Path logging.basicConfig(level=logging.DEBUG) _logger = logging.getLogger(__name__) def get_config(): """Parse command line arguments, extracting the config file name, returning the union of config file and command line arguments :return: dict of config file settings and command line arguments """ parser = argparse.ArgumentParser( formatter_class=argparse.RawDescriptionHelpFormatter, description="""\ Search all method class and give a list separate by ; """, epilog="""\ """, ) parser.add_argument( "-d", "--directory", dest="directory", required=True, help="Directory to execute search in recursive.", ) parser.add_argument( "-t", "--template_dir", dest="template_dir", help=( "Overwrite value template_model_name in template module, give only" " template source directory, will update file hooks.py" ), ) parser.add_argument( "--with_inherit", action="store_true", help="Will search inherit model", ) parser.add_argument( "-q", "--quiet", action="store_true", help="Don't show output of found model.", ) args = parser.parse_args() return args def search_and_replace( f_lines, hooks_file_path, models_name, search_word="template_model_name" ): if not models_name: return f_lines t_index = f_lines.find(search_word) if t_index == -1: _logger.error(f"Cannot find {search_word} in file {hooks_file_path}") return -1 t_index_equation = f_lines.index("=", t_index + 1) if t_index_equation == -1: _logger.error(f"Cannot find {search_word} = in file {hooks_file_path}") return -1 # find next character i = 1 while f_lines[t_index_equation + i] in (" ", "\n"): i += 1 first_char = f_lines[t_index_equation + i] if first_char == "(": second_char = ")" elif f_lines[t_index_equation + i : t_index_equation + i + 3] == '"""': first_char = '"""' second_char = '"""' else: second_char = first_char # t_index_first_quote = f_lines.index(first_char, t_index + 1) t_index_second_quote = f_lines.index(first_char, t_index_equation + i) if t_index_second_quote == -1: _logger.error( f'Cannot find {search_word} = "##" in file {hooks_file_path}' ) return -1 t_index_third_quote = f_lines.index(second_char, t_index_second_quote + 1) if t_index_third_quote == -1: _logger.error(f"Cannot find third quote in file {hooks_file_path}") return -1 # if "\n" in models_name: # new_file_content = ( # f'{f_lines[:t_index_second_quote]}"""{models_name}"""{f_lines[t_index_third_quote + len(second_char):]}' # ) # else: # new_file_content = ( # f'{f_lines[:t_index_second_quote]}"{models_name}"{f_lines[t_index_third_quote + len(second_char):]}' # ) new_file_content = ( f'{f_lines[:t_index_second_quote]}"{models_name}"{f_lines[t_index_third_quote + len(second_char):]}' ) return new_file_content def main(): config = get_config() if not os.path.exists(config.directory): _logger.error(f"Path directory {config.directory} not exist.") return -1 lst_model_name = [] lst_model_inherit_name = [] lst_search_target = ("_name",) lst_search_inherit_target = ("_inherit",) if config.with_inherit else [] # lst_py_file = glob.glob(os.path.join(config.directory, "***", "*.py")) lst_py_file = Path(config.directory).rglob("*.py") for py_file in lst_py_file: if py_file == "__init__.py": continue with open(py_file, "r") as source: f_lines = source.read() try: f_ast = ast.parse(f_lines) except Exception as e: _logger.error(f"Cannot parse file {py_file}") continue for children in f_ast.body: if type(children) == ast.ClassDef: # Detect good _name for node in children.body: if ( type(node) is ast.Assign and node.targets and type(node.targets[0]) is ast.Name # and node.targets[0].id in ("_name",) and type(node.value) is ast.Str ): if ( lst_search_target and node.targets[0].id in lst_search_target ): if node.value.s in lst_model_name: _logger.warning( "Duplicated model name" f" {node.value.s} from file {py_file}" ) else: lst_model_name.append(node.value.s) if ( lst_search_inherit_target and node.targets[0].id in lst_search_inherit_target ): if node.value.s in lst_model_inherit_name: _logger.warning( "Duplicated model inherit name" f" {node.value.s} from file {py_file}" ) else: lst_model_inherit_name.append(node.value.s) lst_model_name.sort() lst_model_inherit_name.sort() models_name = "; ".join(lst_model_name) # TODO temporary fix, remove this when it's supported lst_ignored_inherit = ["portal.mixin", "mail.thread"] for ignored_inherit in lst_ignored_inherit: if ignored_inherit in lst_model_inherit_name: lst_model_inherit_name.remove(ignored_inherit) models_inherit_name = "; ".join(lst_model_inherit_name) if not models_name: _logger.warning(f"Missing models class in {config.directory}") elif not config.quiet: # _logger.info(models_name) print(models_name) print(models_inherit_name) if config.template_dir: if not os.path.exists(config.template_dir): _logger.error( f"Path template dir {config.template_dir} not exist." ) return -1 hooks_file_path = os.path.join(config.template_dir, "hooks.py") if not os.path.exists(hooks_file_path): _logger.error( f"Path template hooks.py file {hooks_file_path} not exist." ) return -1 with open(hooks_file_path, "r") as source: f_lines = source.read() # Throw exception if not found new_file_content = search_and_replace( f_lines, hooks_file_path, models_name ) if models_inherit_name: new_file_content = search_and_replace( new_file_content, hooks_file_path, models_inherit_name, search_word="template_inherit_model_name", ) with open(hooks_file_path, "w") as source: source.write(new_file_content) # Call format all file os.system(f"./script/maintenance/format.sh {hooks_file_path}") return 0 if __name__ == "__main__": status = main() sys.exit(status)