Как работать с несколькими форматами JSON?

Question

Я играю со скриптом Python для извлечения, анализа и форматирования тренда в Twitter JSON. Специфичный для местоположения формат вкладывает тренды в массив:

[
  {
    "created_at": "2010-07-15T22:31:11Z",
    "trends": [
      {
        "name": "trendy",
        "url": "http://search.twitter.com/search?q=trendy",
        "query": "trendy"
      }, ...

В то время как ежедневные и еженедельные форматы json не:

{
  "trends": {
    "2011-01-14 15:20": [
      {
        "name": "#trendy",
        "events": null,
        "promoted_content": null,
        "query": "#trendy"
      },

Я использую этот питон для перечисления тенденций:

class trend:
        #initialize a "trend" object with foo = trend(query,name ...)
        def __init__(self, query, name, promoted_content, events, url):
                self.query = query
                self.name = name
                self.promoted_content = promoted_content
                self.events = events
                self.url = url 

class trending:
        def __init__(self,api_url,title):
                self.api_url = api_url
                self.title = title

        def get_trending(self):
                import simplejson as json
                import urllib2

                trends_all = json.loads(urllib2.urlopen(self.api_url).read())
                return trends_all

        def list_trending(self):
                trends_all = self.get_trending()
                print "%s\n" % self.title

                for x in trends_all[0]['trends']:
                    thistrend = trend(x['query'], x['name'], x['promoted_content'], x['events'], x['url'])
                    print "\t %s (%s) %s" %(thistrend.name, thistrend.url, thistrend.promoted_content)

Это работает для форматов местоположения (первый), но не для ежедневных / еженедельных форматов. Поэтому мне интересно, есть ли разумный способ различить две структуры JSON и реструктурировать их, чтобы я мог работать с обеими.

Answer 1

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

В своем коде вы можете просто заменить:

for x in trends_all[0]['trends']:
    ...

на

if isinstance(trends_all, list):
    trends_all = trends_all[0]
for x in trends_all['trends']:
    ...