import pymongo

client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))


def update_user_info_by_cid(cid, data):
    """
    更新用户信息
    Parameters:
        cid: 企业ID
        data: 更新数据
    Returns:
        info: 信息
    """
    collection = client['用户']['用户信息']
    res = collection.update_one({"企业ID": cid}, {"$set": data})
    info = res.raw_result['updatedExisting']
    return info


def find_user_info_by_cid(cid):
    """
    更新用户信息
    Parameters:
        cid: 企业ID
    Returns:
        info: 查询结果
    """
    collection = client['用户']['用户信息']
    data = collection.find({"企业ID": cid}, {'_id': False})
    record = list(data)
    return record