uptime/mongo_agent.py at master · hpandeycodeit/uptime · GitHub

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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
from pymongo import MongoClient
import bson


class MongoAgent:
    db = None

    def __init__(self, connection_string, db_name):
        """
        This class provides a MongoDB agent used to query data files where user data is stored
        :param connection_string: Connection details of the MongoDB database
        :param db_name: Database name
        """
        client = MongoClient(connection_string)
        self.db = client[db_name]

    def get_dashboard_data(self, email):
        """
        Retrieves the dashboard data from the dashboard collection in the database
        :param email: user email whose data will be returned
        :return:
        """
        collection = self.db["dashboard"]
        return collection.find_one({"email": email})

    def get_metrics_data(self, email):
        """
        Retrieves the metrics data from the metrics collection in the database
        :param email: user email whose data will be returned
        :return:
        """
        collection = self.db["metrics"]
        return collection.find_one({"email": email})

    def get_historical_data(self, email):
        """
        Retrieves the metrics data from the historical collection in the database
        :param email: user email whose data will be returned
        :return:
        """

        collection = self.db["historical"]
        data = collection.find_one({"email": email})
        return data

    def add_form_data(self, product_category, short_heading, web_url, contact_person, email, long_description):
        """
        Add the weblink data from the form entered by the user to the database
        """
        collection = self.db["weblinks"]
        collection.insert_one({
            "product_category": product_category,
            "short_heading": short_heading,
            "web_url": web_url,
            "contact_person": contact_person,
            "email": email,
            "long_description": long_description
        })
        return True

    def update_form_data(self, id, product_category, short_heading, web_url, contact_person, email, long_description):
        """
        Update the weblink data from the form entered by the user to the database
        """
        collection = self.db["weblinks"]
        collection.update_one({
            "_id": bson.ObjectId(id)}, {
            '$set': {
                "product_category": product_category,
                "short_heading": short_heading,
                "web_url": web_url,
                "contact_person": contact_person,
                "email": email,
                "long_description": long_description
            }
        }, upsert=False)
        return True

    def get_weblink_data(self, component):
        """
        Retrieves the weblink data from the database.
        :param component: pivotal product
        :return:
        """

        links = []
        collection = self.db["weblinks"]
        data = collection.find({"product_category": component})
        for link in data:
            link['_id'] = str(link['_id'])
            links.append(link)
        return links

    def get_kbanalytics_data(self):
        """
        Retrieves the data for kb analytics
        :param component: analytics component
        :return:
        """

        collection = self.db["kbanalytics"]
        data = collection.find_one({}, {'_id': False})
        return data

    def get_kb_data(self):
        """
        Retrieves the data for kb analytics
        :param component: analytics component
        :return:
        """

        collection = self.db["kbdata"]
        data = collection.find_one({}, {'_id': False})
        return data

    def get_mykb_data(self, email):
        """
        Retrieves the metrics or all my articles
        :param email: user email whose data will be returned
        :return:
        """

        collection = self.db["mykb"]
        data = collection.find_one({"email": email})
        return data

    def get_trends_data(self):
        """
        Retrieves the data for trends
        :param component: analytics component
        :return:
        """

        collection = self.db["trends"]
        data = collection.find_one({}, {'_id': False})
        return data


    def update_mykb_data(self, id):
        """
        Update the myKb with KB Completed as TRUE
        """
        collection = self.db["mykb_completed"]
        collection.insert_one({
            "ticket_id": id,
            "kb_completed": "TRUE",
        })
        return True