mirror of
https://github.com/rwinkhart/sshyp.git
synced 2026-08-29 13:26:30 -04:00
RandyTheSilly
1c68226ab9
Former-commit-id: 7b67fcdcfa2993ef7967f2300b979c321e5966c2 Former-commit-id: 0e88fa41dcc7dfa0f1d69ebefa48c5f5d56739cc
30 lines
1.0 KiB
Python
Executable File
30 lines
1.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
from os import listdir
|
|
from re import match as rematch
|
|
from sys import argv
|
|
|
|
# read arguments
|
|
arguments = argv[1:]
|
|
|
|
# loop through all Python files in the working directory
|
|
for filename in listdir('working'):
|
|
if filename.endswith('.py'):
|
|
filepath = 'working/' + filename
|
|
# 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 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)
|