將嵌套的OrderedDict嵌套到不同級別的嵌套列表

Question

我使用simple_salesforce來查詢CRM關係數據庫。使用SOQL查詢數據，這是SQL專門針對Salesforce的變體。

它返回嵌套的有序字典。在「加入」中，如果沒有記錄，則返回None。如果有記錄，則返回OrderedDict。我的循環似乎並不是最有效的，因爲它重複了幾行。我正在尋找更簡潔的方式來編寫循環。

查詢使用：

select Id, Subject, Account.Name, Opportunity.Name From Case where ID = '500xxxxxxxxxxxxx' 

代碼：

from collections import OrderedDict 

od = OrderedDict([ 
    (u'totalSize', 1), 
    (u'done', True), 
    (u'records', [OrderedDict([ 
     (u'attributes', OrderedDict([ 
      (u'type', u'Case'), 
      (u'url', u'/services/data/v29.0/sobjects/Case/500xxxxxxxxxxxxxxxx') 
     ]) 
     ), 
     (u'Id', u'500xxxxxxxxxxxxxxxx'), 
     (u'Subject', u"Sample Case 1"), 
     (u'Account', OrderedDict([ 
      (u'attributes', OrderedDict([ 
       (u'type', u'Account'), 
       (u'url', u'/services/data/v29.0/sobjects/Account/001xxxxxxxxxxxxxxxx') 
       ]) 
      ), 
      (u'Name', u'Sample Account 1') 
     ])), 
     (u'Opportunity', None)] 
    )])]) 


output = [] 

def get_items(this_item): 
    for key, value in this_item.iteritems(): 
     if key <> 'attributes': 
      if type(value) is OrderedDict: 
       for key, value in value.iteritems(): 
        if key <> 'attributes': 
         record.append(str(value)) 
      else: 
       record.append(str(value)) 


for item in od['records']: 
    record = [] 
    record.append(get_items(item)) 


    output.append(record) 
print output 

輸出：

[['500xxxxxxxxxxxxxxxx', 'Sample Case 1', 'Sample Account 1', 'None']] 

Answer 1

試試這個：

def get_rec_vals(rec): 
    ret_list = [] 
    for key,val in rec.items(): 
     if key == 'attributes': continue 
     if isinstance(val, OrderedDict): 
      # recursive call 
      ret_list.extend(get_rec_vals(val)) 
     else: 
      ret_list.append(val) 
    return ret_list 


output = [get_rec_vals(rec) for rec in od['records']] 
print(output) 

輸出：

[['500xxxxxxxxxxxxxxxx', 'Sample Case 1', 'Sample Account 1', None]] 

PS我不知道它是否會正常工作了多條記錄，所以你需要提供至少兩個記錄...