Как создать атрибут класса и свойство класса в Elixir / SqlAlchemy - PullRequest
0 голосов
/ 13 октября 2011

У меня есть класс:

from sys import stderr
from elixir import *
from types import *

class User(Entity):
    using_options(tablename="users")
    first_name = Field(String(50))
    middle_name = Field(String(50))
    last_name = Field(String(50))

    def __get_name__ (self):
        first_name = self.first_name if self.first_name is not None else ""
        middle_name = self.middle_name if self.middle_name is not None else ""
        last_name = self.last_name if self.last_name is not None else ""
        return " ".join((first_name, middle_name, last_name)).strip()

    def __set_name__ (self,string):
        first_name = ""
        middle_name = ""
        last_name = ""
        split_string = string.split(' ')
        if len(split_string) == 1:
            first_name = string
        elif len(split_string) == 2:
            first_name, last_name = split_string
        elif len(split_string) == 3:
            first_name, middle_name, last_name = split_string
        else: #len(split_string) > 3:
            first_name = split_string[0]
            last_name = split_string[-1]
            middle_name = " ".join(split_string[1:-2])
        self.first_name = first_name
        self.middle_name = middle_name
        self.last_name = last_name

    name = property(__get_name__,__set_name__)

Я бы хотел выполнить запрос следующим образом:

def get_user(user):
    found = None
    if type(user) in [IntType,StringType]:
        if type(user) is StringType:
            where = or_(User.first_name==user,
                        User.middle_name==user,
                        User.last_name==user,
                        User.name==user)
            qry = User.query.filter(where)
        elif type(user) is IntType:
            where = or_(User.id==user,
                        User.employee_id==user)
            qry = User.query.filter(where)
        try:
            found = qry.one()
        except NoResultFound:
            print >> stderr, "Couldn't find '%s'" % user
    elif type(user) == User:
        found=user
    return found

Однако результирующий запрос SQL выглядит примерно так:

SELECT users.first_name AS users_first_name, 
       users.middle_name AS users_middle_name, 
       users.last_name AS users_last_name
FROM users 
WHERE users.first_name = 'Joseph'
   OR users.middle_name = 'M'
   OR users.last_name = 'Schmoe'
   OR false

Обратите внимание на ' false ' вместо поля User.name.

Я получаю эту ошибку:

sqlalchemy.exc.OperationalError: (OperationalError) no such column: false 

Я думаю, что SQL-запрос должен выглядеть следующим образом:

SELECT users.name
FROM users 
WHERE users.name = 'Joseph M Schmoe'

Редактировать: требуемый / второй SQL-запрос был неправильным для того, что я действительно хотел: какой-то пассивный способ создать поле «имя» в базе данных, которое соответствует конкатенации «first_name», «middle_name», «last_name .

Edit2: я верю, что следующее приведёт меня почти туда. Тем не менее, я все еще борюсь с правильным выражением.

Edit3: похоже, это работает для того, что мне нужно сделать. Так что я включаю это как ответ.

Ответы [ 2 ]

2 голосов
/ 14 октября 2011
from sqlalchemy.ext.hybrid import hybrid_property

@hybrid_property
def name (self):
    first_name = self.first_name if self.first_name is not None else ""
    middle_name = self.middle_name if self.middle_name is not None else ""
    last_name = self.last_name if self.last_name is not None else ""
    return " ".join((first_name, middle_name, last_name)).strip()

@name.setter
def name (self,string):
    first_name = ""
    middle_name = ""
    last_name = ""
    split_string = string.split(' ')
    if len(split_string) == 1:
        first_name = string
    elif len(split_string) == 2:
        first_name, last_name = split_string
    elif len(split_string) == 3:
        first_name, middle_name, last_name = split_string
    else: #len(split_string) > 3:
        first_name = split_string[0]
        last_name = split_string[-1]
        middle_name = " ".join(split_string[1:-2])
    self.first_name = first_name
    self.middle_name = middle_name
    self.last_name = last_name

Выражение здесь:

@name.expression
def name (cls):
    f = cls.first_name
    m = cls.middle_name
    l = cls.last_name
    return f+' '+m+' '+l
2 голосов
/ 14 октября 2011

Вы определяете свойство, и оно будет доступно после назначения объекта.

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

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, DateTime, String
from datetime import datetime

Base = declarative_base()
class User(Base):
    first_name = Column('first_name', String)
    middle_name = Column('middle_name', String)
    last_name = Column('last_name', String)

    def __get_name__ (self):
        first_name = self.first_name if self.first_name is not None else ""
        middle_name = self.middle_name if self.middle_name is not None else ""
        last_name = self.last_name if self.last_name is not None else ""
        return " ".join((first_name, middle_name, last_name)).strip()

    def __set_name__ (self,string):
        first_name = ""
        middle_name = ""
        last_name = ""
        split_string = string.split(' ')
        if len(split_string) == 1:
            first_name = string
        elif len(split_string) == 2:
            first_name, last_name = split_string
        elif len(split_string) == 3:
            first_name, middle_name, last_name = split_string
        else: #len(split_string) > 3:
            first_name = split_string[0]
            last_name = split_string[-1]
            middle_name = " ".join(split_string[1:-2])
        self.first_name = first_name
        self.middle_name = middle_name
        self.last_name = last_name

    name = property(__get_name__,__set_name__)

То есть first_name, middle_name and last_name являются атрибутами класса. Когда вы определяете свойство, ему нужен экземпляр этого объекта.

In [13]: User.first_name
Out[13]: Column('first_name', String(), )

In [14]: User.name
Out[14]: <property object at 0x26fef70>

В приведенном выше примере вы можете увидеть разницу. Пока вы не установите это свойство или какие-либо поля этого свойства, оно всегда будет пустым.

Вы должны дать

In [16]: u1 = User()

In [17]: u1.name = "first middle last"

In [18]: u1.name
Out[18]: 'first middle last'

In [19]: u1.first_name
Out[19]: 'first'

После этого вы можете использовать свойство.

Это поможет вам понять вашу проблему. Ваша собственность связана с экземпляром, к которому у вас нет доступа по классу.

...