lookyloo/tools/generate_sri.py

24 lines
632 B
Python
Executable File

#!/usr/bin/env python3
import base64
import hashlib
import json
from typing import Dict, Any
from lookyloo.default import get_homedir
if __name__ == '__main__':
dest_dir = get_homedir() / 'website' / 'web'
to_save: Dict[str, Any] = {'static': {}}
for resource in (dest_dir / 'static').glob('*'):
if resource.name[0] == '.':
continue
with resource.open('rb') as f:
to_save['static'][resource.name] = base64.b64encode(hashlib.sha512(f.read()).digest()).decode('utf-8')
with (dest_dir / 'sri.txt').open('w') as fw:
json.dump(to_save, fw, indent=2, sort_keys=True)