63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
import os
|
|
import hashlib
|
|
import json
|
|
import argparse
|
|
|
|
excluded_paths = [
|
|
'manifest.json',
|
|
'Updater.exe'
|
|
]
|
|
|
|
def is_excluded(path):
|
|
for excluded in excluded_paths:
|
|
|
|
if path.startswith(excluded):
|
|
return True
|
|
return False
|
|
|
|
def file_hash(path):
|
|
h = hashlib.sha1()
|
|
with open(path, 'rb') as f:
|
|
while chunk := f.read(8192):
|
|
h.update(chunk)
|
|
return h.hexdigest()
|
|
|
|
def generate_manifest(directory):
|
|
manifest = {}
|
|
for root, dirs, files in os.walk(directory):
|
|
for file in files:
|
|
file_path = os.path.join(root, file)
|
|
relative_path = os.path.relpath(file_path, directory).replace("\\", "/")
|
|
|
|
if is_excluded(relative_path):
|
|
continue
|
|
|
|
hash_value = file_hash(file_path)
|
|
manifest[relative_path] = hash_value
|
|
return manifest
|
|
|
|
def save_manifest(manifest, output_path):
|
|
with open(output_path, 'w') as f:
|
|
json.dump(manifest, f, indent=4)
|
|
print(f"Manifest saved to {output_path}")
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Generate manifest.json with SHA1 hashes.")
|
|
parser.add_argument('directory', type=str, help="Folder to scan")
|
|
parser.add_argument('--output', type=str, help="Optional output path for manifest.json")
|
|
args = parser.parse_args()
|
|
|
|
directory = args.directory
|
|
if not os.path.isdir(directory):
|
|
print(f"Directory '{directory}' does not exist.")
|
|
return
|
|
|
|
output_path = args.output if args.output else os.path.join(directory, 'manifest.json')
|
|
print(f"Scanning directory: {directory}...")
|
|
|
|
manifest = generate_manifest(directory)
|
|
save_manifest(manifest, output_path)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|