summaryrefslogtreecommitdiff
path: root/test/Fortran/fixture/myfortran_flags.py
blob: 2b433eaaaa44ed821c3e12dcd5e7e5c2db1a6d5e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import getopt
import sys

comment = ('#' + sys.argv[1]).encode()

opts, args = getopt.getopt(sys.argv[2:], 'cf:o:xyz')
optstring = ''
length = len(comment)
for opt, arg in opts:
    if opt == '-o': out = arg
    elif opt not in ('-f', '-K'): optstring = optstring + ' ' + opt

with open(args[0], 'rb') as infile, open(out, 'wb') as outfile:
    outfile.write((optstring + "\n").encode())
    for l in infile:
        if not l.startswith(comment):
            outfile.write(l)

sys.exit(0)