fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-06-28 03:06:14 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Version 0.8.

Browse source
This commit is contained in:
Cédric Bonhomme 2017-05-06 23:20:54 +02:00
parent a770f402cb
commit cbb9df7675
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
6 changed files with 34 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

9
stegano/red/red.py
View file

@ -28,8 +28,9 @@ __license__ = "GPLv3"
import sys
from PIL import Image
from typing import Union, IO
def hide(input_image_file, message):
def hide(input_image: Union[str, IO[bytes]], message: str):
"""
Hide a message (string) in an image.
@ -40,7 +41,7 @@ def hide(input_image_file, message):
message_length = len(message)
assert message_length != 0, "message message_length is zero"
assert message_length < 255, "message is too long"
img = Image.open(input_image_file)
img = Image.open(input_image)
# Use a copy of image to hide the text in
encoded = img.copy()
width, height = img.size
@ -61,7 +62,7 @@ def hide(input_image_file, message):
img.close()
return encoded
def reveal(input_image_file):
def reveal(input_image: Union[str, IO[bytes]]):
"""
Find a message in an image.
@ -69,7 +70,7 @@ def reveal(input_image_file):
hidden message characters (ASCII values).
The red value of the first pixel is used for message_length of string.
"""
img = Image.open(input_image_file)
img = Image.open(input_image)
width, height = img.size
message = ""
index = 0