def test_sqlalchemy(self):
tbl = Table(self.merovingians)
generated = tbl.sqlalchemy()
self.assertIn("Column('reign_from'", generated)
self.assertIn("Integer()", generated)
tbl = Table(self.canada)
generated = tbl.sqlalchemy()
self.assertIn("Column('capital', Unicode", generated)
def test_sqlalchemy(self):
tbl = Table(self.merovingians)
generated = tbl.sqlalchemy()
self.assertIn("Column('reign_from'", generated)
self.assertIn("Integer()", generated)
tbl = Table(self.canada)
generated = tbl.sqlalchemy()
self.assertIn("Column('capital', Unicode", generated)
def generate_one(tbl, args, table_name=None, file=None):
"""
Prints code (SQL, SQLAlchemy, etc.) to define a table.
"""
table = Table(tbl,
table_name=table_name,
varying_length_text=args.text,
uniques=args.uniques,
pk_name=args.key,
force_pk=args.force_key,
reorder=args.reorder,
data_size_cushion=args.cushion,
save_metadata_to=args.save_metadata_to,
metadata_source=args.use_metadata_from,
loglevel=args.log,
limit=args.limit)
if args.dialect.startswith('sqla'):
if not args.no_creates:
print(table.sqlalchemy(), file=file)
if args.inserts:
print("\n".join(table.inserts(dialect=args.dialect)), file=file)
elif args.dialect.startswith('dj'):
table.django_models()
else:
print(table.sql(dialect=args.dialect,
inserts=args.inserts,
creates=(not args.no_creates),
drops=args.drops,
metadata_source=args.use_metadata_from),
file=file)
return table
def generate_one(datafile, args, table_name=None):
table = Table(datafile, table_name=table_name, varying_length_text=args.text, uniques=args.uniques,
pk_name = args.key, force_pk=args.force_key, reorder=args.reorder, data_size_cushion=args.cushion,
save_metadata_to=args.save_metadata_to, metadata_source=args.use_metadata_from,
loglevel=args.log, limit=args.limit)
if args.dialect.startswith('sqla'):
print(table.sqlalchemy())
if args.inserts:
print("\n".join(table.inserts(dialect=args.dialect)))
#inserter.compile().bindtemplate
elif args.dialect.startswith('dj'):
table.django_models()
else:
print(table.sql(dialect=args.dialect, inserts=args.inserts,
creates=(not args.no_creates), drops=args.drops,
metadata_source=args.use_metadata_from))
def generate_one(tbl, args, table_name=None, file=None):
"""
Prints code (SQL, SQLAlchemy, etc.) to define a table.
"""
table = Table(tbl, table_name=table_name, varying_length_text=args.text, uniques=args.uniques,
pk_name = args.key, force_pk=args.force_key, reorder=args.reorder, data_size_cushion=args.cushion,
save_metadata_to=args.save_metadata_to, metadata_source=args.use_metadata_from,
loglevel=args.log, limit=args.limit)
if args.dialect.startswith('sqla'):
if not args.no_creates:
print(table.sqlalchemy(), file=file)
if args.inserts:
print("\n".join(table.inserts(dialect=args.dialect)), file=file)
elif args.dialect.startswith('dj'):
table.django_models()
else:
print(table.sql(dialect=args.dialect, inserts=args.inserts,
creates=(not args.no_creates), drops=args.drops,
metadata_source=args.use_metadata_from), file=file)
return table
def generate():
args = read_args()
set_logging(args)
logging.info(str(args))
if args.dialect in ('pg', 'pgsql', 'postgres'):
args.dialect = 'postgresql'
if args.dialect not in dialect_names:
raise NotImplementedError('First arg must be one of: %s' % ", ".join(dialect_names))
for datafile in args.datafile:
table = Table(datafile, varying_length_text=args.text, uniques=args.uniques,
pk_name = args.key, force_pk=args.force_key, reorder=args.reorder,
save_metadata_to=args.save_metadata_to, metadata_source=args.use_metadata_from,
loglevel=args.log, limit=args.limit)
if args.dialect.startswith('sqla'):
print(table.sqlalchemy())
if args.inserts:
print("\n".join(table.inserts(dialect=args.dialect)))
#inserter.compile().bindtemplate
else:
print(table.sql(dialect=args.dialect, inserts=args.inserts,
creates=(not args.no_creates), drops=args.drops,
metadata_source=args.use_metadata_from))
