mirror of
https://github.com/veekun/pokedex.git
synced 2024-08-20 18:16:34 +00:00
Added command_ prefix to CLI commands to fix import problems.
This commit is contained in:
parent
b13ffac247
commit
d4077cc71d
1 changed files with 7 additions and 7 deletions
|
@ -15,14 +15,14 @@ def main():
|
||||||
args = sys.argv[2:]
|
args = sys.argv[2:]
|
||||||
|
|
||||||
# Find the command as a function in this file
|
# Find the command as a function in this file
|
||||||
func = globals().get(command, None)
|
func = globals().get("command_%s" % command, None)
|
||||||
if func and callable(func) and command != 'main':
|
if func:
|
||||||
func(*args)
|
func(*args)
|
||||||
else:
|
else:
|
||||||
help()
|
command_help()
|
||||||
|
|
||||||
|
|
||||||
def csvimport(engine_uri, directory='.'):
|
def command_csvimport(engine_uri, directory='.'):
|
||||||
import csv
|
import csv
|
||||||
|
|
||||||
from sqlalchemy.orm.attributes import instrumentation_registry
|
from sqlalchemy.orm.attributes import instrumentation_registry
|
||||||
|
@ -129,7 +129,7 @@ def csvimport(engine_uri, directory='.'):
|
||||||
else:
|
else:
|
||||||
print 'loaded'
|
print 'loaded'
|
||||||
|
|
||||||
def csvexport(engine_uri, directory='.'):
|
def command_csvexport(engine_uri, directory='.'):
|
||||||
import csv
|
import csv
|
||||||
session = connect(engine_uri)
|
session = connect(engine_uri)
|
||||||
|
|
||||||
|
@ -160,7 +160,7 @@ def csvexport(engine_uri, directory='.'):
|
||||||
|
|
||||||
writer.writerow(csvs)
|
writer.writerow(csvs)
|
||||||
|
|
||||||
def lookup(engine_uri, name):
|
def command_lookup(engine_uri, name):
|
||||||
# XXX don't require uri! somehow
|
# XXX don't require uri! somehow
|
||||||
session = connect(engine_uri)
|
session = connect(engine_uri)
|
||||||
|
|
||||||
|
@ -174,7 +174,7 @@ def lookup(engine_uri, name):
|
||||||
print object.__tablename__, object.name
|
print object.__tablename__, object.name
|
||||||
|
|
||||||
|
|
||||||
def help():
|
def command_help():
|
||||||
print u"""pokedex -- a command-line Pokédex interface
|
print u"""pokedex -- a command-line Pokédex interface
|
||||||
|
|
||||||
help Displays this message.
|
help Displays this message.
|
||||||
|
|
Loading…
Reference in a new issue