Skip to content

Commit

Permalink
use simple setup for mysql endpoints (#885)
Browse files Browse the repository at this point in the history
  • Loading branch information
delliott90 committed May 4, 2022
1 parent 021596b commit e59752b
Show file tree
Hide file tree
Showing 5 changed files with 40 additions and 73 deletions.
19 changes: 1 addition & 18 deletions stix_shifter_modules/mysql/entry_point.py
Original file line number Diff line number Diff line change
@@ -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):
Expand All @@ -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')
39 changes: 39 additions & 0 deletions stix_shifter_modules/mysql/stix_transmission/connector.py
Original file line number Diff line number Diff line change
@@ -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
10 changes: 0 additions & 10 deletions stix_shifter_modules/mysql/stix_transmission/delete_connector.py

This file was deleted.

23 changes: 0 additions & 23 deletions stix_shifter_modules/mysql/stix_transmission/ping_connector.py

This file was deleted.

22 changes: 0 additions & 22 deletions stix_shifter_modules/mysql/stix_transmission/results_connector.py

This file was deleted.

0 comments on commit e59752b

Please sign in to comment.