Fix error handling

Aqcurate · May 14, 2018 · ba58e37 · ba58e37
1 parent 19358d1
commit ba58e37
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/steglsb.py b/steglsb.py
@@ -38,9 +38,10 @@ def _save_img(self, img, outfile):
  print('[!] {} image could not be written.'.format(outfile))
  print('[!] {}'.format(e))
  sys.exit()
- except KeyError as e:
- print('[!] Format unable to be determined by filename.'.format(outfile))
+ except Exception as e:
+ print('[!] Unable to save file.')
  print('[!] {}'.format(e))
+ sys.exit()
 
 class LSBEncode(LSB):
  def __init__(self, cover, secret, bits, outfile, mode=None):