add support for Rapid API instance of judge0

- add authenticate() method for adding API_KEY
- change base API URLs

Author: Bhupesh-V

coderunner/coderunner.py

@@ -52,7 +52,13 @@
     "max_file_size": "1024",
 }
 
-API_URL = "https://api.judge0.com/submissions/"
+HEADERS = {
+    "x-rapidapi-host": "judge0.p.rapidapi.com",
+    "useQueryString": True
+}
+
+HOST_URL = "https://judge0.p.rapidapi.com/"
+API_URL = "https://judge0.p.rapidapi.com/submissions/"
 FIELDS = "?fields=stdout,memory,time,status,stderr,exit_code,created_at"
 
 
@@ -135,7 +141,7 @@ def __readStatus(self, token: str):
         Check Submission status
         """
         while True:
-            req = urllib.request.Request(API_URL + token["token"] + FIELDS)
+            req = urllib.request.Request(API_URL + token["token"] + FIELDS, headers=HEADERS)
             with urllib.request.urlopen(req) as response:
                 req = response.read()
 
@@ -161,13 +167,17 @@ def __submit(self):
         api_params["source_code"] = self.source
 
         post_data = urllib.parse.urlencode(api_params).encode("ascii")
-        req = urllib.request.Request(API_URL, post_data)
+        req = urllib.request.Request(API_URL, post_data, headers=HEADERS)
         with urllib.request.urlopen(req) as response:
             req = response.read()
         token = json.loads(req.decode("utf-8"))
 
         return token
 
+    def authenticate(self, key: str):
+        self.API_KEY = key
+        HEADERS["x-rapidapi-key"] = self.API_KEY
+
     def getSubmissionDate(self):
         """Submission date/time of program"""
         return self.__response["created_at"]

demo.py

@@ -1,30 +1,39 @@
 from coderunner import coderunner
 import pprint
+import os
+
+from dotenv import load_dotenv
+load_dotenv()
 
 source_code = "testfiles/" + "test_python_input.py"
 language = "Python3"
 output = "testfiles/output/" + "output2.txt"
 Input = "testfiles/input/" + "input.txt"
+
+
+API_KEY = os.environ["API_KEY"]
+
 r = coderunner.code(source_code, language, output, Input)
+r.authenticate(API_KEY)
 
-r2 = coderunner.code("print(\"yo\")", "Python3", "YO", path=False)
+# r2 = coderunner.code("print(\"yo\")", "Python3", "YO", path=False)
 
-# run the code
+# # run the code
 r.run()
 
 print("Run r :")
 print("Status : " + r.getStatus())
 
-r2.run()
+# r2.run()
 
-print("Run r2 :")
-print("Status : " + r2.getStatus())
+# print("Run r2 :")
+# print("Status : " + r2.getStatus())
 
-# check if any error occured
-if r.getError() is not None:
-    pprint.pprint("Error : " + r.getError())
-else:
-    print("Standard Output : ")
-    pprint.pprint(r.getOutput())
-print("Execution Time : " + r.getTime())
-print("Memory : " + str(r.getMemory()))
+# # check if any error occured
+# if r.getError() is not None:
+#     pprint.pprint("Error : " + r.getError())
+# else:
+#     print("Standard Output : ")
+#     pprint.pprint(r.getOutput())
+# print("Execution Time : " + r.getTime())
+# print("Memory : " + str(r.getMemory()))

requirements.txt

@@ -2,3 +2,4 @@ isort
 black
 flake8
 pylint
+python-dotenv