SSnR.py

@@ -1,4 +1,4 @@
-#!/usr/bin/env python3
+#!/usr/bin/env python
 # -*- coding: utf-8 -*-
 #
 #  SSnR.py
@@ -33,24 +33,21 @@ def main():
 	parser = argparse.ArgumentParser(description='Search and replace tool', prog='SSnR')
 	parser.add_argument('-e', '--regex', help='Regex', required=True)
 	parser.add_argument('-s', '--string', help='String', required=False)
-	parser.add_argument('-i', '--input', help='Input file', required=False)
-	parser.add_argument('-o', '--output', help='Output file', required=False)
+	parser.add_argument('-i', '--input', help='Input file', required=False, type=open)
 	parser.add_argument('-r', '--replace', help='Replace', required=False)
 	parser.add_argument('-m', '--print_nb_match', help='Print the number of match in replace',
 	                    required=False, action='store_true')
 	parser.add_argument('-c', '--ignore_case', help='Ignore the case',
 	                    required=False, action='store_true')
-
+	try:
 		args = vars(parser.parse_args())
+	except FileNotFoundError as exception:
+		print("Error: file not found: " + str(exception))
+		return -1
 
 	# Get input
 	if args["input"] is not None:
-		try:
-			with open(args["input"], "r") as file:
-				string = file.read()
-		except OSError as exception:
-			print("Error: file not found: " + str(exception))
-			return -1
+		string = args["input"].read()
 		is_file = True
 	elif args["string"] is not None:
 		string = args["string"]
@@ -59,17 +56,6 @@ def main():
 		print("Error: You need an input string or file")
 		return -1
 
-	# Get output
-	if args["output"] is not None:
-		try:
-			output_file = open(args["output"], "w")
-		except OSError as exception:
-			print("Error: file not found: " + str(exception))
-			return -1
-		use_output_file = True
-	else:
-		use_output_file = False
-
 	# Compile regex
 	try:
 		ex = compile_regex(args["regex"])
@@ -82,11 +68,7 @@ def main():
 
 	# Search or replace
 	if args["replace"] is not None:
-		replace_string = replace(ex, string, args["replace"], args["print_nb_match"])
-		if use_output_file:
-			output_file.write(replace_string)
-		else:
-			print(replace_string)
+		replace(ex, string, args["replace"], args["print_nb_match"])
 	else:
 		search(ex, string, is_file)
 	return 0
@@ -134,9 +116,9 @@ def replace(ex, string, replace_string, print_nb):
 		:param print_nb: Print the number of match
 	"""
 	res = ex.subn(replace_string, string)
+	print(res[0])
 	if print_nb:
 		print("Number of match: " + str(res[1]))
-	return res[0]
 
 def get_line_pos(string):
 	"""