# Copyright (c) 2024, Oracle and/or its affiliates. # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License, version 2.0, as # published by the Free Software Foundation. # # This program is designed to work with certain software (including # but not limited to OpenSSL) that is licensed under separate terms, # as designated in a particular file or component or in included license # documentation. The authors of MySQL hereby grant you an # additional permission to link the program and your derivative works # with the separately licensed software that they have either included with # the program or referenced in the documentation. # # Without limiting anything contained in the foregoing, this file, # which is part of MySQL Connector/Python, is also subject to the # Universal FOSS Exception, version 1.0, a copy of which can be found at # http://oss.oracle.com/licenses/universal-foss-exception. # # This program is distributed in the hope that it will be useful, but # WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. # See the GNU General Public License, version 2.0, for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software Foundation, Inc., # 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA """ Script to update the copyright header of the source files of connectors distributed with `connector-python`. This program traverses each connector package included in the `connector-python` looking for source code, configuration, installation, documentation, and supporting files that include the `Copyright` header pattern. As a result, such a `Copyright` header is updated based on the rules defined in this very script. Also, this script expects no command-line arguments, however, a relative location to the root folder `connector-python` is assumed, therefore, if you move this script to another location, it won't work. It's up to the developer to decide where or when this program will be executed. Use case example: In the event of a "Copyright" change, the developer can update the rules found in this program and run it to make the changes effective across the relevant code base. """ import datetime import logging import os import re from pathlib import Path from typing import List, Tuple, Union CURRENT_YEAR = datetime.date.today().year DEFAULT_FOUND_YEAR = 2023 # `sol` stands for start-of-line COPYRIGHT_HEADER = """{sol} Copyright (c) {creation_year}{current_year}, Oracle and/or its affiliates. {sol} {sol} This program is free software; you can redistribute it and/or modify {sol} it under the terms of the GNU General Public License, version 2.0, as {sol} published by the Free Software Foundation. {sol} {sol} This program is designed to work with certain software (including {sol} but not limited to OpenSSL) that is licensed under separate terms, {sol} as designated in a particular file or component or in included license {sol} documentation. The authors of MySQL hereby grant you an {sol} additional permission to link the program and your derivative works {sol} with the separately licensed software that they have either included with {sol} the program or referenced in the documentation. {sol} {sol} Without limiting anything contained in the foregoing, this file, {sol} which is part of MySQL Connector/Python, is also subject to the {sol} Universal FOSS Exception, version 1.0, a copy of which can be found at {sol} http://oss.oracle.com/licenses/universal-foss-exception. {sol} {sol} This program is distributed in the hope that it will be useful, but {sol} WITHOUT ANY WARRANTY; without even the implied warranty of {sol} MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. {sol} See the GNU General Public License, version 2.0, for more details. {sol} {sol} You should have received a copy of the GNU General Public License {sol} along with this program; if not, write to the Free Software Foundation, Inc., {sol} 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA""" RE_FLAGS = re.MULTILINE | re.DOTALL PATTERN = r"^{sol_pattern}\sCopyright\s.*?\sMA\s+02110-1301\s+USA$" PATTERN_HEADER_PY = re.compile(PATTERN.format(sol_pattern=r"#"), flags=RE_FLAGS) PATTERN_HEADER_C = re.compile(PATTERN.format(sol_pattern=r"\s\*"), flags=RE_FLAGS) PATTERN_HEADER_PLAIN = re.compile( r"^Copyright\s.*?\sMA\s+02110-1301\s+USA$", flags=RE_FLAGS ) PATTERN_YEARS = r"Copyright\s+\(c\)\s+\d{4},\s*\d{4},\s+Oracle" TARGET_SRC_PY_EXTS = [".py", ".sh", ".spec", ".postinst", ".postrm", ".toml"] TARGET_SRC_C_EXTS = [".c", ".h", ".cpp", ".cc", ".proto"] TARGET_SRC_PLAIN_EXTS = [".rst"] TARGET_SRC_EXTS = TARGET_SRC_PY_EXTS + TARGET_SRC_C_EXTS + TARGET_SRC_PLAIN_EXTS SRC_BUT_EXTENSIONLESS = ["rules", "Dockerfile"] SHARED_FILES = [ "pyproject.toml", ] MYSQL_CONNECTORS = { "classic": "mysql-connector-python", "xdevapi": "mysqlx-connector-python", } EXCLUDE = [ os.path.join(MYSQL_CONNECTORS["classic"], "build"), os.path.join(MYSQL_CONNECTORS["classic"], "lib", "mysql", "opentelemetry"), os.path.join(MYSQL_CONNECTORS["xdevapi"], "build"), ] logging.basicConfig(level=logging.DEBUG) logger = logging.getLogger("cpy.headerfy") def update_header(root: Union[str, Path], filename: str) -> None: """Updates the Copyright header of the provided file. Args: root: Absolute path to the folder where `file` can be found. filename: File name to be updated. Returns: None Raises: PermissionError, FileNotFoundError, FileExistsError: If the opening of the file fails. RuntimeError: The file extension isn't supported. """ path_to_file = os.path.join(root, filename) pattern = None if ( any((filename.endswith(ext) for ext in TARGET_SRC_PY_EXTS)) or filename in SRC_BUT_EXTENSIONLESS ): pattern = PATTERN_HEADER_PY elif any((filename.endswith(ext) for ext in TARGET_SRC_C_EXTS)): pattern = PATTERN_HEADER_C elif any((filename.endswith(ext) for ext in TARGET_SRC_PLAIN_EXTS)): pattern = PATTERN_HEADER_PLAIN else: raise RuntimeError(f"Couldn't resolve extension for {filename}") sol = None if pattern == PATTERN_HEADER_PY: sol = "#" elif pattern == PATTERN_HEADER_C: sol = " *" elif pattern == PATTERN_HEADER_PLAIN: sol = "" with open(path_to_file, mode="r", encoding="utf8") as fp: text_old = fp.read() match_header = re.search(pattern, text_old) # no copyright header if match_header is None: return two_years_snippet = re.search(PATTERN_YEARS, match_header.group()) creation_year = ( two_years_snippet.group().split(",")[0].replace("Copyright (c)", "").strip() if two_years_snippet is not None else "" ) logger.info("%s (%s)", path_to_file, creation_year if creation_year else "NULL") creation_year = f"{creation_year}, " if creation_year else f"{DEFAULT_FOUND_YEAR}, " text_new = re.sub( pattern, COPYRIGHT_HEADER.format( sol=sol, creation_year=creation_year, current_year=CURRENT_YEAR ), text_old, ) with open(path_to_file, mode="w", encoding="utf8") as fp: fp.write(text_new) def headerify( path_base: Union[str, Path], package_name: str ) -> List[Tuple[str, Exception]]: """Traverses the given package directory and hands over the relevant* files to be updated to `update_header` which drives the actual update. [*]: Including source, configuration, installation and further supporting files such as documentation files. Args: path_base: Absolute path to the folder where `package_name` can be found. package_name: The connector to be processed. Returns: issues: A list of 2-tuple items; 1st includes the absolute path to a file that couldn't be updated and 2nd informs the corresponding raised error. """ warns: List[Tuple[str, Exception]] = [] for root, _, files in os.walk(Path(path_base, package_name)): if any( (os.path.join(path_base, path_exclude) in root for path_exclude in EXCLUDE) ): continue for filename in files: if ( any((filename.endswith(ext) for ext in TARGET_SRC_EXTS)) or filename in SRC_BUT_EXTENSIONLESS ): try: update_header(root, filename) except (PermissionError, FileNotFoundError, FileExistsError) as err: warns.append((os.path.join(root, filename), err)) return warns if __name__ == "__main__": path_to_base = Path(__file__).parent.parent issues: List[Tuple[str, Exception]] = [] for pkg_name in MYSQL_CONNECTORS.values(): issues.extend( headerify( path_base=path_to_base, package_name=pkg_name, ) ) for file in SHARED_FILES: try: update_header(path_to_base, file) except (PermissionError, FileNotFoundError, FileExistsError) as error: issues.append((os.path.join(path_to_base, file), error)) if issues: print("-" * 10 + "WARNING" + "-" * 10) for path_file, exc in issues: logger.warning("Couldn't update %s. Got %s", path_file, exc)