DjangoBB LoFi version

Полная версия: Динамическое создание свойств у объекта

Начало » Python для новичков » Динамическое создание свойств у объекта

Дек. 27, 2014 00:01:28

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

 books = {}
for item in items:
            itemData = {}
            itemData['id'] = item.id
            itemData['title'] = item.card.title
            itemData['description'] = item.card.description
            book_id = str(item.card.book.id)
            if books[book_id]  == None:
                books[book_id] = {}
                books[book_id]['title'] = item.card.book.title
                books[book_id]['description'] = item.card.book.description
                books[book_id]['cards'] = []
            books[book_id]['cards'].append (itemData)

На строчке

 if books[book_id]  == None:

он выкидывает KeyError. в двух словах мне требуется убедится существует ли уже отдельный массив который доступен как свойство объекта books и если существует то просто запихнуть в него карту если нет, то создать этот массив.

Дек. 27, 2014 01:03:54

Напишу решение, мне пришлось написать отдельный метод. Если есть что-то элегантнее отпишитесь ниже:

  if hasAttr(books, book_id) == False:
               #someCode

def hasAttr (obj, attr):
    try:
        obj[attr]
        return True
    except AttributeError:
        return False
    except KeyError:
        return False

py.user.next

Дек. 27, 2014 03:08:51

in
выкидывает KeyError

if book_id not in books:

GreyZmeem

Дек. 27, 2014 03:28:13

Чтобы проверить присутствует ли ключ в словаре:

>>> my_dict = {'a': 1, 'b': 2}
>>> 'a' in my_dict
True
>>> 'c' in my_dict
False

Чтобы получить какое-то значения из словаря по ключу или None/дефолтное значение, есть метод get:

>>> print(my_dict.get('a'))
1
>>> print(my_dict.get('c'))
None
>>> print(my_dict.get('c', 'default value'))
default value
>>>

Также правила хорошего тона - делать проверку на None при помощи оператора is, а не равенством.

Потому вашу проверку:

if books[book_id]  == None

Можно заменить на:

if book_id not in books

Или, если ключ может присутствовать в словаре и содержать None:

if books.get(book_id) is None: