def __init__(self, base_url=None, auth_url=None, api_url=None, stream_url=None, api_version=None, cacert=None, debug=False, token=None, api_key=None): # Get CLI options. If not given, then try to get it from the environment. self.endpoints = dict() # Populate the endpoints if base_url: self.endpoints['base'] = base_url else: self.endpoints['base'] = os.environ.get('ST2_BASE_URL', DEFAULT_BASE_URL) api_version = api_version or os.environ.get('ST2_API_VERSION', DEFAULT_API_VERSION) if api_url: self.endpoints['api'] = api_url else: self.endpoints['api'] = os.environ.get( 'ST2_API_URL', '%s:%s/%s' % (self.endpoints['base'], DEFAULT_API_PORT, api_version)) if auth_url: self.endpoints['auth'] = auth_url else: self.endpoints['auth'] = os.environ.get( 'ST2_AUTH_URL', '%s:%s' % (self.endpoints['base'], DEFAULT_AUTH_PORT)) if stream_url: self.endpoints['stream'] = auth_url else: self.endpoints['stream'] = os.environ.get( 'ST2_STREAM_URL', '%s:%s/%s' % (self.endpoints['base'], DEFAULT_STREAM_PORT, api_version)) if cacert is not None: self.cacert = cacert else: self.cacert = os.environ.get('ST2_CACERT', None) # Note: boolean is also a valid value for "cacert" is_cacert_string = isinstance(self.cacert, six.string_types) if (self.cacert and is_cacert_string and not os.path.isfile(self.cacert)): raise ValueError('CA cert file "%s" does not exist.' % (self.cacert)) self.debug = debug # Note: This is a nasty hack for now, but we need to get rid of the decrator abuse if token: os.environ['ST2_AUTH_TOKEN'] = token self.token = token if api_key: os.environ['ST2_API_KEY'] = api_key self.api_key = api_key # Instantiate resource managers and assign appropriate API endpoint. self.managers = dict() self.managers['Token'] = ResourceManager(models.Token, self.endpoints['auth'], cacert=self.cacert, debug=self.debug) self.managers['RunnerType'] = ResourceManager(models.RunnerType, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Action'] = ResourceManager(models.Action, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['ActionAlias'] = ActionAliasResourceManager( models.ActionAlias, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['ActionAliasExecution'] = ResourceManager( models.ActionAliasExecution, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['ApiKey'] = ResourceManager(models.ApiKey, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['LiveAction'] = LiveActionResourceManager( models.LiveAction, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Pack'] = PackResourceManager(models.Pack, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Policy'] = ResourceManager(models.Policy, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['PolicyType'] = ResourceManager(models.PolicyType, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Rule'] = ResourceManager(models.Rule, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Sensor'] = ResourceManager(models.Sensor, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['TriggerType'] = ResourceManager(models.TriggerType, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Trigger'] = ResourceManager(models.Trigger, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['TriggerInstance'] = TriggerInstanceResourceManager( models.TriggerInstance, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['KeyValuePair'] = ResourceManager(models.KeyValuePair, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Webhook'] = ResourceManager(models.Webhook, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Trace'] = ResourceManager(models.Trace, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['RuleEnforcement'] = ResourceManager( models.RuleEnforcement, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Stream'] = StreamManager(self.endpoints['stream'], cacert=self.cacert, debug=self.debug)
def __init__(self, base_url=None, auth_url=None, api_url=None, stream_url=None, api_version=None, cacert=None, debug=False, token=None, api_key=None): # Get CLI options. If not given, then try to get it from the environment. self.endpoints = dict() # Populate the endpoints if base_url: self.endpoints['base'] = base_url else: self.endpoints['base'] = os.environ.get('ST2_BASE_URL', DEFAULT_BASE_URL) api_version = api_version or os.environ.get('ST2_API_VERSION', DEFAULT_API_VERSION) self.endpoints['exp'] = '%s:%s/%s' % (self.endpoints['base'], DEFAULT_API_PORT, 'exp') if api_url: self.endpoints['api'] = api_url else: self.endpoints['api'] = os.environ.get( 'ST2_API_URL', '%s:%s/%s' % (self.endpoints['base'], DEFAULT_API_PORT, api_version)) if auth_url: self.endpoints['auth'] = auth_url else: self.endpoints['auth'] = os.environ.get( 'ST2_AUTH_URL', '%s:%s' % (self.endpoints['base'], DEFAULT_AUTH_PORT)) if stream_url: self.endpoints['stream'] = stream_url else: self.endpoints['stream'] = os.environ.get( 'ST2_STREAM_URL', '%s:%s/%s' % ( self.endpoints['base'], DEFAULT_STREAM_PORT, api_version ) ) if cacert is not None: self.cacert = cacert else: self.cacert = os.environ.get('ST2_CACERT', None) # Note: boolean is also a valid value for "cacert" is_cacert_string = isinstance(self.cacert, six.string_types) if (self.cacert and is_cacert_string and not os.path.isfile(self.cacert)): raise ValueError('CA cert file "%s" does not exist.' % (self.cacert)) self.debug = debug # Note: This is a nasty hack for now, but we need to get rid of the decrator abuse if token: os.environ['ST2_AUTH_TOKEN'] = token self.token = token if api_key: os.environ['ST2_API_KEY'] = api_key self.api_key = api_key # Instantiate resource managers and assign appropriate API endpoint. self.managers = dict() self.managers['Token'] = ResourceManager( models.Token, self.endpoints['auth'], cacert=self.cacert, debug=self.debug) self.managers['RunnerType'] = ResourceManager( models.RunnerType, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Action'] = ResourceManager( models.Action, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['ActionAlias'] = ActionAliasResourceManager( models.ActionAlias, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['ActionAliasExecution'] = ActionAliasExecutionManager( models.ActionAliasExecution, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['ApiKey'] = ResourceManager( models.ApiKey, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Config'] = ConfigManager( models.Config, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['ConfigSchema'] = ResourceManager( models.ConfigSchema, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Execution'] = ExecutionResourceManager( models.Execution, self.endpoints['api'], cacert=self.cacert, debug=self.debug) # NOTE: LiveAction has been deprecated in favor of Execution. It will be left here for # backward compatibility reasons until v3.2.0 self.managers['LiveAction'] = self.managers['Execution'] self.managers['Inquiry'] = InquiryResourceManager( models.Inquiry, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Pack'] = PackResourceManager( models.Pack, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Policy'] = ResourceManager( models.Policy, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['PolicyType'] = ResourceManager( models.PolicyType, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Rule'] = ResourceManager( models.Rule, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Sensor'] = ResourceManager( models.Sensor, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['TriggerType'] = ResourceManager( models.TriggerType, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Trigger'] = ResourceManager( models.Trigger, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['TriggerInstance'] = TriggerInstanceResourceManager( models.TriggerInstance, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['KeyValuePair'] = ResourceManager( models.KeyValuePair, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Webhook'] = WebhookManager( models.Webhook, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Timer'] = ResourceManager( models.Timer, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Trace'] = ResourceManager( models.Trace, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['RuleEnforcement'] = ResourceManager( models.RuleEnforcement, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['Stream'] = StreamManager( self.endpoints['stream'], cacert=self.cacert, debug=self.debug) self.managers['Workflow'] = WorkflowManager( self.endpoints['api'], cacert=self.cacert, debug=self.debug) # Service Registry self.managers['ServiceRegistryGroups'] = ServiceRegistryGroupsManager( models.ServiceRegistryGroup, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['ServiceRegistryMembers'] = ServiceRegistryMembersManager( models.ServiceRegistryMember, self.endpoints['api'], cacert=self.cacert, debug=self.debug) # RBAC self.managers['Role'] = ResourceManager( models.Role, self.endpoints['api'], cacert=self.cacert, debug=self.debug) self.managers['UserRoleAssignment'] = ResourceManager( models.UserRoleAssignment, self.endpoints['api'], cacert=self.cacert, debug=self.debug)
def __init__( self, base_url=None, auth_url=None, api_url=None, stream_url=None, api_version=None, cacert=None, debug=False, token=None, api_key=None, basic_auth=None, ): # Get CLI options. If not given, then try to get it from the environment. self.endpoints = dict() # Populate the endpoints if base_url: self.endpoints["base"] = base_url else: self.endpoints["base"] = os.environ.get("ST2_BASE_URL", DEFAULT_BASE_URL) api_version = api_version or os.environ.get("ST2_API_VERSION", DEFAULT_API_VERSION) self.endpoints["exp"] = "%s:%s/%s" % ( self.endpoints["base"], DEFAULT_API_PORT, "exp", ) if api_url: self.endpoints["api"] = api_url else: self.endpoints["api"] = os.environ.get( "ST2_API_URL", "%s:%s/%s" % (self.endpoints["base"], DEFAULT_API_PORT, api_version), ) if auth_url: self.endpoints["auth"] = auth_url else: self.endpoints["auth"] = os.environ.get( "ST2_AUTH_URL", "%s:%s" % (self.endpoints["base"], DEFAULT_AUTH_PORT)) if stream_url: self.endpoints["stream"] = stream_url else: self.endpoints["stream"] = os.environ.get( "ST2_STREAM_URL", "%s:%s/%s" % (self.endpoints["base"], DEFAULT_STREAM_PORT, api_version), ) if cacert is not None: self.cacert = cacert else: self.cacert = os.environ.get("ST2_CACERT", None) # Note: boolean is also a valid value for "cacert" is_cacert_string = isinstance(self.cacert, six.string_types) if self.cacert and is_cacert_string and not os.path.isfile( self.cacert): raise ValueError('CA cert file "%s" does not exist.' % (self.cacert)) self.debug = debug # Note: This is a nasty hack for now, but we need to get rid of the decrator abuse if token: os.environ["ST2_AUTH_TOKEN"] = token self.token = token if api_key: os.environ["ST2_API_KEY"] = api_key self.api_key = api_key if basic_auth: # NOTE: We assume username can't contain colons if len(basic_auth.split(":", 1)) != 2: raise ValueError( "basic_auth config options needs to be in the " "username:password notation") self.basic_auth = tuple(basic_auth.split(":", 1)) else: self.basic_auth = None # Instantiate resource managers and assign appropriate API endpoint. self.managers = dict() self.managers["Token"] = ResourceManager( models.Token, self.endpoints["auth"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["RunnerType"] = ResourceManager( models.RunnerType, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Action"] = ActionResourceManager( models.Action, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["ActionAlias"] = ActionAliasResourceManager( models.ActionAlias, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["ActionAliasExecution"] = ActionAliasExecutionManager( models.ActionAliasExecution, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["ApiKey"] = ResourceManager( models.ApiKey, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Config"] = ConfigManager( models.Config, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["ConfigSchema"] = ResourceManager( models.ConfigSchema, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Execution"] = ExecutionResourceManager( models.Execution, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) # NOTE: LiveAction has been deprecated in favor of Execution. It will be left here for # backward compatibility reasons until v3.2.0 self.managers["LiveAction"] = self.managers["Execution"] self.managers["Inquiry"] = InquiryResourceManager( models.Inquiry, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Pack"] = PackResourceManager( models.Pack, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Policy"] = ResourceManager( models.Policy, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["PolicyType"] = ResourceManager( models.PolicyType, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Rule"] = ResourceManager( models.Rule, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Sensor"] = ResourceManager( models.Sensor, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["TriggerType"] = ResourceManager( models.TriggerType, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Trigger"] = ResourceManager( models.Trigger, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["TriggerInstance"] = TriggerInstanceResourceManager( models.TriggerInstance, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["KeyValuePair"] = ResourceManager( models.KeyValuePair, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Webhook"] = WebhookManager( models.Webhook, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Timer"] = ResourceManager( models.Timer, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Trace"] = ResourceManager( models.Trace, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["RuleEnforcement"] = ResourceManager( models.RuleEnforcement, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Stream"] = StreamManager( self.endpoints["stream"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["Workflow"] = WorkflowManager( self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) # Service Registry self.managers["ServiceRegistryGroups"] = ServiceRegistryGroupsManager( models.ServiceRegistryGroup, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers[ "ServiceRegistryMembers"] = ServiceRegistryMembersManager( models.ServiceRegistryMember, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) # RBAC self.managers["Role"] = ResourceManager( models.Role, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, ) self.managers["UserRoleAssignment"] = ResourceManager( models.UserRoleAssignment, self.endpoints["api"], cacert=self.cacert, debug=self.debug, basic_auth=self.basic_auth, )