Completed
Pull Request — master (#590)
by Anthony
03:18
created

CiscoSparkAction.connect()   A

Complexity

Conditions 1

Size

Total Lines 3

Duplication

Lines 0
Ratio 0 %

Importance

Changes 2
Bugs 0 Features 1
Metric Value
cc 1
c 2
b 0
f 1
dl 0
loc 3
rs 10
1
try:
2
    from ciscosparkapi import CiscoSparkAPI, DEFAULT_BASE_URL
3
except ImportError:
4
    message = ('Missing "ciscosparkapi", please install it using pip:\n'
5
               'pip install ciscosparkapi')
6
    raise ImportError(message)
7
8
from .ciscospark_parsers import CiscoSparkResultSets
9
from st2actions.runners.pythonrunner import Action
10
11
__all__ = [
12
    'CiscoSparkAction',
13
]
14
15
16
class CiscoSparkAction(Action):
17
    def __init__(self, config):
18
        super(CiscoSparkAction, self).__init__(config)
19
        self._access_token = self.config['access_token']
20
        self._base_url = self.config.get('base_url', DEFAULT_BASE_URL)
21
        self.connect()
22
        self._parser = CiscoSparkResultSets()
23
24
    def connect(self):
25
        self.connection = CiscoSparkAPI(access_token=self._access_token,
26
                                        base_url=self._base_url)
27
28
    def _parse_result(self, result):
29
        return self._parser.formatter(result)
30