Support user_agent http header

CrowdStrike · Nov 16, 2022 · 147cd82 · 147cd82
1 parent 08aea27
commit 147cd82
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/ccib/__init__.py b/ccib/__init__.py
@@ -1 +1,3 @@
-__version__ = '0.0.1'
+from .version import __version__
+
+__all__ = ['__version__']
diff --git a/ccib/falcon.py b/ccib/falcon.py
@@ -4,6 +4,7 @@
 from .config import config
 from .log import log
 from .helper import thousands
+from .version import __version__
 
 
 class ApiError(Exception):
@@ -19,10 +20,11 @@ class FalconAPI():
     }
 
     def __init__(self):
-        client_id = config.get('falcon', 'client_id')
-        client_secret = config.get('falcon', 'client_secret')
-        crowdstrike_url = self.__class__.base_url()
-        self.intel = Intel(client_id=client_id, client_secret=client_secret, base_url=crowdstrike_url)
+        self.intel = Intel(client_id=config.get('falcon', 'client_id'),
+                           client_secret=config.get('falcon', 'client_secret'),
+                           base_url=self.__class__.base_url(),
+                           user_agent=f"chronicle-intel-bridge/{__version__}"
+                           )
 
     @classmethod
     def base_url(cls):

diff --git a/ccib/version.py b/ccib/version.py
@@ -0,0 +1 @@
+__version__ = '0.0.1'