diff --git a/bin/tests/base2img.py b/bin/tests/base2img.py index e9db269..6cbe0dd 100644 --- a/bin/tests/base2img.py +++ b/bin/tests/base2img.py @@ -6,9 +6,10 @@ # @File Name: base2img.py import base64 +import os -def convert() -> None: +def convert_single() -> None: """ :return: @@ -20,4 +21,22 @@ def convert() -> None: print(base64_code) -convert() +def convert_multi() -> None: + """ + + :return: + """ + file_list = os.listdir('../res/images') + dict_count = 0 + img_dict = {} + for i in file_list: + with open(f'../res/images/theme2{i}', 'rb') as f: + base64_data = base64.b64encode(f.read()) + base64_code = 'data:image/svg+xml;base64,' + base64_data.decode('utf-8') + img_dict[str(dict_count)] = base64_code + dict_count += 1 + + print(img_dict) + + +convert_multi() \ No newline at end of file