|
|
@@ -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
|
philemon