简单正则表达遍历Mongo

有心人可以把它封装成函数哦

import pymongo,re
client = pymongo.MongoClient('127.0.0.1', 27017)  # 缺少一步骤进行属性的清洗操作，确定是否有这个值
dbname='WHU教务系统'
db = client.course
documents = db[dbname].find()
for document in documents:
    mongoid=document['_id']
    myquery = {"_id": mongoid}
    cour_area=document['上课时间地点']

    pattern = re.compile(r'[1-3]区')  # 查找数字
    school=None
    if cour_area!=None:
        result=pattern.findall(cour_area)
        if len(result)==1:
            print(result[0])
            school=result[0]

    listdata={
        '校区':school
    }
    newvalues = {"$set": listdata}
    db[dbname].update_one(myquery, newvalues)
    print('插入一个')