优化数据库类

WorkData/Argus/problem.py

@@ -119,6 +119,27 @@ class DataProblem(object):
             session.close()
             return error
 
+    def delete_like_eventid(self, session, eventid):
+        """
+        :param session: 指针
+        :param eventid:
+        :return:
+        """
+        self.log.debug("查询数据库:")
+        try:
+            self.log.debug("删除相关测试数据")
+            session.query(self.table).filter(self.table.eventid.like(str(eventid/100) + "%")).delete(synchronize_session=False)
+            # session.delete(sql)
+            session.commit()
+            session.close()
+            self.log.debug("删除成功")
+            return True
+        except UtilsDataBase().errors as error:
+            self.log.error("异常:")
+            self.log.error(error)
+            session.close()
+            return error
+
     def insert_one(self, session, eventid=None, source=None, _object=None, objectid=None, clock=None, ns=None,
                    r_clock=None, r_ns=None, hostid=None, name=None, acknowledged=None, acknowledged_time=None,
                    severity=None, status=None,):

WorkData/Zabbix/events.py

@@ -128,4 +128,25 @@ class DataEvents(object):
             self.log.error("异常:")
             self.log.error(error)
             session.close()
-            return error
+            return error
\ No newline at end of file
+
+    def delete_like_eventid(self, session, eventid):
+        """
+        :param session: 指针
+        :param eventid:
+        :return:
+        """
+        self.log.debug("查询数据库:")
+        try:
+            self.log.debug("删除相关测试数据")
+            session.query(self.table).filter(self.table.eventid.like(str(eventid/100) + "%")).delete(synchronize_session=False)
+            # session.delete(sql)
+            session.commit()
+            session.close()
+            self.log.debug("删除成功")
+            return True
+        except UtilsDataBase().errors as error:
+            self.log.error("异常:")
+            self.log.error(error)
+            session.close()
+            return error

WorkData/Zabbix/problem.py

@@ -91,6 +91,27 @@ class DataProblem(object):
             session.close()
             return error
 
+    def delete_like_eventid(self, session, eventid):
+        """
+        :param session: 指针
+        :param eventid:
+        :return:
+        """
+        self.log.debug("查询数据库:")
+        try:
+            self.log.debug("删除相关测试数据")
+            session.query(self.table).filter(self.table.eventid.like(str(eventid/100) + "%")).delete(synchronize_session=False)
+            # session.delete(sql)
+            session.commit()
+            session.close()
+            self.log.debug("删除成功")
+            return True
+        except UtilsDataBase().errors as error:
+            self.log.error("异常:")
+            self.log.error(error)
+            session.close()
+            return error
+
     def insert_one(self, session, eventid=None, source=None, _object=None, objectid=None, clock=None,
                    r_eventid=None, r_clock=None, r_ns=None, correlationid=None, userid=None,
                    acknowledged=None, ns=None, name=None, severity=None):