Added .env for API key and updated request headers

Author: saaskilahtij

.gitignore

@@ -0,0 +1 @@
+.env

__pycache__/indicators.cpython-311.pyc

@@ -4,10 +4,12 @@
 Author: Cody Winkler (twitter: c2thewinkler | github: m0rph-1)
 Date: 2/12/2020
 """
+import os
 import time
 import requests
 import json
 import sys
+from dotenv import load_dotenv
 from termcolor import colored
 from base64 import b64decode
 from indicators import *
@@ -27,9 +29,13 @@ def parse_options():
     args = parser.parse_args()
     return args
 
-def get_code_snippet(git_url_key):
+def get_code_snippet(git_url_key, authorization):
+    authorization
+    headers = {
+        "Authorization":"Bearer {authorization}"
+    }
 
-    this_req = requests.get(git_url_key, verify=True)
+    this_req = requests.get(git_url_key, verify=True, headers=headers)
     json_data = json.loads(this_req.content)
     if this_req.status_code == 200:
         repo_code = str(b64decode(json_data["content"]).decode('utf-8')).split('\n')
@@ -39,14 +45,25 @@ def get_code_snippet(git_url_key):
                      print(str(repo_code[i]))
 
 def check_rate_limit():
-
-    this_req = requests.get("https://api.github.com/rate_limit", verify=True)
+    this_req = requests.get("https://api.github.com/rate_limit", verify=True, headers=headers)
     json_data = json.loads(this_req.content)
     pprint(json_data["rate"])
     return
 
 def main(args):
 
+    load_dotenv()
+
+    api_key = os.getenv("API_KEY")
+
+    if api_key == None or api_key == '':
+        print(colored("[!] API key is not set. Specify it in .env.", "red"))
+        exit(1)
+
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+
     separator = "-"*100
 
     if args.rate:
@@ -60,7 +77,7 @@ def main(args):
         try:
 
             github_api = ("https://api.github.com/search/code?q=user:{} {}").format(args.org, args.query)
-            this_req = requests.get(github_api, verify=True)
+            this_req = requests.get(github_api, verify=True, headers=headers)
             json_data = json.loads(this_req.content)
 
             if this_req.status_code == 200:
@@ -78,7 +95,7 @@ def main(args):
 
                             print(colored("This should be printed if -g is specified", "yellow"))
                             git_url_key = key["git_url"]
-                            get_code_snippet(git_url_key)
+                            get_code_snippet(git_url_key, )
 
                     except KeyError:
                         print(colored("[-] Didn't find anything. Moving on!", "red"))

github_autopwn.py

@@ -0,0 +1 @@
+add token to .env and test it