mirror of
https://github.com/rwinkhart/sshyp.git
synced 2026-09-04 08:07:15 -04:00
port-jobs: No longer remove shebang with COMMENTS.py
Former-commit-id: 7b67fcdcfa2993ef7967f2300b979c321e5966c2 Former-commit-id: 0e88fa41dcc7dfa0f1d69ebefa48c5f5d56739cc
This commit is contained in:
@@ -13,12 +13,17 @@ for filename in listdir('working'):
|
|||||||
# read input file
|
# read input file
|
||||||
with open(filepath, 'r') as file:
|
with open(filepath, 'r') as file:
|
||||||
new_file = []
|
new_file = []
|
||||||
|
first_line = True
|
||||||
for line in file:
|
for line in file:
|
||||||
# determine whether to remove all or only PORT comments
|
# determine whether to remove all or only PORT comments
|
||||||
if len(arguments) > 0 and arguments[0] == 'ALL':
|
if first_line:
|
||||||
if not rematch(r'^\s*#', line):
|
|
||||||
new_file.append(line)
|
|
||||||
elif not rematch(r'^\s*# PORT', line):
|
|
||||||
new_file.append(line)
|
new_file.append(line)
|
||||||
|
first_line = False
|
||||||
|
else:
|
||||||
|
if len(arguments) > 0 and arguments[0] == 'ALL':
|
||||||
|
if not rematch(r'^\s*#', line):
|
||||||
|
new_file.append(line)
|
||||||
|
elif not rematch(r'^\s*# PORT', line):
|
||||||
|
new_file.append(line)
|
||||||
# write the updated file contents
|
# write the updated file contents
|
||||||
open(filepath, 'w').writelines(new_file)
|
open(filepath, 'w').writelines(new_file)
|
||||||
|
|||||||
Reference in New Issue
Block a user