Блог

В последнее время это возникало несколько раз, и я хотел бы разобраться с этим лучше, чем раньше: у меня есть ряд атрибутов, на которые я делаю перекрестные ссылки между объектом и словарем. Если значение между ними различается, я хочу присвоить object.attribute значению dictionary[‘атрибут’]. Я также хочу отслеживать, что меняется.

Теперь моя первая мысль — просто использовать оператор if else для каждого атрибута, но после написания нескольких из них становится очевидно, что я снова и снова переписываю один и тот же код. Для этого должен быть ПРОСТОЙ способ, где я указываю только те части, которые меняются каждый раз, а затем перебираю все атрибуты.

В производственном коде есть 15 различных атрибутов, но в моем примере ниже для простоты будет использоваться только 2. У меня есть некоторая идея о том, как сделать это умным способом, но я пропускаю последний шаг фактической установки object.attribute, равного значению dictionary[‘attribute’].

 # Simulated data setup - not under my control IRL
class someClass:
    def __init__(self, name, version):
        self.name = name
        self.version = version

objA = someClass('Test1','1.1')        
dictA = {'name':'Test1','revision':'1.2'}

# My code below        

# option 1 - a series of for loops
def updateAttributesSimple(obj, adict, msg):
    if obj.name == adict['name']:
        msg.append('Name is the same')
    else:
        msg.append('Name was updated from %s to %s' % (obj.name, adict['name']))
        obj.name = adict['name']

    if obj.version == adict['revision']:
        msg.append('Version is the same')
    else:
        msg.append('Version was updated from %s to %s' % (obj.version, adict['revision']))
        obj.version = adict['revision']        

# option 2 - trying to be clever about this
def updateAttributesClever(obj, adict, msg):
    attributeList = (('Name', obj.name, adict['name']),
                     ('Version', obj.version, adict['revision']))

    for valTuple in attributeList:
        if valTuple[1] == valTuple[2]:
            msg.append('%s is the same' % (valTuple[0]))
        else:
            msg.append('%s was updated from %s to %s' % (valTuple[0], valTuple[1], valTuple[2]))
            # code to set valTuple[1] = valTuple[2] goes here, but what is it?
            # valTuple[1] = valTuple[2] attempts to set the desired value to a string, rather than the attribute of obj itself            


msg = ['Updating Attributes simple way:']
updateAttributesSimple(objA, dictA, msg)
print 'nt'.join(msg)

#reset data
objA = someClass('Test1','1.1')        
dictA = {'name':'Test1','revision':'1.2'}

msg = ['Updating Attributes clever way:']
updateAttributesClever(objB, dictB, msg)
print 'nt'.join(msg)
  

Идея заключается в том, что таким образом, всякий раз, когда мне нужно добавить другой атрибут, я могу просто обновить список проверяемых атрибутов, а остальной код уже написан. Каков питонический способ добиться этого?

setattr() это то, что вы ищете:

 attributeList = (('Name',    'name',    'name'),
                 ('Version', 'version', 'revision'))

for title, obj_attribute, dict_key in attributeList:
    obj_value = getattr(obj, obj_attribute)
    adict_value = adict[dict_key]

    if obj_value == adict_value:
        msg.append('%s is the same' % (obj_value))
    else:
        msg.append('%s was updated from %s to %s' % (title, obj_value, adict_value))

        setattr(obj, obj_attribute, adict_value)
  

Это должно сработать для вашего:

 class X(object):
    def __init__(self):
        self.a = 1
        self.b = 2

x = X()

d = dict()
d['a'] = 1
d['b'] = 3

def updateAttributes(obj,dic):
    def update(name):
        val = dic[name]
        if getattr(obj,name)==val:
            print name,"was equal"
        else:
            print "setting %s to %s" % (name,val)
            setattr(obj,name,val)

    for name in ['a','b']:
        update(name)

updateAttributes(x,d)
print x.a
print x.b
  

Возможно, вам захочется подумать о создании функции, которая может принимать произвольный объект и преобразовывать словарь пар имя / значение во что-то более значимое. Это не совсем «питоновская» стратегия, но то, что довольно легко сделать на Python из-за его поддержки замыканий и того, как он обрабатывает объекты под капотом:

 def checkUpdates( obj ):
    def updated( dictionaryPrevious, msg ):
        for name, value in dictionaryPrevious.items():
            if( obj.__dict__[name] == value ):
                msg.append('Name is the same')
            else:
                msg.append(name   'has been changed!')
                obj.__dict__[name] = value
    return updated
  

Я делаю одно предположение, имена в словаре всегда соответствуют объектным переменным. Если они не совпадают, вам нужно будет выполнить сопоставление.

Редактировать:

() => [] и object => obj . спасибо, ребята. Иногда вы переходите с одного языка на несколько других, и все это запутывается.

Пара ответов близки, но чтобы справиться с тем фактом, что имя ключа в dict не совпадает с именем атрибута соответствующего объекта, вам понадобится какой-то способ справиться с этим. Это можно легко сделать, добавив еще один словарь, сопоставляющий имена ключей в dict с именами атрибутов объекта.

 class someClass:
    def __init__(self, name, version):
        self.name = name
        self.version = version

objA = someClass('Test1','1.1')
dictA = {'name':'Test1','revision':'1.2'}
keymap = {'name':'name', 'revision':'version'}

def updateAttributesGeneric(obj, adict, key2attr, msg):
    for key, value in adict.iteritems():
        attrname = key2attr[key]
        if getattr(obj, attrname) == value:
            msg.append('%s is the same' % attrname)
        else:
            msg.append('%s has been changed' % attrname)
            setattr(obj, attrname, adict[key])

msg = ['Updating Attributes:']
updateAttributesGeneric(objA, dictA, keymap, msg)
print 'nt'.join(msg)

# Updating Attributes:
#   name is the same
#   version has been changed