Как обеспечить, чтобы ключи для ввода в Python были строчными? - PullRequest
Новая
       5

Как обеспечить, чтобы ключи для ввода в Python были строчными?

8 голосов
/ 19 ноября 2010

У меня есть диктат, который я хочу преобразовать в JSON с помощью simplejson.

Как я могу убедиться, что все ключи моего dict строчные? 

    {
        "DISTANCE": 17.059918745802999, 
        "name": "Foo Bar", 
        "Restaurant": {
            "name": "Foo Bar", 
            "full_address": {
                "country": "France", 
                "street": "Foo Bar", 
                "zip_code": "68190", 
                "city": "UNGERSHEIM"
            }, 
            "phone": "+33.389624300", 
            "longitude": "7.3064454", 
            "latitude": "47.8769091", 
            "id": "538"
        }, 
        "price": "", 
        "composition": "", 
        "profils": {}, 
        "type_menu": "", 
        "ID": ""
    },

РЕДАКТИРОВАТЬ: Спасибо всем, что взглянули на мой вопрос, мне жаль, что я не объяснил подробно, почему я хотел этого. Это было исправление JSONEmitter из django-piston.

Ответы [ 6 ]

23 голосов
/ 19 ноября 2010 
>>> d = {"your": "DATA", "FROM": "above"}
>>> dict((k.lower(), v) for k, v in d.iteritems())
{'from': 'above', 'your': 'DATA'}
>>> def lower_keys(x):
...   if isinstance(x, list):
...     return [lower_keys(v) for v in x]
...   elif isinstance(x, dict):
...     return dict((k.lower(), lower_keys(v)) for k, v in x.iteritems())
...   else:
...     return x
...
>>> lower_keys({"NESTED": {"ANSWER": 42}})
{'nested': {'answer': 42}}
8 голосов
/ 19 ноября 2010

Вот решение, которое запрещает установку строчной клавиши: 

class LowerCaseDict(dict):
    def __setitem__(self, key, val):
        if not key.islower():
            raise TypeError, "%s key must be lowercase" % key
        dict.__setitem__(self, key, val)

ld = LowerCaseDict()
ld['g']='g'
4 голосов
/ 19 ноября 2010

Поскольку вы не указали четко, что вы хотите сделать:

Преобразовать все ключи в нижний регистр: 

>>> y = dict((k.lower(), v) for k, v in x.iteritems())

Проверьте ключи: 

>>> for k in x.iterkeys():
    if k.islower():
        print k, 'True'
    else:
        print k, 'False'
4 голосов
/ 19 ноября 2010

Вот мое решение: 

def lower_key(in_dict):
    if type(in_dict) is dict:
        out_dict = {}
        for key, item in in_dict.items():
            out_dict[key.lower()] = lower_key(item)
        return out_dict
    elif type(in_dict) is list:
        return [lower_key(obj) for obj in in_dict]
    else:
        return in_dict
2 голосов
/ 11 января 2015

Вот полное решение 

from requests import CaseInsensitiveDict

Или, если вы хотите увидеть код: 

class CaseInsensitiveDict(dict):

    """Basic case insensitive dict with strings only keys."""

    proxy = {}

    def __init__(self, data):
        self.proxy = dict((k.lower(), k) for k in data)
        for k in data:
            self[k] = data[k]

    def __contains__(self, k):
        return k.lower() in self.proxy

    def __delitem__(self, k):
        key = self.proxy[k.lower()]
        super(CaseInsensitiveDict, self).__delitem__(key)
        del self.proxy[k.lower()]

    def __getitem__(self, k):
        key = self.proxy[k.lower()]
        return super(CaseInsensitiveDict, self).__getitem__(key)

    def get(self, k, default=None):
        return self[k] if k in self else default

    def __setitem__(self, k, v):
        super(CaseInsensitiveDict, self).__setitem__(k, v)
        self.proxy[k.lower()] = k
2 голосов
/ 19 ноября 2010

Если вы просто хотите проверить , если они все строчные (ваша формулировка, использующая слово «обеспечить», не ясна, но я подозреваю, что это не то, что вы хотите), вы можете сделать это компактно в одну строку: 

all(k.islower() for k in x.iterkeys())
...