Completed
Pull Request — master (#442)
by
unknown
02:14
created

Check.run()   B

Complexity

Conditions 1

Size

Total Lines 28

Duplication

Lines 0
Ratio 0 %
Metric Value
cc 1
dl 0
loc 28
rs 8.8571
1
#!/usr/bin/env python
2
3
# Licensed to the StackStorm, Inc ('StackStorm') under one or more
4
# contributor license agreements.  See the NOTICE file distributed with
5
# this work for additional information regarding copyright ownership.
6
# The ASF licenses this file to You under the Apache License, Version 2.0
7
# (the "License"); you may not use this file except in compliance with
8
# the License.  You may obtain a copy of the License at
9
#
10
#     http://www.apache.org/licenses/LICENSE-2.0
11
#
12
# Unless required by applicable law or agreed to in writing, software
13
# distributed under the License is distributed on an "AS IS" BASIS,
14
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15
# See the License for the specific language governing permissions and
16
# limitations under the License.
17
18
from st2actions.runners.pythonrunner import Action
19
import duo_client
20
21
22
class Check(Action):
23
    def run(self):
24
        """
25
        Ping the Duo Platorm.
26
27
        Returns: An dict with info returned by Duo.
28
29
        Raises:
30
          RuntimeError: On Invalid keys.
31
        """
32
33
        try:
34
            ikey = self.config['auth']['ikey']
35
            skey = self.config['auth']['skey']
36
            host = self.config['auth']['host']
37
        except KeyError:
38
            raise ValueError("Duo config not found in config.")
39
40
        auth = duo_client.Auth(ikey=ikey,
41
                               skey=skey,
42
                               host=host)
43
44
        try:
45
            data = auth.check()
46
        except RuntimeError, e:
47
            print e
48
            raise RuntimeError("Check failed! '%s'" % e)
49
        else:
50
            return data
51