Add feature to set custom API key

Author: yasht01

src/arguments/utility.py

@@ -7,6 +7,12 @@
 from .save import SaveSearchResults
 from .markdown import MarkdownRenderer
 
+# Required for OAuth
+import json
+import webbrowser
+from oauthlib.oauth2 import MobileApplicationClient
+from requests_oauthlib import OAuth2Session
+
 console = Console()
 
 
@@ -85,4 +91,38 @@ def get_ans(self, questions_list):
 
                     console.print(output_text)
             ans.append(json_ans_data["items"])
-        return ans
+        return ans
+
+    # Get an access token and extract to a JSON file "access_token.json"
+    def setCustomKey(self):
+        client_id = 20013
+
+        # scopes possible values:
+        # read_inbox - access a user's global inbox
+        # no_expiry - access_token's with this scope do not expire
+        # write_access - perform write operations as a user 
+        # private_info - access full history of a user's private actions on the site
+        scopes = 'read_inbox'
+
+        authorization_url = 'https://stackoverflow.com/oauth/dialog'
+        redirect_uri = 'https://stackexchange.com/oauth/login_success'
+
+        # Create an OAuth session and open the auth_url in a browser for the user to authenticate
+        stackApps = OAuth2Session(client=MobileApplicationClient(client_id=client_id), scope=scopes, redirect_uri=redirect_uri)
+        auth_url, state = stackApps.authorization_url(authorization_url)
+        print(f"Visit this page in your browser: {auth_url}")
+        webbrowser.open(auth_url)
+
+        # Extract the access token data and save to a variable
+        callback_url = input("Paste URL after authenticating here: ")
+        accessTokenData = stackApps.token_from_fragment(callback_url)
+
+        # Store the access token data in a JSON file
+        jsonDict = {
+            "access_token": accessTokenData['access_token'],
+            "expires": accessTokenData['expires'],
+            "state": accessTokenData['state']
+        }
+
+        with open('access_token.json', 'w') as jsonFile:
+            json.dump(jsonDict, jsonFile)