mkheader.py: Allow to print file to stdout

This is needed for vim integration.

1 files changed, 21 insertions, 9 deletions

diff --git a/mkheader.py b/mkheader.py
index 89c3620..dcd374e 100755
--- a/mkheader.py
+++ b/mkheader.py
@@ -9,11 +9,12 @@ def usage():
 
  -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:], "efh")
+		opts, args = getopt.getopt(sys.argv[1:], "efsh")
 	except getopt.GetoptError, err:
 		print str(err)
 		usage()
@@ -25,36 +26,47 @@ def main():
 
 	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 overwrite and os.path.exists(f):
+		if not to_stdout and not overwrite and os.path.exists(f):
 			print "Header file '%s' already exists. Use -f to override." % f
 			continue
 
-		print "Creating %s..." % f
 		q = re.compile('[.-]')
 		define = q.sub('_', f.upper())
-		f = open(f, 'w')
-		f.write('#ifndef _%s_\n' % define)
-		f.write('#define _%s_\n\n' % define)
-		f.write('/* TODO: Your code here */\n\n')
-		f.write('#endif /* _%s_ */' % define)
-		f.close()
+		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")