1. import os
  2. import argparse
  3. import logging
  5. # Configure logging
  6. logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
  8. def process_files(input_dir, output_dir, operation, file_pattern):
  9. """
  10. Processes files in a directory based on a specified operation.
  11. """
  12. if not os.path.exists(input_dir):
  13. logging.error(f"Input directory '{input_dir}' does not exist.")
  14. return
  16. if not os.path.exists(output_dir):
  17. os.makedirs(output_dir)
  18. logging.info(f"Created output directory '{output_dir}'.")
  20. files = [f for f in os.listdir(input_dir) if file_pattern in f]
  22. if not files:
  23. logging.warning(f"No files matching pattern '{file_pattern}' found in '{input_dir}'.")
  24. return
  26. for filename in files:
  27. input_path = os.path.join(input_dir, filename)
  28. output_filename = filename # Default: keep the same name
  29. output_path = os.path.join(output_dir, output_filename)
  31. try:
  32. if operation == "copy":
  33. # Simple copy operation
  34. with open(input_path, 'rb') as infile, open(output_path, 'wb') as outfile:
  35. outfile.write(infile.read())
  36. logging.info(f"Copied '{filename}' to '{output_filename}'.")
  38. elif operation == "rename":
  39. # Rename the file
  40. new_filename = filename.replace(".txt", "_processed.txt") #example
  41. output_path = os.path.join(output_dir, new_filename)
  42. os.rename(input_path, output_path)
  43. logging.info(f"Renamed '{filename}' to '{new_filename}'.")
  45. elif operation == "validate":
  46. # Basic validation (example: check file size)
  47. file_size = os.path.getsize(input_path)
  48. if file_size > 1000000: # 1MB
  49. logging.warning(f"File '{filename}' is too large ({file_size} bytes). Skipping.")
  50. continue
  51. else:
  52. logging.info(f"Validated '{filename}' (size: {file_size} bytes).")
  54. else:
  55. logging.error(f"Invalid operation: {operation}")
  56. continue
  58. except Exception as e:
  59. logging.error(f"Error processing '{filename}': {e}")
  61. def main():
  62. """
  63. Main function to parse arguments and execute the file processing.
  64. """
  65. parser = argparse.ArgumentParser(description='Batch file processing tool.')
  66. parser.add_argument('input_dir', help='Input directory containing files.')
  67. parser.add_argument('output_dir', help='Output directory for processed files.')
  68. parser.add_argument('operation', choices=['copy', 'rename', 'validate'], help='Operation to perform.')
  69. parser.add_argument('file_pattern', help='File pattern to filter files (e.g., *.txt)')
  71. args = parser.parse_args()
  73. process_files(args.input_dir, args.output_dir, args.operation, args.file_pattern)
  76. if __name__ == "__main__":
  77. main()

Add your comment