default_value = column_desc[4]
if default_value != "":
return default_value
return None
# Args check
if len(sys.argv) != 2:
print("Usage: python3 schema-to-yaml.py schema_name")
sys.exit()
# Connect to database
db = Database(sys.argv[1])
# Get table list
with open("schema.yaml", "w") as out:
for table in db.get_table_list():
table_name = table[0]
# Table name
out.write(table_name.split("_")[1] + ":\n")
out.write(" lock_attributes: true\n")
# Attributes
out.write(" attributes:\n")
table_desc = db.get_table_definition(table_name)
for column_desc in table_desc:
# Skip "_id"
field_name = get_field_name(column_desc)
