def choose_engine(opts, choice=True):
"""Prompts the user to select a database engine"""
if "engine" in list(opts.keys()):
enginename = opts["engine"]
elif opts["command"] == "download":
enginename = "download"
else:
if not choice:
return None
print("Choose a database engine:")
for engine in engine_list:
if engine.abbreviation:
abbreviation = "(" + engine.abbreviation + ") "
else:
abbreviation = ""
print(" " + abbreviation + engine.name)
enginename = input(": ")
enginename = enginename.lower()
engine = Engine()
if not enginename:
engine = engine_list[0]
else:
for thisengine in engine_list:
if (enginename == thisengine.name.lower() or thisengine.abbreviation and
enginename == thisengine.abbreviation):
engine = thisengine
engine.opts = opts
return engine
def choose_engine(opts, choice=True):
"""Prompts the user to select a database engine"""
if "engine" in list(opts.keys()):
enginename = opts["engine"]
elif opts["command"] == "download":
enginename = "download"
else:
if not choice:
return None
print("Choose a database engine:")
for engine in engine_list:
if engine.abbreviation:
abbreviation = "(" + engine.abbreviation + ") "
else:
abbreviation = ""
print(" " + abbreviation + engine.name)
enginename = input(": ")
enginename = enginename.lower()
engine = Engine()
if not enginename:
engine = engine_list[0]
else:
for thisengine in engine_list:
if (enginename == thisengine.name.lower() or thisengine.abbreviation and
enginename == thisengine.abbreviation):
engine = thisengine
if 'table_name' in opts:
if opts['table_name'] and "{table}" not in opts['table_name'] or "{db}" not in opts['table_name']:
for opt in engine.required_opts:
if opt[0] == 'table_name':
raise Exception('Accepted Table format {fom}'.format(fom=opt[2]))
engine.opts = opts
return engine
def choose_engine(opts, choice=True):
"""Prompts the user to select a database engine"""
if "engine" in list(opts.keys()):
enginename = opts["engine"]
elif opts["command"] == "download":
enginename = "download"
else:
if not choice:
return None
print("Choose a database engine:")
for engine in engine_list:
if engine.abbreviation:
abbreviation = "(" + engine.abbreviation + ") "
else:
abbreviation = ""
print(" " + abbreviation + engine.name)
enginename = input(": ")
enginename = enginename.lower()
engine = Engine()
if not enginename:
engine = engine_list[0]
else:
for thisengine in engine_list:
if (enginename == thisengine.name.lower()
or thisengine.abbreviation
and enginename == thisengine.abbreviation):
engine = thisengine
if 'table_name' in opts:
if opts['table_name'] \
and "{table}" not in opts['table_name'] \
or "{db}" not in opts['table_name']:
for opt in engine.required_opts:
if opt[0] == 'table_name':
raise Exception('Accepted Table format '
'{fom}'.format(fom=opt[2]))
engine.opts = opts
return engine
import os
from StringIO import StringIO
from retriever.lib.engine import Engine
from retriever.lib.table import Table
from retriever.lib.templates import BasicTextTemplate
from retriever.lib.tools import getmd5
from retriever import DATA_WRITE_PATH
from nose.tools import with_setup
# Create simple engine fixture
test_engine = Engine()
test_engine.table = Table("test")
test_engine.script = BasicTextTemplate(tables={'test':test_engine.table},
shortname='test')
test_engine.opts = {'database_name': '{db}_abc'}
HOMEDIR = os.path.expanduser('~')
def test_auto_get_columns():
"""Basic test of getting column labels from header"""
test_engine.table.delimiter = ","
columns, column_values = test_engine.table.auto_get_columns("a,b,c,d")
assert columns == [['a', None], ['b', None], ['c', None], ['d', None]]
def test_auto_get_columns_cleanup():
"""Test of automatically cleaning up column labels from header"""
test_engine.table.delimiter = ","
columns, column_values = test_engine.table.auto_get_columns("a),b.b,c/c,d___d,group")
assert columns == [['a', None], ['b_b', None], ['c_c', None], ['d_d', None],
['grp', None]]