summaryrefslogtreecommitdiff
path: root/migrate/changeset/util.py
blob: f8ecd817c77d0562e23f6b2ad0486af5a090d19e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from migrate.changeset import SQLA_10

"""
Safe quoting method
"""

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


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]