Pull common script code into igor.script and add bin/igorpackedexperiment.py.

[igor.git] / igor / script.py

diff --git a/igor/script.py b/igor/script.py
new file mode 100644 (file)
index 0000000..be34b61
--- /dev/null
+++ b/igor/script.py
@@ -0,0 +1,42 @@
+# Copyright
+
+"Common code for scripts distributed with the `igor` package."
+
+from __future__ import absolute_import
+import argparse as _argparse
+import logging as _logging
+import sys as _sys
+
+from . import __version__
+from . import LOG as _LOG
+
+
+class Script (object):
+    log_levels = [_logging.ERROR, _logging.WARNING, _logging.INFO, _logging.DEBUG]
+
+    def __init__(self, description=None, filetype='IGOR Binary Wave (.ibw) file'):
+        self.parser = _argparse.ArgumentParser(
+            description=description, version=__version__)
+        self.parser.add_argument(
+            '-f', '--infile', metavar='FILE', default='-',
+            help='input {}'.format(filetype))
+        self.parser.add_argument(
+            '-o', '--outfile', metavar='FILE', default='-',
+            help='file for ASCII output')
+        self.parser.add_argument(
+            '-V', '--verbose', action='count', default=0,
+            help='increment verbosity')
+
+    def run(self, *args, **kwargs):
+        args = self.parser.parse_args(*args, **kwargs)
+        if args.infile == '-':
+            args.infile = _sys.stdin
+        if args.outfile == '-':
+            args.outfile = _sys.stdout
+        if args.verbose > 1:
+            log_level = self.log_levels[min(args.verbose-1, len(self.log_levels)-1)]
+            _LOG.setLevel(log_level)
+        self._run(args)
+
+    def _run(self, args):
+        raise NotImplementedError()