From eea4c4de8ea2637ba4d87a24f9bbb729d1306361 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Bonhomme?= <cedric@cedricbonhomme.org>
Date: Sun, 13 Mar 2016 18:19:43 +0100
Subject: [PATCH] fix the compatibility problem between 'input' and 'raw_input'
 for Python 2 and 3.

---
 CHANGELOG.rst      | 5 +++++
 stegano/slsb.py    | 7 ++++++-
 stegano/slsbset.py | 7 ++++++-
 3 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.rst b/CHANGELOG.rst
index d60dab1..36ed070 100644
--- a/CHANGELOG.rst
+++ b/CHANGELOG.rst
@@ -1,6 +1,11 @@
 Release History
 ===============
 
+0.4.7 (not yet released)
+------------------------
+
+* management of greyscale images;
+
 0.4.6 (2016-03-12)
 ------------------
 
diff --git a/stegano/slsb.py b/stegano/slsb.py
index 359a892..3919117 100755
--- a/stegano/slsb.py
+++ b/stegano/slsb.py
@@ -30,6 +30,11 @@ from PIL import Image
 
 from . import tools
 
+try:
+   input = raw_input
+except NameError:
+   pass
+
 def hide(input_image_file, message):
     """
     Hide a message (string) in an image with the
@@ -49,7 +54,7 @@ def hide(input_image_file, message):
     width, height = img.size
     index = 0
 
-    message = str(len(message)) + ":" + message
+    message = str(len(message)) + ":" + str(message)
     #message_bits = tools.a2bits(message)
     message_bits = "".join(tools.a2bits_list(message))
     message_bits += '0' * ((3 - (len(message_bits) % 3)) % 3)
diff --git a/stegano/slsbset.py b/stegano/slsbset.py
index 28772ba..81e19bd 100644
--- a/stegano/slsbset.py
+++ b/stegano/slsbset.py
@@ -31,6 +31,11 @@ from PIL import Image
 from . import tools
 from . import generators
 
+try:
+   input = raw_input
+except NameError:
+   pass
+
 def hide(input_image_file, message, generator_function):
     """
     Hide a message (string) in an image with the
@@ -50,7 +55,7 @@ def hide(input_image_file, message, generator_function):
     width, height = img.size
     index = 0
 
-    message = str(len(message)) + ":" + message
+    message = str(len(message)) + ":" + str(message)
     #message_bits = tools.a2bits(message)
     message_bits = "".join(tools.a2bits_list(message))
     message_bits += '0' * ((3 - (len(message_bits) % 3)) % 3)