SaltLocal

Complexity

Total Complexity

2

Size/Duplication

Total Lines	49
Duplicated Lines	0 %

Importance

Changes	1
Bugs	1	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
B	run()	0	34	2

import json
from requests import Session

from lib.base import SaltAction


class SaltLocal(SaltAction):
    __explicit__ = [
        'cmdmod',
        'event',
        'file',
        'grains',
        'pillar',
        'pkg',
        'saltcloudmod',
        'schedule',
        'service',
        'state',
        'status'
    ]

    def run(self, module, target, expr_form, args, **kwargs):
        self.verify_ssl = self.config.get('verify_ssl', True)
        '''
        CLI Examples:

            st2 run salt.local module=test.ping matches='web*'
            st2 run salt.local module=test.ping expr_form=grain target='os:Ubuntu'
        '''
        if module not in ['test.ping','test.version']:
            self.generate_package('local',
                                  cmd=module,
                                  target=target,
                                  expr_form=expr_form,
                                  args=args,
                                  data=kwargs)
        else:
            self.generate_package('local',
                                  cmd=module,
                                  target=target,
                                  expr_form=expr_form,
                                  args=None,
                                  data=kwargs)
        self.generate_package('local',
                              cmd=module,
                              target=target,
                              expr_form=expr_form,
                              args=args,
                              data=kwargs)
        request = self.generate_request()
        self.logger.info('[salt] Request generated')
        request.prepare_body(json.dumps(self.data), None)
        self.logger.info('[salt] Preparing to send')
        resp = Session().send(request, verify=self.verify_ssl)
        return resp.json()