1. import os
  2. import sys
  3. import json
  4. import argparse
  7. def validate_filename(filename):
  8. """Basic filename validation."""
  9. if not filename:
  10. return False, "Filename cannot be empty."
  11. if not filename.endswith(('.bin')):
  12. return False, "Filename must end with .bin"
  13. return True, None
  16. def extract_metadata(filepath):
  17. """Extracts metadata from a binary file (simulated)."""
  18. try:
  19. with open(filepath, 'rb') as f:
  20. # Simulate reading metadata - replace with actual parsing logic
  21. metadata = {
  22. "size": os.path.getsize(filepath),
  23. "creation_date": file_creation_date(filepath),
  24. "author": "Unknown",
  25. "description": "Binary file metadata"
  26. }
  27. return metadata
  28. except FileNotFoundError:
  29. return None
  30. except Exception as e:
  31. print(f"Error reading file {filepath}: {e}")
  32. return None
  35. def file_creation_date(filepath):
  36. """Simulates getting creation date."""
  37. return os.path.getctime(filepath)
  40. def attach_metadata(filepath, metadata_file):
  41. """Attaches metadata to a binary file."""
  43. is_valid, error_message = validate_filename(filepath)
  44. if not is_valid:
  45. print(f"Error: {error_message}")
  46. return False
  48. metadata = extract_metadata(filepath)
  49. if metadata is None:
  50. return False
  52. try:
  53. with open(metadata_file, 'r') as f:
  54. try:
  55. metadata_data = json.load(f)
  56. except json.JSONDecodeError:
  57. print(f"Error: Invalid JSON in {metadata_file}. Creating a new metadata file.")
  58. metadata_data = {}
  59. except FileNotFoundError:
  60. print(f"Creating new metadata file {metadata_file}")
  61. metadata_data = {}
  62. except Exception as e:
  63. print(f"Error reading metadata file {metadata_file}: {e}")
  64. return False
  66. metadata_data[os.path.basename(filepath)] = metadata
  68. try:
  69. with open(metadata_file, 'w') as f:
  70. json.dump(metadata_data, f, indent=4)
  71. print(f"Metadata attached to {filepath} and saved to {metadata_file}")
  72. return True
  73. except Exception as e:
  74. print(f"Error writing to metadata file {metadata_file}: {e}")
  75. return False
  77. def main():
  78. parser = argparse.ArgumentParser(description="Attach metadata to binary files.")
  79. parser.add_argument("binary_files", nargs="+", help="List of binary files to process.")
  80. parser.add_argument("-m", "--metadata_file", default="metadata.json", help="File to store metadata.")
  82. args = parser.parse_args()
  84. for binary_file in args.binary_files:
  85. attach_metadata(binary_file, args.metadata_file)
  88. if __name__ == "__main__":
  89. main()

Add your comment