33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
from c_comment import generateAllComments as c_generateAllComments
|
|
from py_comment import generateAllComments as py_generateAllComments
|
|
import sys
|
|
|
|
def fileExists(filename: str) -> bool:
|
|
try:
|
|
with open(filename, 'r'):
|
|
return True
|
|
except FileNotFoundError:
|
|
return False
|
|
|
|
if __name__ == '__main__':
|
|
if sys.argv[1] == '-h' or sys.argv[1] == "-H" or sys.argv[1] == '--help' or sys.argv[1] == 'help':
|
|
print('ComGen: A comment generator for C and Python files')
|
|
print('Usage: python3 comgen.py [path/to/file1] [path/to/file2] ...')
|
|
sys.exit()
|
|
elif sys.argv[1] == '-v' or sys.argv[1] == "-V" or sys.argv[1] == '--version' or sys.argv[1] == 'version':
|
|
print('ComGen version 1.0.0')
|
|
sys.exit()
|
|
filenames = sys.argv[1:]
|
|
for filename in filenames:
|
|
if not fileExists(filename):
|
|
print('File not found')
|
|
sys.exit()
|
|
extension = filename.split('.')[-1]
|
|
if extension == 'c' or extension == 'h':
|
|
c_generateAllComments(filename)
|
|
elif extension == 'py':
|
|
py_generateAllComments(filename)
|
|
else:
|
|
print('Language not supported')
|
|
sys.exit()
|