adding env vars

librato · Aug 25, 2017 · b728eb2 · b728eb2
1 parent 3f3521a
commit b728eb2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/librato/__init__.py b/librato/__init__.py
@@ -26,6 +26,7 @@
 import platform
 import time
 import logging
+import os
 from six.moves import http_client
 from six.moves import map
 from six import string_types
@@ -560,11 +561,15 @@ def set_timeout(self, timeout):
         self.timeout = timeout
 
 
-def connect(username, api_key, hostname=HOSTNAME, base_path=BASE_PATH, sanitizer=sanitize_no_op,
+def connect(username=None, api_key=None, hostname=HOSTNAME, base_path=BASE_PATH, sanitizer=sanitize_no_op,
             protocol="https", tags={}):
     """
     Connect to Librato Metrics
     """
+
+    username = username if username else os.getenv('LIBRATO_USER', '')
+    api_key = api_key if api_key else os.getenv('LIBRATO_TOKEN', '')
+
     return LibratoConnection(username, api_key, hostname, base_path, sanitizer=sanitizer, protocol=protocol, tags=tags)