def handle(self, **options):
try:
schema_tool = SchemaTool()
except UnsupportedDBError as e:
raise CommandError(e)
result = SchemaDump()
if options['tables']:
result.set_table_list(schema_tool.get_tables())
self.stdout.write(str(result))
else:
result.load({
'defaults': schema_tool.get_defaults()})
for app_label in schema_tool.app_configs:
app_dump = SchemaAppDump(app_label)
table_names = schema_tool.get_app_tables(app_label)
for table_name in table_names:
table_dump = SchemaTableDump(table_name)
table_dump.load({
'fields': schema_tool.get_table_fields(table_name),
'indices': schema_tool.get_table_indices(table_name),
'constraints':
schema_tool.get_table_constraints(table_name)})
app_dump.add_table(table_dump)
if table_names:
result.add_app(app_dump)
self.stdout.write(str(result))
def handle(self, **options):
try:
schema_tool = SchemaTool()
except UnsupportedDBError as e:
raise CommandError(e)
result = SchemaDump()
if options['tables']:
result.set_table_list(schema_tool.get_tables())
self.stdout.write(str(result))
else:
result.load({'defaults': schema_tool.get_defaults()})
for app_label in schema_tool.app_configs:
app_dump = SchemaAppDump(app_label)
table_names = schema_tool.get_app_tables(app_label)
for table_name in table_names:
table_dump = SchemaTableDump(table_name)
table_dump.load({
'fields':
schema_tool.get_table_fields(table_name),
'indices':
schema_tool.get_table_indices(table_name),
'constraints':
schema_tool.get_table_constraints(table_name)
})
app_dump.add_table(table_dump)
if table_names:
result.add_app(app_dump)
self.stdout.write(str(result))
def handle(self, **options):
schema_tool = SchemaTool()
if options['tables']:
self.stdout.write(jsonify(schema_tool.get_tables()))
else:
self.stdout.write(
jsonify(schema_tool.get_defaults())
)
def test_schema(capfd):
if get_current_db_type() != 'mysql':
pytest.skip("unsupported database")
call_command('schema')
out, err = capfd.readouterr()
schema_tool = SchemaTool()
result = SchemaDump()
result.load(json.loads(out))
assert result.defaults == schema_tool.get_defaults()
def test_schema_tool():
if get_current_db_type() != 'mysql':
pytest.skip("unsupported database")
schema_tool = SchemaTool()
defaults = schema_tool.get_defaults()
assert set(defaults.keys()) == TEST_MYSQL_SCHEMA_PARAM_NAMES['defaults']
for app_label in schema_tool.app_configs:
for table in schema_tool.get_app_tables(app_label):
row = schema_tool.get_table_fields(table).values()[0]
assert (set([x.lower() for x in row.keys()]).issubset(
TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['fields']))
row = schema_tool.get_table_indices(table).values()[0]
assert (set([x.lower() for x in row.keys()]).issubset(
TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['indices']))
row = schema_tool.get_table_constraints(table).values()[0]
assert (set([x.lower() for x in row.keys()]).issubset(
TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['constraints']))
def test_schema_tool():
schema_tool = SchemaTool()
if isinstance(schema_tool.schema_dumper, MySQLSchemaDumper):
defaults = schema_tool.get_defaults()
assert set(
defaults.keys()) == TEST_MYSQL_SCHEMA_PARAM_NAMES['defaults']
for app_label in schema_tool.app_configs:
for table in schema_tool.get_app_tables(app_label):
row = schema_tool.get_table_fields(table)[0]
assert (set([
x.lower() for x in row.keys()
]) == TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['fields'])
row = schema_tool.get_table_indices(table)[0]
assert (set([
x.lower() for x in row.keys()
]) == TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['indices'])
row = schema_tool.get_table_constraints(table)[0]
assert (set([
x.lower() for x in row.keys()
]) == TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['constraints'])
def test_schema_dump(test_fs):
if get_current_db_type() != 'mysql':
pytest.skip("unsupported database")
schema_tool = SchemaTool()
expected_result = SchemaDump()
with test_fs.open(['data', 'schema.json']) as f:
expected_result.load(data=json.loads(f.read()))
assert expected_result.defaults == schema_tool.get_defaults()
for app_label in schema_tool.app_configs:
expected_app_result = expected_result.get_app(app_label)
for table_name in schema_tool.get_app_tables(app_label):
expected_table_result = expected_app_result.get_table(table_name)
assert (schema_tool.get_table_fields(table_name) ==
expected_table_result.fields)
assert (schema_tool.get_table_indices(table_name) ==
expected_table_result.indices)
assert (schema_tool.get_table_constraints(table_name) ==
expected_table_result.constraints)
def test_schema_dump(test_fs):
if get_current_db_type() != 'mysql':
pytest.skip("unsupported database")
schema_tool = SchemaTool()
expected_result = SchemaDump()
with test_fs.open(['data', 'schema.json']) as f:
expected_result.load(data=json.loads(f.read()))
assert expected_result.defaults == schema_tool.get_defaults()
for app_label in schema_tool.app_configs:
expected_app_result = expected_result.get_app(app_label)
for table_name in schema_tool.get_app_tables(app_label):
expected_table_result = expected_app_result.get_table(table_name)
assert (schema_tool.get_table_fields(table_name) ==
expected_table_result.fields)
assert (schema_tool.get_table_indices(table_name) ==
expected_table_result.indices)
assert (schema_tool.get_table_constraints(table_name) ==
expected_table_result.constraints)
def test_schema_tool():
schema_tool = SchemaTool()
if isinstance(schema_tool.schema_dumper, MySQLSchemaDumper):
defaults = schema_tool.get_defaults()
assert set(defaults.keys()) == TEST_MYSQL_SCHEMA_PARAM_NAMES['defaults']
for app_label in schema_tool.app_configs:
for table in schema_tool.get_app_tables(app_label):
row = schema_tool.get_table_fields(table)[0]
assert (
set([x.lower() for x in row.keys()]) ==
TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['fields']
)
row = schema_tool.get_table_indices(table)[0]
assert (
set([x.lower() for x in row.keys()]) ==
TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['indices']
)
row = schema_tool.get_table_constraints(table)[0]
assert (
set([x.lower() for x in row.keys()]) ==
TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['constraints']
)
def test_schema_tool():
if get_current_db_type() != 'mysql':
pytest.skip("unsupported database")
schema_tool = SchemaTool()
defaults = schema_tool.get_defaults()
assert set(defaults.keys()) == TEST_MYSQL_SCHEMA_PARAM_NAMES['defaults']
for app_label in schema_tool.app_configs:
for table in schema_tool.get_app_tables(app_label):
row = schema_tool.get_table_fields(table).values()[0]
assert (
set([x.lower() for x in row.keys()]).issubset(
TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['fields'])
)
row = schema_tool.get_table_indices(table).values()[0]
assert (
set([x.lower() for x in row.keys()]).issubset(
TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['indices'])
)
row = schema_tool.get_table_constraints(table).values()[0]
assert (
set([x.lower() for x in row.keys()]).issubset(
TEST_MYSQL_SCHEMA_PARAM_NAMES['tables']['constraints'])
)
def handle(self, **options):
schema_tool = SchemaTool()
if options['tables']:
self.stdout.write(jsonify(schema_tool.get_tables()))
else:
self.stdout.write(jsonify(schema_tool.get_defaults()))
def test_schema(capfd):
call_command('schema')
out, err = capfd.readouterr()
schema_tool = SchemaTool()
assert jsonify(schema_tool.get_defaults()) in out
