Add command line argument to setup custom API key

Author: yasht01

main.py

@@ -34,6 +34,11 @@
                     help="Save answer to a file",
                     action="store_true")
 
+parser.add_argument("-c",
+                    "--custom",
+                    help="Set a custom API key",
+                    action="store_true")
+
 ARGV = parser.parse_args()
 
 search_flag = Search(ARGV)

requirements.txt

@@ -5,3 +5,5 @@ requests==2.24.0
 termcolor==1.1.0
 urllib3==1.25.10
 rich==9.9.0
+oauthlib==3.1.0
+requests-oauthlib==1.3.0

src/arguments/search.py

@@ -39,6 +39,8 @@ def search_args(self):
                 webbrowser.open(f"{url}?title={self.arguments.new}")
             else:
                 webbrowser.open(url)
+        elif self.arguments.custom:
+            self.utility_object.setCustomKey()
 
     def search_for_results(self, save=False):
         queries = ["What do you want to search", "Tags"]