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

chg: [typing] Make Mypy Happy Again.

Browse source
This commit is contained in:
Cédric Bonhomme 2025-06-22 12:03:30 +02:00
parent 21004219fa
commit 18fd8a8a49
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
5 changed files with 43 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

14
stegano/red/red.py
View file

@ -23,7 +23,7 @@ __date__ = "$Date: 2010/10/01 $"
__revision__ = "$Date: 2017/02/06 $"
__license__ = "GPLv3"
from typing import IO, Union
from typing import IO, Union, cast
from stegano import tools
@ -40,13 +40,17 @@ def hide(input_image: Union[str, IO[bytes]], message: str):
assert message_length != 0, "message message_length is zero"
assert message_length < 255, "message is too long"
img = tools.open_image(input_image)
# Ensure image mode is RGB
if img.mode != "RGB":
img = img.convert("RGB")
# Use a copy of image to hide the text in
encoded = img.copy()
width, height = img.size
index = 0
for row in range(height):
for col in range(width):
(r, g, b) = img.getpixel((col, row))
pixel = cast(tuple[int, int, int], img.getpixel((col, row)))
r, g, b = pixel
# first value is message_length of message
if row == 0 and col == 0 and index < message_length:
asc = message_length
@ -70,12 +74,16 @@ def reveal(input_image: Union[str, IO[bytes]]):
The red value of the first pixel is used for message_length of string.
"""
img = tools.open_image(input_image)
# Ensure image mode is RGB
if img.mode != "RGB":
img = img.convert("RGB")
width, height = img.size
message = ""
index = 0
for row in range(height):
for col in range(width):
r, g, b = img.getpixel((col, row))
pixel = cast(tuple[int, int, int], img.getpixel((col, row)))
r, g, b = pixel
# First pixel r value is length of message
if row == 0 and col == 0:
message_length = r