diff --git a/Forums/Utilities/utilities.py b/Forums/Utilities/utilities.py index 741ec1f..2a5e2f0 100644 --- a/Forums/Utilities/utilities.py +++ b/Forums/Utilities/utilities.py @@ -353,29 +353,24 @@ def encrypt_encode_image_to_base64(driver, xpath): return None -def decode_decrypt_image_in_base64(html_content): +def decode_decrypt_image_in_base64(string_image): - soup = BeautifulSoup(html_content, 'html.parser') - - for img_tag in soup.find_all('img'): - - src_attr = img_tag.get('src') + try: - if src_attr and src_attr.startswith('data:image'): + base64_image = bytes(string_image, encoding='utf-8') + encrypted_image = base64.b64decode(base64_image) + decrypted_image = aes_decryption(encrypted_image) - try: + im = Image.open(io.BytesIO(decrypted_image)) + im.show() - string_image = src_attr.split('base64,')[-1] - base64_image = bytes(string_image, encoding='utf-8') - encrypted_image = base64.b64decode(base64_image) - decrypted_image = aes_decryption(encrypted_image) + return decrypted_image - im = Image.open(io.BytesIO(decrypted_image)) - im.show() + except Exception as e: + print(e) + pass - except Exception as e: - print(e) - pass + return None def replace_image_sources(driver, html_content):