Merge pull request #49 from equinor/119166

Added console script

Author: oihaa

setup.py

@@ -15,7 +15,7 @@
         'Programming Language :: Python :: 3.8'
         ],
     version='0.1.1',
-    author='Lindvar Lagran',
+    author='Equinor ASA',
     author_email='llag@equinor.com',
     install_requires=[
                     'requests',
@@ -28,4 +28,7 @@
     package_dir={"": "src"},
     include_package_data=True,
     zip_safe=False,
+    entry_points={
+        "console_scripts": ["sumologin = sumo.wrapper.login:main"]
+        }
 )

src/sumo/wrapper/_auth.py

@@ -29,6 +29,7 @@ def __init__(self, client_id, resource_id, authority=AUTHORITY_URI, client_crede
         if self._cache_available():
             if not self.accounts:
                 print("Token cache found but have no accounts")
+                self._oauth_device_code()
             else:
                 print("Get token and maybe refresh")
                 self._oauth_get_token_silent()

src/sumo/wrapper/login.py

@@ -0,0 +1,27 @@
+import os
+import sys
+from argparse import ArgumentParser
+from sumo.wrapper import CallSumoApi
+
+
+def get_parser() -> ArgumentParser:
+    parser = ArgumentParser(
+        description="Login to Sumo on azure"
+    )
+    parser.add_argument(
+        "--env", dest="env", action='store', default="prod", help="Environment to log into"
+    )
+    return parser
+
+
+def main():
+    args = get_parser().parse_args()
+
+    env = args.env
+
+    print("Login to Sumo environment: " + env)
+
+    id = CallSumoApi(env=env)
+
+if __name__ == '__main__':
+    main()