From 3e0e4b6c8fc5d63c417643e283c3ff52c2ca0312 Mon Sep 17 00:00:00 2001
From: Ryan Petrello <rpetrell@redhat.com>
Date: Fri, 28 Feb 2020 11:12:26 -0500
Subject: [PATCH] cli: add the ability to specify a token description w/ `awx
 login`

see: https://github.com/ansible/awx/issues/6122
---
 awxkit/awxkit/api/pages/base.py | 3 ++-
 awxkit/awxkit/cli/resource.py   | 3 +++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/awxkit/awxkit/api/pages/base.py b/awxkit/awxkit/api/pages/base.py
index 482895764e..e3e64d7db7 100644
--- a/awxkit/awxkit/api/pages/base.py
+++ b/awxkit/awxkit/api/pages/base.py
@@ -140,6 +140,7 @@ class Base(Page):
     load_default_authtoken = load_authtoken
 
     def get_oauth2_token(self, username='', password='', client_id=None,
+                         description='AWX CLI',
                          client_secret=None, scope='write'):
         default_cred = config.credentials.default
         username = username or default_cred.username
@@ -176,7 +177,7 @@ class Base(Page):
             resp = self.connection.post(
                 '/api/v2/users/{}/personal_tokens/'.format(username),
                 json={
-                    "description": "Tower CLI",
+                    "description": description,
                     "application": None,
                     "scope": scope
                 },
diff --git a/awxkit/awxkit/cli/resource.py b/awxkit/awxkit/cli/resource.py
index 1fea0133a8..20e08c6f18 100644
--- a/awxkit/awxkit/cli/resource.py
+++ b/awxkit/awxkit/cli/resource.py
@@ -76,6 +76,7 @@ class Login(CustomCommand):
 
     def handle(self, client, parser):
         auth = parser.add_argument_group('OAuth2.0 Options')
+        auth.add_argument('--description', help='description of the generated OAuth2.0 token', metavar='TEXT')
         auth.add_argument('--conf.client_id', metavar='TEXT')
         auth.add_argument('--conf.client_secret', metavar='TEXT')
         auth.add_argument(
@@ -90,6 +91,8 @@ class Login(CustomCommand):
             'client_secret': getattr(parsed, 'conf.client_secret', None),
             'scope': getattr(parsed, 'conf.scope', None),
         }
+        if getattr(parsed, 'description', None):
+            kwargs['description'] = parsed.description
         try:
             token = api.Api().get_oauth2_token(**kwargs)
         except Exception as e: