from pymongo import MongoClient
from urllib.parse import quote
class test_mongo:
    def __init__(self, host,port,user,pwd,db):
        self.host = host
        self.port = port
        self.user = user
        self.pwd = pwd
        self.db = db

    def build_conn_mongo(self):
        """
        功能：實現創(chuàng)建客戶端
        :return: client
        """
        escaped_username = quote(self.user, safe='')
        escaped_password = quote(self.pwd, safe='')
        # 創(chuàng)建客戶端對象，連接MongoDB服務器
        client = MongoClient('mongodb://{0}:{1}@{2}:{3}/{4}'.format(escaped_username, escaped_password, self.host, self.port, self.db))
        return client

    def check_mongo(self,query=None):
        """
        功能：查詢信息
        :param query:輸入查詢條件  eg：{"paId": "38ffefbe29ddcf3c9f574aa"}
        :return:返回信息，以及獲取查詢的條數
        """
        client = self.build_conn_mongo()
        db = client.iot_ota
        collection = db.集合名稱
        results = collection.find(query)
        count= collection.count_documents(query)
        client.close()
        return results,count

    def update_mongo_by_query(self,query, set_value):
        """
        功能：按照條件修改一個記錄
        :param query:輸入查詢條件  eg：query1 = {"paId": "38ffefbe291a375c4aa"}
        :param set_value：eg：{"$set":{"description":"yz測試"}}
        :return:返回基本信息
        """
        client = self.build_conn_mongo()
        db = client.iot_ota
        collection = db.集合名稱
        results = collection.update_one(query, set_value)
        client.close()
        return results

    def delete_package_by_query(self,query):
        """
        功能：按照條件刪除一個記錄
        :param query:輸入查詢條件  eg：{"otId":202501221}
        :return:
        """
        client = self.build_conn_mongo()
        db = client.iot_ota
        collection = db.集合名稱
        results = collection.delete_one(query)
        client.close()
        return results