pokedex/__init__.py

   1 # encoding: utf8
   2 from optparse import OptionParser
   3 import sys
   4
   5 from .db import connect, metadata
   6 import pokedex.db.load
   7 from pokedex.lookup import lookup as pokedex_lookup
   8
   9 def main():
  10     if len(sys.argv) <= 1:
  11         command_help()
  12
  13     command = sys.argv[1]
  14     args = sys.argv[2:]
  15
  16     # Find the command as a function in this file
  17     func = globals().get("command_%s" % command, None)
  18     if func:
  19         func(*args)
  20     else:
  21         command_help()
  22
  23
  24 def command_dump(*args):
  25     parser = OptionParser()
  26     parser.add_option('-e', '--engine', dest='engine_uri', default=None)
  27     parser.add_option('-d', '--directory', dest='directory', default=None)
  28     parser.add_option('-q', '--quiet', dest='verbose', default=True, action='store_false')
  29     options, _ = parser.parse_args(list(args))
  30
  31     session = connect(options.engine_uri)
  32     pokedex.db.load.dump(session, directory=options.directory,
  33                                   verbose=options.verbose)
  34
  35 def command_load(*args):
  36     parser = OptionParser()
  37     parser.add_option('-e', '--engine', dest='engine_uri', default=None)
  38     parser.add_option('-d', '--directory', dest='directory', default=None)
  39     parser.add_option('-D', '--drop-tables', dest='drop_tables', default=False, action='store_true')
  40     parser.add_option('-q', '--quiet', dest='verbose', default=True, action='store_false')
  41     options, _ = parser.parse_args(list(args))
  42
  43     session = connect(options.engine_uri)
  44
  45     pokedex.db.load.load(session, directory=options.directory,
  46                                   drop_tables=options.drop_tables,
  47                                   verbose=options.verbose)
  48
  49
  50 def command_lookup(engine_uri, name):
  51     # XXX don't require uri!  somehow
  52     session = connect(engine_uri)
  53
  54     results, exact = pokedex_lookup(session, name)
  55     if exact:
  56         print "Matched:"
  57     else:
  58         print "Fuzzy-matched:"
  59
  60     for object in results:
  61         print object.__tablename__, object.name
  62
  63
  64 def command_help():
  65     print u"""pokedex -- a command-line Pokédex interface
  66 usage: pokedex {command} [options...]
  67 Run `pokedex setup` first, or nothing will work!
  68
  69 Commands:
  70     help                Displays this message.
  71     lookup [thing]      Look up something in the Pokédex.
  72
  73 System commands:
  74     load                Load Pokédex data into a database from CSV files.
  75     dump                Dump Pokédex data from a database into CSV files.
  76
  77 Options:
  78     -d|--directory      By default, load and dump will use the CSV files in the
  79                         pokedex install directory.  Use this option to specify
  80                         a different directory.
  81     -D|--drop-tables    With load, drop all tables before loading data.
  82     -e|--engine=URI     By default, all commands try to use a SQLite database
  83                         in the pokedex install directory.  Use this option to
  84                         specify an alternate database.
  85     -q|--quiet          Turn off any unnecessary status output from dump/load.
  86 """.encode(sys.getdefaultencoding(), 'replace')
  87
  88     sys.exit(0)