unrhodecode/rhodecode/lib/dbmigrate/migrate/changeset/util.py
2025-01-13 17:47:37 +01:00

20 lines
488 B
Python
Executable file

"""
Safe quoting method
"""
from rhodecode.lib.dbmigrate.migrate.changeset import SQLA_10
def fk_column_names(constraint):
if SQLA_10:
return [constraint.columns[key].name for key in constraint.column_keys]
else:
return [element.parent.name for element in constraint.elements]
def safe_quote(obj):
# this is the SQLA 0.9 approach
if hasattr(obj, "name") and hasattr(obj.name, "quote"):
return obj.name.quote
else:
return obj.quote