Mercurial > hg > tvii

annotate tvii/cli.py @ 84:0f3af15bb29a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
add noise introduction function
author Jeff Hammel <k0scist@gmail.com>
date Sun, 17 Dec 2017 13:55:35 -0800
parents a17d43d7bf1b
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
75
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
1 """
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
2 Command-Line Interface support
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
3 """
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
4
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
5 import argparse
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
6 import os
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
7
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
8
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
9 class CLIParser(argparse.ArgumentParser):
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
10 """ABC for CLI parsing"""
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
11
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
12 def __init__(self, **kwargs):
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
13 argparse.ArgumentParser.__init__(self, **kwargs)
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
14 self.add_arguments()
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
15 self.options = None
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
16
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
17 def add_arguments(self):
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
18 self.add_argument('-v', '--verbose', dest='verbose',
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
19 action='store_true', default=False,
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
20 help="be more verbose")
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
21
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
22 def parse_args(self, args):
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
23
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
24 options = argparse.ArgumentParser.parse_args(self, args)
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
25 self.options = self.validate(options)
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
26 return self.options
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
27
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
28 def validate(self, options):
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
29
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
30 if not getattr(options, 'verbose', False):
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
31 # https://github.com/tensorflow/tensorflow/issues/7778
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
32 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
33
a17d43d7bf1b add simple CLI abstraction
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
34 return options