Commit 7994b5d5 authored by Cosmia Fu's avatar Cosmia Fu

Add SQLAlchemy v0.7.x compatibility

parent 1993309a
from sqlalchemy import tuple_, or_, and_, inspect from sqlalchemy import tuple_, or_, and_
from sqlalchemy.sql.operators import eq from sqlalchemy.sql.operators import eq
from sqlalchemy.exc import DBAPIError from sqlalchemy.exc import DBAPIError
from ast import literal_eval from ast import literal_eval
...@@ -36,7 +36,7 @@ def get_primary_key(model): ...@@ -36,7 +36,7 @@ def get_primary_key(model):
:param model: :param model:
Model class Model class
""" """
mapper = inspect(model) mapper = model._sa_class_manager.mapper
pks = [mapper.get_property_by_column(c).key for c in mapper.primary_key] pks = [mapper.get_property_by_column(c).key for c in mapper.primary_key]
if len(pks) == 1: if len(pks) == 1:
return pks[0] return pks[0]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment