Fixed #12. More tests to come.

This commit is contained in:
Cédric Bonhomme 2017-03-08 22:26:12 +01:00
parent 091f33c521
commit 7a5c23ef2f
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
3 changed files with 6 additions and 15 deletions

View file

@ -81,13 +81,6 @@ if arguments.command == 'hide':
elif arguments.command == 'reveal':
secret = lsb.reveal(arguments.input_image_file)
if arguments.secret_binary != None:
"""data = tools.base642binary(secret)
from PIL import Image
import io
file_like = io.BytesIO(data)
file_like.seek(0)
image = Image.open(file_like)
image.save(arguments.secret_binary)"""
data = tools.base642binary(secret)
with open(arguments.secret_binary, "wb") as f:
f.write(data)