1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# mkheader.py - create a C header file skeleton (#ifndef/#define) from filename
#
# Copyright © 2009 Tobias Klauser <tklauser@distanz.ch>
#
# THE BEER-WARE LICENSE (Revision 42):
# <tklauser@distanz.ch> wrote this file. As long as you retain this notice you
# can do whatever you want with this stuff. If we meet some day, and you think
# this stuff is worth it, you can buy me a beer in return.
# Tobias Klauser
import os, sys
import getopt
import re
def usage():
print """usage: %s [OPTION] FILE...
-e launch editor after creating files. Uses editor from $EDITOR envvar.
-f overwrite existing files
-s print generated to stdout instead of writing it to file
-h show this help and exit""" % (os.path.basename(sys.argv[0]))
def main():
try:
opts, args = getopt.getopt(sys.argv[1:], "efsh")
except getopt.GetoptError, err:
print str(err)
usage()
sys.exit(2)
if len(args) < 1:
usage()
sys.exit(2)
overwrite = False
launch_editor = False
to_stdout = False
for o, a in opts:
if o == "-e":
launch_editor = True
elif o == "-f":
overwrite = True
elif o == "-s":
to_stdout = True
elif o == "-h":
usage()
sys.exit()
else:
assert False, "unhandled option"
if to_stdout and launch_editor:
print "Options -e and -s cannot be used together"
sys.exit(2)
p = re.compile('[a-zA-Z0-9._-]+\.h')
for f in args:
if not p.match(f):
print "Invalid header filename '%s'. Header not created" % f
continue
if not to_stdout and not overwrite and os.path.exists(f):
print "Header file '%s' already exists. Use -f to override." % f
continue
q = re.compile('[.-]')
define = q.sub('_', f.upper())
if to_stdout:
fd = sys.stdout
else:
print "Creating %s..." % f
fd = open(f, 'w')
fd.write('#ifndef _%s_\n' % define)
fd.write('#define _%s_\n\n' % define)
fd.write('/* TODO: Your code here */\n\n')
fd.write('#endif /* _%s_ */' % define)
if not to_stdout:
fd.close()
if launch_editor:
editor = os.getenv("EDITOR")
if editor is None:
editor = "vi"
# Fork an editor for each file as not all editors support
# editing multiple files at once (e.g. splitted view in vim)
for f in args:
print "Editing header '%s' with '%s'" % (f, editor)
pid = os.fork()
if pid == 0: # child process
os.execvp(editor, [editor, f])
sys.exit(0)
else: # parent process
os.waitpid(pid, 0)
if __name__ == '__main__':
main()
|
