mirror of
https://github.com/rwinkhart/sshyp.git
synced 2026-08-29 21:36:31 -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
|
||||
with open(filepath, 'r') as file:
|
||||
new_file = []
|
||||
first_line = True
|
||||
for line in file:
|
||||
# determine whether to remove all or only PORT comments
|
||||
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):
|
||||
if first_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
|
||||
open(filepath, 'w').writelines(new_file)
|
||||
|
||||
Reference in New Issue
Block a user