def setUp(self):
self.engine = create_fixtures()
self.config = Config({})
self.config._engine = self.engine
self.config.setup()
spec = load_config(os.path.join(FIXTURE_PATH, 'spec.yaml'))
self.spec = Spec(self.config, spec)
self.spec._engine = self.engine
self.gen = Generator(self.spec)
def map(ctx, spec_file):
""" Map data from the database into modeled objects. """
try:
config = ctx.obj['CONFIG']
spec = load_config(spec_file)
spec = Spec(config, spec, path=spec_file)
mapper = Mapper(config, spec)
mapper.map()
except LoomException as le:
raise click.ClickException(le.message)
def cli(ctx, debug, config_file):
""" Map data from a SQL database into a statement-based graph. """
ctx.obj = ctx.obj or {}
ctx.obj['DEBUG'] = debug
config = load_config(config_file)
ctx.obj['CONFIG'] = Config(config, path=config_file)
ctx.obj['CONFIG'].setup()
fmt = '[%(levelname)-8s] %(name)-12s: %(message)s'
level = logging.DEBUG if debug else logging.INFO
logging.basicConfig(level=level, format=fmt)
logging.getLogger('requests').setLevel(logging.WARNING)
logging.getLogger('urllib3').setLevel(logging.WARNING)
logging.getLogger('elasticsearch').setLevel(logging.WARNING)