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

View File

@ -5,16 +5,18 @@ import base64
import hashlib
import json
from typing import Dict
from lookyloo.helpers import get_homedir
if __name__ == '__main__':
dest_dir = get_homedir() / 'website' / 'web'
to_save = {'static': {}}
to_save: Dict = {'static': {}}
for resource in (dest_dir / 'static').glob('*'):
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 f:
json.dump(to_save, f, indent=2)
with (dest_dir / 'sri.txt').open('w') as fw:
json.dump(to_save, fw, indent=2)