MSSQLAction

Complexity

Total Complexity

6

Size/Duplication

Total Lines	23
Duplicated Lines	0 %

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	3	1
A	_connect_params()	0	13	4
A	connect()	0	3	1

import _mssql

from st2actions.runners.pythonrunner import Action

__all__ = [
    'MSSQLAction'
]


class MSSQLAction(Action):

    def __init__(self, config):
        super(MSSQLAction, self).__init__(config=config)
        self.config = config

    def connect(self, database=None, server=None, user=None, password=None):
        # pylint: disable=no-member
        return _mssql.connect(**self._connect_params(database, server, user, password))

    def _connect_params(self, database=None, server=None, user=None, password=None):
        database = database or self.config.get('default')
        db_config = self.config.get(database, {})
        params = {
            'database': db_config.get('database') or database,
            'server': server or db_config.get('server'),
            'user': user or db_config.get('user'),
            'password': password or db_config.get('password')
        }
        unspecified = [param for param, value in params.iteritems() if value is None]
        if unspecified:
            raise Exception("Must specify or configure in config.yaml: %s" % ', '.join(unspecified))
        return params