From 482ff6418915c25d45dc115ebd5aba610a4ee867 Mon Sep 17 00:00:00 2001 From: Marc Michalsky Date: Mon, 1 Aug 2022 17:34:43 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=89=20initial=20commit?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .idea/.gitignore | 8 ++ .idea/.name | 1 + .idea/inspectionProfiles/Project_Default.xml | 14 +++ .../inspectionProfiles/profiles_settings.xml | 6 ++ .idea/misc.xml | 4 + .idea/modules.xml | 8 ++ .idea/rename_viruses.iml | 10 ++ rename_viruses.py | 98 +++++++++++++++++++ 8 files changed, 149 insertions(+) create mode 100644 .idea/.gitignore create mode 100644 .idea/.name create mode 100644 .idea/inspectionProfiles/Project_Default.xml create mode 100644 .idea/inspectionProfiles/profiles_settings.xml create mode 100644 .idea/misc.xml create mode 100644 .idea/modules.xml create mode 100644 .idea/rename_viruses.iml create mode 100644 rename_viruses.py diff --git a/.idea/.gitignore b/.idea/.gitignore new file mode 100644 index 0000000..13566b8 --- /dev/null +++ b/.idea/.gitignore @@ -0,0 +1,8 @@ +# Default ignored files +/shelf/ +/workspace.xml +# Editor-based HTTP Client requests +/httpRequests/ +# Datasource local storage ignored files +/dataSources/ +/dataSources.local.xml diff --git a/.idea/.name b/.idea/.name new file mode 100644 index 0000000..4b5cc45 --- /dev/null +++ b/.idea/.name @@ -0,0 +1 @@ +rename_viruses \ No newline at end of file diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml new file mode 100644 index 0000000..99e1656 --- /dev/null +++ b/.idea/inspectionProfiles/Project_Default.xml @@ -0,0 +1,14 @@ + + + + \ No newline at end of file diff --git a/.idea/inspectionProfiles/profiles_settings.xml b/.idea/inspectionProfiles/profiles_settings.xml new file mode 100644 index 0000000..105ce2d --- /dev/null +++ b/.idea/inspectionProfiles/profiles_settings.xml @@ -0,0 +1,6 @@ + + + + \ No newline at end of file diff --git a/.idea/misc.xml b/.idea/misc.xml new file mode 100644 index 0000000..2b251db --- /dev/null +++ b/.idea/misc.xml @@ -0,0 +1,4 @@ + + + + \ No newline at end of file diff --git a/.idea/modules.xml b/.idea/modules.xml new file mode 100644 index 0000000..b068cff --- /dev/null +++ b/.idea/modules.xml @@ -0,0 +1,8 @@ + + + + + + + + \ No newline at end of file diff --git a/.idea/rename_viruses.iml b/.idea/rename_viruses.iml new file mode 100644 index 0000000..b437c3d --- /dev/null +++ b/.idea/rename_viruses.iml @@ -0,0 +1,10 @@ + + + + + + + + + + \ No newline at end of file diff --git a/rename_viruses.py b/rename_viruses.py new file mode 100644 index 0000000..3b25f8f --- /dev/null +++ b/rename_viruses.py @@ -0,0 +1,98 @@ +import argparse + +import os +import sys +import re + +my_parser = argparse.ArgumentParser(description='Rename all suspicious files found by clamav') + +my_parser.add_argument('File', + metavar='file', + type=str, + help='path to clamav log file') + +my_parser.add_argument('-s', + '--suffix', + type=str, + help='suffix to add to the end of the file names', + default='.VIRUS') + +my_parser.add_argument('-u', + '--undo', + action='store_true', + help='undo renaming') + +my_parser.add_argument('--dry-run', + action='store_true', + help='perform a test run where no file names are actually changed') + + +def main(): + args = my_parser.parse_args() + file = args.File + success_count = 0 + error_count = 0 + warning_count = 0 + + if not os.path.isfile(file): + print(f"The file '{file}' does not exist") + sys.exit() + + with open(file) as f: + lines = f.readlines() + files_to_rename = [re.split(r":", line.strip())[0] for line in lines if re.search(r"FOUND$", line)] + + for file_to_rename in files_to_rename: + new_name = file_to_rename + args.suffix + if not args.undo: + if not os.path.isfile(file_to_rename): + if os.path.isfile(new_name): + print(f"WARNING: The file {file_to_rename} was already renamed to {new_name}") + warning_count += 1 + else: + print(f"ERROR: Could not rename {file_to_rename}") + error_count += 1 + continue + else: + try: + if not args.dry_run: + os.rename(file_to_rename, new_name) + print(f"SUCCESS: {file_to_rename} -> {new_name}") + success_count += 1 + except Exception as e: + print(f"ERROR: Could not rename {file_to_rename}: {e}") + error_count += 1 + else: + if not os.path.isfile(new_name): + if os.path.isfile(file_to_rename): + print(f"WARNING: The file {file_to_rename} was not yet renamed") + warning_count += 1 + else: + print(f"ERROR: Could not undo renaming of {new_name}") + error_count += 1 + continue + else: + try: + if not args.dry_run: + os.rename(new_name, file_to_rename) + print(f"SUCCESS: {new_name} -> {file_to_rename}") + success_count += 1 + except Exception as e: + print(f"ERROR: Could not undo renaming of {new_name}: {e}") + error_count += 1 + + if error_count or warning_count: + print(f"--------------------------------------------------------------\n" + f"Renaming {'would have' if args.dry_run else ''} finished with " + f"{error_count} error{'s' if error_count > 1 else ''} and " + f"{warning_count} warning{'s' if warning_count > 1 else ''}.\n" + f"{success_count if success_count > 0 else 'No'} files {'would have been ' if args.dry_run else 'were '} " + "renamed successfully.") + else: + print(f"--------------------------------------------------------------\n" + f"{success_count if success_count > 0 else 'No'} files {'would have been ' if args.dry_run else 'were '}" + "renamed successfully.") + + +if __name__ == '__main__': + main()