WIP oie implementation, oauth2 only, authorization code flow.

.gitignore

@@ -91,6 +91,7 @@ venv/
 ENV/
 env.bak/
 venv.bak/
+.vscode/
 
 # Spyder project settings
 .spyderproject

tokendito/__init__.py

@@ -5,10 +5,11 @@
 import os
 from os.path import expanduser
 import sys
+from requests import Session
 
 from platformdirs import user_config_dir
 
-__version__ = "2.1.2"
+__version__ = "2.2.0"
 __title__ = "tokendito"
 __description__ = "Get AWS STS tokens from Okta SSO"
 __long_description_content_type__ = "text/markdown"
@@ -53,6 +54,7 @@ class Config(object):
             password="",
             mfa=None,
             mfa_response=None,
+            oauth_client_id=None,
             tile=None,
             org=None,
         ),
@@ -132,3 +134,12 @@ def get_defaults(self):
 
 
 config = Config()
+Session = Session()
+Session.headers.update(
+    {
+        "User-Agent": f"tokendito/{__version__}",
+        "content-type": "application/json",
+        "accept": "application/json",
+    }
+)
+

tokendito/__main__.py

@@ -12,7 +12,7 @@ def main(args=None):  # needed for console script
 
         path = os.path.dirname(os.path.dirname(__file__))
         sys.path[0:0] = [path]
-    from tokendito.tool import cli
+    from tokendito.user import cli
 
     try:
         return cli(args)

tokendito/aws.py

@@ -46,7 +46,7 @@ def get_output_types():
     return ["json", "text", "csv", "yaml", "yaml-stream"]
 
 
-def authenticate_to_roles(urls, cookies=None):
+def authenticate_to_roles(urls, cookies):
     """Authenticate AWS user with saml.
 
     :param urls: list of tuples or tuple, with tiles info