61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
|
from stegano import lsb
|
|||
|
|
|
|||
|
|
class auth():
|
|||
|
|
def __init__(self):
|
|||
|
|
global username
|
|||
|
|
username = input('Enter USERNAME: ')
|
|||
|
|
|
|||
|
|
if len(username.split()) > 0:
|
|||
|
|
main()
|
|||
|
|
else:
|
|||
|
|
exit()
|
|||
|
|
|
|||
|
|
|
|||
|
|
class main():
|
|||
|
|
def __init__(self):
|
|||
|
|
menu = '''
|
|||
|
|
[1] - спрятать сообщение в изображении
|
|||
|
|
[2] - показать спрятанное сообщение
|
|||
|
|
[3] - спрятать сообщение в изображении (с подписью)
|
|||
|
|
'''
|
|||
|
|
print(menu)
|
|||
|
|
|
|||
|
|
cmd = input('CMD: ')
|
|||
|
|
|
|||
|
|
if cmd == '1':
|
|||
|
|
image = input('Введите абсолютный путь до файла PNG or JPG: ')
|
|||
|
|
msg = input('Введите секретное сообщение: ')
|
|||
|
|
|
|||
|
|
if len(image.split()) > 0 and len(msg.split()) > 0:
|
|||
|
|
sc_img = lsb.hide(image, msg)
|
|||
|
|
sc_img.save('secret_image.png')
|
|||
|
|
print('Сообщение было спрятано в secret_image.png')
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
exit()
|
|||
|
|
|
|||
|
|
elif cmd == '2':
|
|||
|
|
image = input('Введите абсолютный путь до файла PNG or JPG: ')
|
|||
|
|
|
|||
|
|
if len(image.split()) > 0:
|
|||
|
|
sc_message = lsb.reveal(image)
|
|||
|
|
print(f'Секретное сообщение:\n{sc_message}')
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
exit()
|
|||
|
|
|
|||
|
|
elif cmd == '3':
|
|||
|
|
image = input('Введите абсолютный путь до файла PNG or JPG: ')
|
|||
|
|
msg = input('Введите секретное сообщение: ')
|
|||
|
|
|
|||
|
|
if len(image.split()) > 0 and len(msg.split()) > 0:
|
|||
|
|
sc_img = lsb.hide(image, message=f'{msg}\n\nby {username}')
|
|||
|
|
sc_img.save('secret_image.png')
|
|||
|
|
print('Сообщение было спрятано в secret_image.png')
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
exit()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
auth()
|