在Python中分割字符串并多次出现split关键字

时间:2018-08-30 04:41:15

标签: python mongodb pymongo

因此,我想为MongoDB拆分过滤器和更新参数,以替换Pymongo库中可用的一种方法。

{filter}, {update}

是从文件传递给我的,每行一对。

例如:{"k1":"v1"}, {"k1":"v1", "k2":"v2", "k3":"v3"}

我想做什么?

将它们分割成两个dict变量,

filter = {"k1":"v1"}
update = {"k1":"v1", "k2":"v2", "k3":"v3"}

我尝试了什么?

问题是,我不想更改原始格式,如果我使用“,”拆分它们,那么它可能会突然拆分,我也不能依靠第一次出现的“,”拆分,因为过滤器部件本身可能会拆分有多个“,”

def data_replace_one(host, port, dbname, coll_name, file_path, authdb):

    if LOCALHOST:
        client = pymongo.MongoClient(host, port)       
    else:
        print("Enter credentials:")
        uname = input("Username: ")
        pwd = getpass.getpass()

        client = pymongo.MongoClient(
            host, port, username=uname, password=pwd, authSource=authdb)

    db = client[dbname]
    coll = db[coll_name]

    with open(file_path) as in_file:
        list_dict_queries = [line.strip() for line in in_file]

    list_dict_queries = list(filter(None, list_dict_queries))

    for query in list_dict_queries:
        query_list = query.split("|")
        query_list[0] = query_list[0].strip()
        query_list[1] = query_list[1].strip()
        #print(literal_eval(query_list[0]), literal_eval(query_list[1]))
        coll.replace_one(literal_eval(
            query_list[0]), literal_eval(query_list[1]))

2 个答案:

答案 0 :(得分:0)

我认为,最简单的方法是在每行周围添加一些方括号,然后将其解释为JSON-前提是您的输入格式必须保证与JSON兼容。

类似的东西:

import json

with open(file_path) as in_file:
    list_dict_queries = [('[' + line.strip() + ']') for line in in_file]

query_list = [json.loads(n) for n in list_dict_queries]

答案 1 :(得分:0)

如果您在其他任何地方都没有花括号/花括号,则可以使用以下内容。

>>> filter, update = re.findall('{.*?}', '{"k1":"v1"}, {"k1":"v1", "k2":"v2", "k3":"v3"}')

>>> filter
'{"k1":"v1"}'

>>> update
'{"k1":"v1", "k2":"v2", "k3":"v3"}'