From 96a65a45a284d604d831711574adc33e795471bb Mon Sep 17 00:00:00 2001
From: cedricbonhomme <devnull@localhost>
Date: Wed, 6 Oct 2010 09:08:36 +0200
Subject: [PATCH] Command line improvement.

---
 lsb-s.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/lsb-s.py b/lsb-s.py
index a37ecdc..66fcd8d 100755
--- a/lsb-s.py
+++ b/lsb-s.py
@@ -126,8 +126,11 @@ def reveal_slow(img):
 if __name__ == '__main__':
     # Point of entry in execution mode.
     from optparse import OptionParser
-    usage = "usage: %prog hide|reveal [options]"
-    parser = OptionParser(usage)
+    parser = OptionParser(version=__version__)
+    parser.add_option('--hide', action='store_true', default=False,
+                      help="Hides a message in an image.")
+    parser.add_option('--reveal', action='store_true', default=False,
+                      help="Reveals the message hided in an image.")
     # Original image
     parser.add_option("-i", "--input", dest="input_image_file",
                     help="Input image fil.e")
@@ -153,7 +156,7 @@ if __name__ == '__main__':
     (options, args) = parser.parse_args()
 
 
-    if sys.argv[1] == "hide":
+    if options.hide:
         if options.secret_message != "" and options.secret_file == "":
             secret = options.secret_message
         elif options.secret_message == "" and options.secret_file != "":
@@ -167,7 +170,7 @@ if __name__ == '__main__':
             # If hide() returns an error (Too long message).
             print e
 
-    elif sys.argv[1] == "reveal":
+    elif options.reveal:
         img = Image.open(options.input_image_file)
         secret = reveal(img)
         if options.secret_binary != "":
@@ -175,4 +178,4 @@ if __name__ == '__main__':
             with open(options.secret_binary, "w") as f:
                 f.write(data)
         else:
-            print secret
+            print secret
\ No newline at end of file