Python字典等效pymongo声明

Question

我有一个Python字典这样

{'OR': [{'AND': [{'column': 'XXX', 'operator': '=', 'value': u'M'}, {'column': 'XXX', 'operator': '=', 'value': 'N'}]}, {'column': 'YYY', 'operator': '>=', 'value': '3.0'}]} 

现在，我想把它转换成类似

{'$or': [{'$and': [{'XXX': 'M'}, {'YYY': 'N'}]}, {'YYY': {u'$gte': 3.0}}]} 

这显然是相当于pymongo声明，我相信。

到目前为止，我写的代码是这样的：

FILTMAP = {'>=': '$gte', '<=': '$lte', '>': '$gt', '<': '$lt', "!=":"$ne"} 
CONJUNCTION_MAP = {"AND":"$and", "OR":"$or"} 

def gen_mongo_filters_json(filter, supplied_key="") 
    first_key = filters.keys()[0] 
    if first_key == 'OR' or first_key == 'AND': 

     if supplied_key == "":  

      return_dict[CONJUNCTION_MAP[first_key]] = [] 
     else:    
      temp_dict[CONJUNCTION_MAP[first_key]] = [] 
      #return_dict[supplied_key] = temp_dict 
     for i in range (len(filters[first_key])): 
      if supplied_key == "":    
       return_dict[CONJUNCTION_MAP[first_key]].append(gen_mongo_filters_json(filters[first_key][i], first_key)) 
      else: 
       temp_dict[CONJUNCTION_MAP[first_key]].append(gen_mongo_filters_json(filters[first_key][i], first_key)) 
       return_dict[CONJUNCTION_MAP[supplied_key]].append(temp_dict) 
    else: 
     operator = filters['operator'] 
     if operator == "=": 
      ret_dict = {filters['column']:filters['value'] 
      return ret_dict 
     else: 
      operator = FILTMAP[operator] 
      ret_dict = {filters['column']:{operator:filters['value']}} 
      return ret_dict 
    return return_dict 

它产生的输出是：

{u'$or': [{u'$and': [{u'Engine': u'MSN'}, {u'Engine': u'Google'}]}, {u'$and': [{u'Engine': u'MSN'}, {u'Engine': u'Google'}]}, {...}, {u'Imps': {u'$gte': 3.0}}]} 

这是附近的解决方案，但没有一个确切。它适用于像字典

{'AND': [{'column': 'XXX', 'operator': '=', 'value': 'M'}, {'column': 'XXX', 'operator': '=', 'value': 'N'}]} 

OR 

{'column': 'YYY', 'operator': '>', 'value': '1000'} 

你能指点我一个方向吗？

（这个想法是创建一个通用的。所以，我想产生等同于任何有效的Python解释成pymongo声明。最小的是最后一个）

Answer 1

你的示例代码不运行，但因为一个字典

{'OR': [{'AND': [{'column': 'XXX', 'operator': '=', 'value': 'M'}, {'column': 'YYY', 'operator': '=', 'value': 'N'}]}, {'column': 'YYY', 'operator': '>=', 'value': '3.0'}]} 

应convertet到

{'$or': [{'$and': [{'XXX': 'M'}, {'YYY': 'N'}]}, {'YYY': {'$gte': 3.0}}]} 

使用这样的事情：

FILTMAP = {'>=': '$gte', '<=': '$lte', '>': '$gt', '<': '$lt', "!=":"$ne"} 
CONJUNCTION_MAP = {"AND":"$and", "OR":"$or"} 

def convert_column(dic): 
    if not dic['operator'] in FILTMAP: 
     return {dic['column']: dic['value']} 
    else: 
     value = float(dic['value']) if dic['operator'] == "!=" else dic['value'] 
     return {dic['column']: {FILTMAP[dic['operator']]: value}} 

def convert(dic): 
    for k,v in dic.items(): 
     if isinstance(v, list): 
      if k in CONJUNCTION_MAP: 
       k = CONJUNCTION_MAP[k] 
      return {k: [convert(i) for i in v]} 
     else: 
      return convert_column(dic) 

我不知道将'3.0'转换为3.0是否重要。该行

value = float(dic['value']) if dic['operator'] == "!=" else dic['value'] 

是相当hackish，你想用一些适当的逻辑来代替它来处理这些情况。

Answer 2

感谢您的回答。我也想出了一些东西。这里是完整的代码

FILTMAP = {'>=': '$gte', '<=': '$lte', '>': '$gt', '<': '$lt', "!=":"$ne"} 
CONJUNCTION_MAP = {"AND":"$and", "OR":"$or"} 


def gen_mongo_filters_json(filters, supplied_key=""): 
    return_dict = {} 
    temp_dict = {} 
    first_key = filters.keys()[0] 
    if first_key == 'OR' or first_key == 'AND': 

     if supplied_key == "": 

      return_dict[CONJUNCTION_MAP[first_key]] = [] 
     else:    
      temp_dict[CONJUNCTION_MAP[first_key]] = [] 
      #return_dict[supplied_key] = temp_dict""" 
     for i in range (len(filters[first_key])): 
      if supplied_key == "": 
       return_dict[CONJUNCTION_MAP[first_key]].append(gen_mongo_filters_json(filters[first_key][i], first_key)) 
      else: 
       for i in range (len(filters[first_key])): 
        temp_dict[CONJUNCTION_MAP[first_key]].append(gen_mongo_filters_json(filters[first_key][i], first_key)) 
       return temp_dict 
    else: 
     operator = filters['operator'] 
     if operator == "=": 
      ret_dict = {filters['column']:filters['value']} 
      return ret_dict 
     else: 
      operator = FILTMAP[operator] 
      ret_dict = {filters['column']:{operator:filters['value']}} 
      return ret_dict 
    return return_dict 


if __name__ == "__main__": 

    print gen_mongo_filters_json({'OR': [{'AND': [{'column': 'XXX', 'operator': '=', 'value': 'M'}, {'column': 'XXX', 'operator': '=', 'value': 'N'}]}, {'AND': [{'column': 'PPP', 'operator': '=', 'value': 'R'}, {'column': 'DDD', 'operator': '=', 'value': 'T'}]}]}) 
enter code here 

请让我知道您的评论

我尝试了

{'column': 'YYY', 'operator': '>', 'value': '1000'} 

{'AND': [{'column': 'XXX', 'operator': '=', 'value': 'M'}, {'column': 'XXX', 'operator': '=', 'value': 'N'}]} 

{'OR': [{'AND': [{'column': 'XXX', 'operator': '=', 'value': u'M'}, {'column': 'XXX', 'operator': '=', 'value': 'N'}]}, {'column': 'YYY', 'operator': '>=', 'value': '3.0'}]} 

{'OR': [{'AND': [{'column': 'XXX', 'operator': '=', 'value': u'M'}, {'column': 'XXX', 'operator': '=', 'value': 'N'}]}, {'AND': [{'column': 'PPP', 'operator': '=', 'value': u'R'}, {'column': 'DDD', 'operator': '=', 'value': 'T'}]}]} 

，这里是输出

{u'YYY': {u'$gt': u'1000'}} 

{u'$and': [{u'XXX': u'M'}, {u'XXX': u'N'}]} 

{u'$or': [{u'$and': [{u'XXX': u'M'}, {u'XXX': u'N'}]}, {u'YYY': {u'$gte': u'3.0'}}]} 

{u'$or': [{u'$and': [{u'XXX': u'M'}, {u'XXX': u'N'}]}, {u'$and': [{u'PPP': u'R'}, {u'DDD': u'T'}]}]} 

感谢您的回答。它也很好。我现在会尝试。

还有一件事，我实际上省略了示例代码中的“价值”因素。我很感谢你指出这一点。我已经有了一种方法来实现这一点。不管怎么说，还是要谢谢你。