diff --git a/sqlalchemy_mptt/mixins.py b/sqlalchemy_mptt/mixins.py index 39d666f..2e9a386 100644 --- a/sqlalchemy_mptt/mixins.py +++ b/sqlalchemy_mptt/mixins.py @@ -3,6 +3,7 @@ # vim:fenc=utf-8 # # Copyright © 2014 uralbash +# Copyright © 2016 Jiri Kuncar # # Distributed under terms of the MIT license. @@ -43,12 +44,13 @@ class Tree(Base, BaseNestedSets): def __repr__(self): return "" % self.id """ + @declared_attr def __table_args__(cls): return ( - Index('%s_lft_idx' % cls.__tablename__, "lft"), - Index('%s_rgt_idx' % cls.__tablename__, "rgt"), - Index('%s_level_idx' % cls.__tablename__, "level"), + Index('%s_lft_idx' % cls.__tablename__, cls.left.name), + Index('%s_rgt_idx' % cls.__tablename__, cls.right.name), + Index('%s_level_idx' % cls.__tablename__, cls.level.name), ) @classmethod