dalan/SSnR
1
0
Fork 0
Code Issues Releases Activity

Some improvments

Browse source 
Signed-off-by: Rémi BERTHO <remi.bertho@dalan.fr>
This commit is contained in:
Rémi BERTHO 2017-10-16 22:51:03 +02:00
parent 19e9e23f7e
commit 25d6729e03
Signed by: dalan
GPG key ID: EE3B917931C07B64
1 changed files with 23 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
SSnR.py
View file

@ -33,23 +33,24 @@ def main():
parser = argparse.ArgumentParser(description='Search and replace tool', prog='SSnR') parser = argparse.ArgumentParser(description='Search and replace tool', prog='SSnR')
parser.add_argument('-e', '--regex', help='Regex', required=True) parser.add_argument('-e', '--regex', help='Regex', required=True)
parser.add_argument('-s', '--string', help='String', required=False) parser.add_argument('-s', '--string', help='String', required=False)
parser.add_argument('-i', '--input', help='Input file', required=False, type=open) parser.add_argument('-i', '--input', help='Input file', required=False)
parser.add_argument('-o', '--output', help='Output file', required=False, parser.add_argument('-o', '--output', help='Output file', required=False)
type=argparse.FileType('w'))
parser.add_argument('-r', '--replace', help='Replace', required=False) parser.add_argument('-r', '--replace', help='Replace', required=False)
parser.add_argument('-m', '--print_nb_match', help='Print the number of match in replace', parser.add_argument('-m', '--print_nb_match', help='Print the number of match in replace',
required=False, action='store_true') required=False, action='store_true')
parser.add_argument('-c', '--ignore_case', help='Ignore the case', parser.add_argument('-c', '--ignore_case', help='Ignore the case',
required=False, action='store_true') required=False, action='store_true')
try:
args = vars(parser.parse_args()) args = vars(parser.parse_args())
except FileNotFoundError as exception:
print("Error: file not found: " + str(exception))
return -1
# Get input # Get input
if args["input"] is not None: if args["input"] is not None:
string = args["input"].read() try:
with open(args["input"], "r") as file:
string = file.read()
except OSError as exception:
print("Error: file not found: " + str(exception))
return -1
is_file = True is_file = True
elif args["string"] is not None: elif args["string"] is not None:
string = args["string"] string = args["string"]
@ -58,6 +59,17 @@ def main():
print("Error: You need an input string or file") print("Error: You need an input string or file")
return -1 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 # Compile regex
try: try:
ex = compile_regex(args["regex"]) ex = compile_regex(args["regex"])
@ -71,8 +83,8 @@ def main():
# Search or replace # Search or replace
if args["replace"] is not None: if args["replace"] is not None:
replace_string = replace(ex, string, args["replace"], args["print_nb_match"]) replace_string = replace(ex, string, args["replace"], args["print_nb_match"])
if args["output"] is not None: if use_output_file:
args["output"].write(replace_string) output_file.write(replace_string)
else: else:
print(replace_string) print(replace_string)
else: else: