CIRCL
/
lookyloo
mirror of https://github.com/CIRCL/lookyloo
2
0
Fork 0
Code Issues Releases Wiki Activity

fix: Typing in SRI generator

pull/219/head
Raphaël Vinot 2021-06-16 17:52:27 -07:00
parent 8b08d76e48
commit e6a027c82d
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tools/generate_sri.py
View File

@ -5,16 +5,18 @@ import base64
import hashlib import hashlib
import json import json
from typing import Dict
from lookyloo.helpers import get_homedir from lookyloo.helpers import get_homedir
if __name__ == '__main__': if __name__ == '__main__':
dest_dir = get_homedir() / 'website' / 'web' dest_dir = get_homedir() / 'website' / 'web'
to_save = {'static': {}} to_save: Dict = {'static': {}}
for resource in (dest_dir / 'static').glob('*'): for resource in (dest_dir / 'static').glob('*'):
with resource.open('rb') as f: with resource.open('rb') as f:
to_save['static'][resource.name] = base64.b64encode(hashlib.sha512(f.read()).digest()).decode('utf-8') to_save['static'][resource.name] = base64.b64encode(hashlib.sha512(f.read()).digest()).decode('utf-8')
with (dest_dir / 'sri.txt').open('w') as f: with (dest_dir / 'sri.txt').open('w') as fw:
json.dump(to_save, f, indent=2) json.dump(to_save, fw, indent=2)