new file: mmgen/opt/Opts.py

mmgen/opt/Opts.py

@@ -0,0 +1,84 @@
+#!/usr/bin/env python
+#
+# Opts.py, an options parsing library for Python.  Copyright (C) 2014 by
+# Philemon <mmgen-py@yandex.com>.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import sys, getopt
+
+def usage(hd):
+	print "USAGE: %s %s" % (hd['prog_name'], hd['usage'])
+	sys.exit(2)
+
+def print_help(help_data):
+	pn = help_data['prog_name']
+	pn_len = str(len(pn)+2)
+	print ("  %-"+pn_len+"s %s") % (pn.upper()+":", help_data['desc'].strip())
+	print ("  %-"+pn_len+"s %s %s")%("USAGE:", pn, help_data['usage'].strip())
+	sep = "\n    "
+	print "  OPTIONS:"+sep+"%s" % sep.join(help_data['options'].strip().split("\n"))
+	if "notes" in help_data:
+		print "  %s" % "\n  ".join(help_data['notes'][1:-1].split("\n"))
+
+
+def process_opts(argv,help_data,short_opts,long_opts):
+
+	long_opts  = [i.replace("_","-") for i in long_opts]
+
+	try: cl_opts, args = getopt.getopt(argv[1:], short_opts, long_opts)
+	except getopt.GetoptError as err:
+		print str(err); sys.exit(2)
+
+	opts,short_opts_l = {},[]
+
+	for i in short_opts:
+		if i == ":": short_opts_l[-1] += i
+		else:        short_opts_l     += i
+
+	for opt, arg in cl_opts:
+		if   opt in ("-h","--help"): print_help(help_data); sys.exit()
+		elif opt[:2] == "--" and opt[2:] in long_opts:
+			opts[opt[2:].replace("-","_")] = True
+		elif opt[:2] == "--" and opt[2:]+"=" in long_opts:
+			opts[opt[2:].replace("-","_")] = arg
+		elif opt[0] == "-" and opt[1]     in short_opts_l:
+			opts[long_opts[short_opts_l.index(opt[1:])].replace("-","_")] = True
+		elif opt[0] == "-" and opt[1:]+":" in short_opts_l:
+			opts[long_opts[short_opts_l.index(
+					opt[1:]+":")][:-1].replace("-","_")] = arg
+		else: assert False, "Invalid option"
+
+	return opts,args
+
+
+def parse_opts(argv,help_data):
+
+	lines = help_data['options'].strip().split("\n")
+	import re
+	pat = r"^-([a-zA-Z0-9]), --([a-zA-Z0-9-]{1,64})(=| )(.+)"
+	rep = r"-{0}, --{1}{w}{3}"
+	opt_data = [list(m.groups()) for m in [re.match(pat,l) for l in lines] if m]
+
+	for d in opt_data:
+		if d[2] == " ": d[2] = ""
+	short_opts = "".join([d[0]+d[2].replace("=",":") for d in opt_data])
+	long_opts = [d[1].replace("-","_")+d[2] for d in opt_data]
+	help_data['options'] = "\n".join(
+		[rep.format(w=" ", *m.groups())
+			if m else k for m,k in [(re.match(pat,l),l) for l in lines]]
+	)
+	opts,args = process_opts(argv,help_data,short_opts,long_opts)
+
+	return opts,args,short_opts,long_opts