Change environ to getenv

Author: rw-access

detection_rules/eswrap.py

@@ -12,7 +12,7 @@
 from kibana import Kibana, RuleResource
 
 from .main import root
-from .misc import environ
+from .misc import getenv
 from .utils import normalize_timing_and_sort, unix_time_to_formatted, get_path
 from .rule_loader import get_rule, rta_mappings, load_rule_files, load_rules
 
@@ -179,10 +179,10 @@ def run(self, agent_hostname, indexes, verbose=True, **match):
 
 @es_group.command('collect-events')
 @click.argument('agent-hostname')
-@click.option('--elasticsearch-url', '-u', default=environ("DR_ELASTICSEARCH_URL"))
-@click.option('--cloud-id', default=environ("DR_ELASTICSEARCH_URL"))
-@click.option('--user', '-u', default=environ("DR_USER"))
-@click.option('--password', '-p', default=environ("DR_PASSWORD"), hide_input=True)
+@click.option('--elasticsearch-url', '-u', default=getenv("DR_ELASTICSEARCH_URL"))
+@click.option('--cloud-id', default=getenv("DR_ELASTICSEARCH_URL"))
+@click.option('--user', '-u', default=getenv("DR_USER"))
+@click.option('--password', '-p', default=getenv("DR_PASSWORD"), hide_input=True)
 @click.option('--index', '-i', multiple=True, help='Index(es) to search against (default: all indexes)')
 @click.option('--agent-type', '-a', help='Restrict results to a source type (agent.type) ex: auditbeat')
 @click.option('--rta-name', '-r', help='Name of RTA in order to save events directly to unit tests data directory')
@@ -229,10 +229,10 @@ def normalize_file(events_file):
 
 @root.command("kibana-upload")
 @click.argument("toml-files", nargs=-1, required=True)
-@click.option('--kibana-url', '-u', default=environ("DR_KIBANA_URL"))
-@click.option('--cloud-id', default=environ("DR_CLOUD_ID"))
-@click.option('--user', '-u', default=environ("DR_USER"))
-@click.option('--password', '-p', default=environ("DR_PASSWORD"))
+@click.option('--kibana-url', '-u', default=getenv("DR_KIBANA_URL"))
+@click.option('--cloud-id', default=getenv("DR_CLOUD_ID"))
+@click.option('--user', '-u', default=getenv("DR_USER"))
+@click.option('--password', '-p', default=getenv("DR_PASSWORD"))
 def kibana_upload(toml_files, kibana_url, cloud_id, user, password):
     """Upload a list of rule .toml files to Kibana."""
     from uuid import uuid4

detection_rules/misc.py

@@ -213,6 +213,6 @@ def parse_config():
     return _CONFIG
 
 
-def environ(name):
+def getenv(name):
     """Callback function for `default` to get an environment variable."""
     return lambda: os.environ.get(name)