From 3c0180bee0170771d0ab6ac435c675ef706c6b55 Mon Sep 17 00:00:00 2001 From: Zubair Shakoor <57657330+zubairshakoorarbisoft@users.noreply.github.com> Date: Wed, 9 Aug 2023 15:35:37 +0500 Subject: [PATCH] fix: providing_args removal codemod added (#413) Co-authored-by: Usama Sadiq --- .../django42/remove_providing_args_arg.py | 46 +++++++++++++++++++ setup.py | 1 + 2 files changed, 47 insertions(+) create mode 100644 edx_repo_tools/codemods/django42/remove_providing_args_arg.py diff --git a/edx_repo_tools/codemods/django42/remove_providing_args_arg.py b/edx_repo_tools/codemods/django42/remove_providing_args_arg.py new file mode 100644 index 00000000..5a85a8f3 --- /dev/null +++ b/edx_repo_tools/codemods/django42/remove_providing_args_arg.py @@ -0,0 +1,46 @@ +import os +import re +import click + +def remove_providing_args(root_dir): + # Regex pattern to match the lines containing providing_args + pattern = r"(.*)[,\s]*providing_args\s*=\s*\[.*?\](.*)" + + # Traverse all Python files in the root directory + for root, _, files in os.walk(root_dir): + for file in files: + if file.endswith(".py"): + file_path = os.path.join(root, file) + updated_lines = [] + + # Open the file and read its content + with open(file_path, "r") as f: + lines = f.readlines() + + # Process each line in the file + for line in lines: + # Check if the line contains providing_args + match = re.match(pattern, line) + if match: + # Remove the providing_args argument along with any preceding comma or whitespace + updated_line = match.group(1).rstrip(", \t") + match.group(2) + "\n" + updated_lines.append(updated_line) + else: + updated_lines.append(line) + + # Write the updated content back to the file + with open(file_path, "w") as f: + f.writelines(updated_lines) + + +@click.command() +@click.option( + '--root_dir', default='.', + help="Path to root of project") +def main(root_dir): + remove_providing_args(root_dir) + print("Providing_args removed from the specified lines.") + + +if __name__ == '__main__': + main() \ No newline at end of file diff --git a/setup.py b/setup.py index 3c39d47b..0ece75b2 100644 --- a/setup.py +++ b/setup.py @@ -111,6 +111,7 @@ def is_requirement(line): 'tag_release = edx_repo_tools.release.tag_release:main', 'modernize_tox_django42 = edx_repo_tools.codemods.django42.tox_moderniser_django42:main', 'modernize_github_actions_django42 = edx_repo_tools.codemods.django42.github_actions_modernizer_django42:main', + 'remove_providing_args = edx_repo_tools.codemods.django42.remove_providing_args_arg:main', ], }, package_data={