Browse Source

updated decryption for forums utilities.py

main
westernmeadow 1 year ago
parent
commit
2a9dbdcba5
1 changed files with 12 additions and 17 deletions
  1. +12
    -17
      Forums/Utilities/utilities.py

+ 12
- 17
Forums/Utilities/utilities.py View File

@ -353,29 +353,24 @@ def encrypt_encode_image_to_base64(driver, xpath):
return None 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): def replace_image_sources(driver, html_content):


Loading…
Cancel
Save