+ got_from = 'default setting'
+
+ session = pokedex.db.connect(engine_uri)
+
+ if options.verbose:
+ print "Connected to database {engine} (from {got_from})" \
+ .format(engine=session.bind.url, got_from=got_from)
+
+ return session
+
+def get_lookup(options, session=None, recreate=False):
+ """Given a parsed options object, opens the whoosh index and returns a
+ PokedexLookup object.
+ """
+
+ if recreate and not session:
+ raise ValueError("get_lookup() needs an explicit session to regen the index")
+
+ index_dir = options.index_dir
+ got_from = None
+ if index_dir:
+ got_from = 'command line'
+ else:
+ index_dir = os.environ.get('POKEDEX_INDEX_DIR', None)
+ if index_dir:
+ got_from = 'environment'
+ else:
+ index_dir = pkg_resources.resource_filename('pokedex',
+ 'data/whoosh-index')
+ got_from = 'default setting'
+
+ if options.verbose:
+ print "Opened lookup index {index_dir} (from {got_from})" \
+ .format(index_dir=index_dir, got_from=got_from)
+
+ lookup = pokedex.lookup.PokedexLookup(index_dir, session=session)
+
+ if recreate:
+ lookup.rebuild_index()
+
+ return lookup
+
+def get_csv_directory(options):
+ """Prints and returns the csv directory we're about to use."""
+
+ if not options.verbose:
+ return
+
+ if options.directory:
+ csvdir = options.directory
+ got_from = 'command line'
+ else:
+ # This is the same as the db.load default
+ csvdir = pkg_resources.resource_filename('pokedex', 'data/csv')
+ got_from = 'default setting'
+
+ print "Using CSV directory {csvdir} (from {got_from})" \
+ .format(csvdir=csvdir, got_from=got_from)
+
+ return csvdir
+
+
+### Plumbing commands
+
+def command_dump(*args):
+ parser = get_parser(verbose=True)
+ parser.add_option('-d', '--directory', dest='directory', default=None)
+ options, tables = parser.parse_args(list(args))
+
+ session = get_session(options)
+ get_csv_directory(options)
+
+ pokedex.db.load.dump(session, directory=options.directory,
+ tables=tables,
+ verbose=options.verbose)
+
+def command_load(*args):
+ parser = get_parser(verbose=True)
+ parser.add_option('-d', '--directory', dest='directory', default=None)
+ parser.add_option('-D', '--drop-tables', dest='drop_tables', default=False, action='store_true')
+ options, tables = parser.parse_args(list(args))
+
+ if not options.engine_uri:
+ print "WARNING: You're reloading the default database, but not the lookup index. They"
+ print " might get out of sync, and pokedex commands may not work correctly!"
+ print "To fix this, run `pokedex reindex` when this command finishes. Or, just use"
+ print "`pokedex setup` to do both at once."
+ print
+
+ session = get_session(options)
+ get_csv_directory(options)
+
+ pokedex.db.load.load(session, directory=options.directory,
+ drop_tables=options.drop_tables,
+ tables=tables,
+ verbose=options.verbose)
+
+def command_reindex(*args):
+ parser = get_parser(verbose=True)
+ options, _ = parser.parse_args(list(args))
+
+ session = get_session(options)
+ lookup = get_lookup(options, session=session, recreate=True)
+
+ print "Recreated lookup index."
+
+
+def command_setup(*args):
+ parser = get_parser(verbose=False)
+ options, _ = parser.parse_args(list(args))
+
+ options.directory = None
+
+ session = get_session(options)
+ get_csv_directory(options)
+ pokedex.db.load.load(session, directory=None, drop_tables=True,
+ verbose=options.verbose)
+
+ lookup = get_lookup(options, session=session, recreate=True)
+
+ print "Recreated lookup index."
+
+
+def command_status(*args):
+ parser = get_parser(verbose=True)
+ options, _ = parser.parse_args(list(args))
+ options.verbose = True
+ options.directory = None
+
+ # Database, and a lame check for whether it's been inited at least once
+ session = get_session(options)
+ print " - OK! Connected successfully."
+
+ if pokedex.db.tables.Pokemon.__table__.exists(session.bind):
+ print " - OK! Database seems to contain some data."
+ else:
+ print " - WARNING: Database appears to be empty."
+
+ # CSV; simple checks that the dir exists
+ csvdir = get_csv_directory(options)
+ if not os.path.exists(csvdir):
+ print " - ERROR: No such directory!"
+ elif not os.path.isdir(csvdir):
+ print " - ERROR: Not a directory!"
+ else:
+ print " - OK! Directory exists."
+
+ if os.access(csvdir, os.R_OK):
+ print " - OK! Can read from directory."
+ else:
+ print " - ERROR: Can't read from directory!"
+
+ if os.access(csvdir, os.W_OK):
+ print " - OK! Can write to directory."
+ else:
+ print " - WARNING: Can't write to directory! " \
+ "`dump` will not work. You may need to sudo."
+
+ # Index; the PokedexLookup constructor covers most tests and will
+ # cheerfully bomb if they fail
+ lookup = get_lookup(options, recreate=False)
+ print " - OK! Opened successfully."
+
+
+### User-facing commands
+
+def command_lookup(*args):
+ parser = get_parser(verbose=False)
+ options, words = parser.parse_args(list(args))
+
+ name = u' '.join(words)
+
+ session = get_session(options)
+ lookup = get_lookup(options, session=session, recreate=False)
+
+ results = lookup.lookup(name)
+ if not results:
+ print "No matches."
+ elif results[0].exact:
projects / zzz-pokedex.git / blobdiff