Merge branch 'develop' into reaqta_connector

stix_shifter_modules/mysql/entry_point.py

@@ -1,9 +1,4 @@
 from stix_shifter_utils.utils.base_entry_point import BaseEntryPoint
-from stix_shifter_utils.modules.base.stix_transmission.base_sync_connector import BaseSyncConnector
-from .stix_transmission.ping_connector import PingConnector
-from .stix_transmission.delete_connector import DeleteConnector
-from .stix_transmission.results_connector import ResultsConnector
-from .stix_transmission.api_client import APIClient
 
 
 class EntryPoint(BaseEntryPoint):
@@ -13,18 +8,6 @@ def __init__(self, connection={}, configuration={}, options={}):
         self.set_async(False)
 
         if connection:
-            api_client = APIClient(connection, configuration)
-            base_sync_connector = BaseSyncConnector()
-            ping_connector = PingConnector(api_client)
-            query_connector = base_sync_connector
-            status_connector = base_sync_connector
-            results_connector = ResultsConnector(api_client)
-            delete_connector = DeleteConnector(api_client)
+            self.setup_transmission_basic(connection, configuration)
 
-            self.set_results_connector(results_connector)
-            self.set_status_connector(status_connector)
-            self.set_delete_connector(delete_connector)
-            self.set_query_connector(query_connector)
-            self.set_ping_connector(ping_connector)
-
         self.setup_translation_simple(dialect_default='default')

stix_shifter_modules/mysql/stix_transmission/connector.py

@@ -0,0 +1,39 @@
+import datetime
+import json
+from stix_shifter_utils.modules.base.stix_transmission.base_sync_connector import BaseSyncConnector
+from .api_client import APIClient
+from stix_shifter_utils.utils.error_response import ErrorResponder
+from stix_shifter_utils.utils import logger
+
+
+class Connector(BaseSyncConnector):
+
+    def __init__(self, connection, configuration):
+        self.api_client = APIClient(connection, configuration)
+        self.logger = logger.set_logger(__name__)
+        self.connector = __name__.split('.')[1]
+
+    def ping_connection(self):
+        response = self.api_client.ping_data_source()
+        response_code = response.get('code')
+        response_txt = response.get('message')
+        return_obj = dict()
+        return_obj['success'] = False
+
+        if len(response) > 0 and response_code == 200:
+            return_obj['success'] = True
+        else:
+            ErrorResponder.fill_error(return_obj, response, ['message'], error=response_txt, connector=self.connector)
+        return return_obj
+
+    def create_results_connection(self, query, offset, length):
+        return_obj = dict()
+        response = self.api_client.run_search(query, start=offset, rows=length)
+        response_code = response.get('code')
+        response_txt = response.get('message')
+        if response_code == 200:
+            return_obj['success'] = True
+            return_obj['data'] = response.get('result')
+        else:
+            ErrorResponder.fill_error(return_obj, response, ['message'], error=response_txt, connector=self.connector)
+        return return_obj