SqlAlchemy отношение к конкретным столбцам - PullRequest
Новая
       23

SqlAlchemy отношение к конкретным столбцам

3 голосов
/ 31 января 2012

Допустим, у меня есть модель SqlAlchemy, примерно такая: 

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, String, Integer, ForeignKey
from sqlalchemy.orm import sessionmaker, relationship
Base = declarative_base()
Session = sessionmaker()

class EmployeeType(Base):
    __tablename__ = 'employee_type'
    id = Column(Integer(), primary_key=True)
    name = Column(String(20))

class Employee(Base):
    __tablename__ = 'employee'
    id = Column(Integer(), primary_key=True)
    type_id = Column(Integer(), ForeignKey(EmployeeType.id))
    type = relationship(EmployeeType, uselist=False)

session = Session()
session.add(EmployeeType(name='drone'))
session.add(EmployeeType(name='PHB'))

Мне бы хотелось иметь какое-то "отношение" от Employee напрямую к EmployeeType.name для удобства, поэтому я могу пропуститьшаг поиска идентификатора или объекта EmployeeType, если у меня есть имя типа: 

emp = Employee()
emp.type_name = "drone"
session.add(emp)
session.commit()
assert (emp.type.id == 1)

Возможно ли такое?

РЕДАКТИРОВАТЬ: я обнаружил, что association_proxy может получить меня там частично1009 * 

class Employee(Base):
    ...
    type_name = association_proxy("type", "name")

единственная проблема заключается в том, что если я назначу ему: 

emp = session.query(Employee).filter_by(EmployeeType.name=='PHB').first()
emp.type_name = 'drone'

, то он изменяет столбец employee_type.name, а не столбец employee.type_id.

Ответы [ 2 ]

3 голосов
/ 06 февраля 2012

Я согласен с общим подходом Джонатана, но я чувствую, что добавление объекта сотрудника в сеанс и установка типа сотрудника должны быть независимыми операциями.Вот реализация, которая имеет type_name в качестве свойства и требует добавления к сеансу перед его установкой: 

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, String, Integer, ForeignKey
from sqlalchemy.orm import sessionmaker, relationship
Base = declarative_base()
Session = sessionmaker()

class EmployeeType(Base):
    __tablename__ = 'employee_type'
    id = Column(Integer(), primary_key=True)
    name = Column(String(20))

class Employee(Base):
    __tablename__ = 'employee'
    id = Column(Integer(), primary_key=True)
    type_id = Column(Integer(), ForeignKey(EmployeeType.id))
    type = relationship(EmployeeType)

    @property
    def type_name(self):
        if self.type is not None:
            return self.type.name
        return None

    @type_name.setter
    def type_name(self, value):
        if value is None:
            self.type = None
        else:
            session = Session.object_session(self)
            if session is None:
                raise Exception("Can't set Employee type by name until added to session")
            self.type = session.query(EmployeeType).filter_by(name=value).one()
2 голосов
/ 01 февраля 2012

Я бы сделал это, создав метод, который сделает это для меня. 

class EmployeeType(Base):
    __tablename__ = 'employee_type'
    id = Column(Integer(), primary_key=True)
    name = Column(String(20))

class Employee(Base):
    __tablename__ = 'employee'
    id = Column(Integer(), primary_key=True)
    type_id = Column(Integer(), ForeignKey(EmployeeType.id))
    type = relationship(EmployeeType, uselist=False)

    def __init__(self, type):
        self.type = type

    def add(self, type_name=None):
        if type_name is not None:
            emp_type = DBSession.query(EmployeeType).filter(EmployeeType.name == type_name).first()
            if emp_type:
                type = emp_type
            else:
                type = EmployeeType(name=type_name)
        else:
            type = None
        DBSession.add(Employee(type=type))

Тогда вы делаете: 

Employee.add(type_name='boss')
...