1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

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):

        """

        功能：实现创建客户端

        :return: client

        """

        escaped_username = quote(self.user, safe='')

        escaped_password = quote(self.pwd, safe='')

        # 创建客户端对象，连接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