Small updates to update_questions.py (#233)

Author: seanprashad

cron/update_questions.py

@@ -38,13 +38,12 @@ def get_question_metadata(api, title_slug):
 
     try:
         response = api.graphql_post(body=graphql_request)
+        return response
     except ApiException as e:
         print(
             f'Exception occurred when contacting the Leetcode GraphQL API: ${e}')
         exit()
 
-    return response
-
 
 def construct_company_tag_list(company_tags_json, sections):
     companies = []
@@ -60,13 +59,26 @@ def construct_company_tag_list(company_tags_json, sections):
     return sorted(companies, key=lambda d: d['frequency'], reverse=True)
 
 
-def update_question_metadata(question, title, difficulty, companies, is_premium):
-    print(f"🔄 Updating question metadata for {title}")
+def update_question_metadata(question, response):
+    print(f'''🔄 Updating question metadata for {question["title"]}''')
+
+    question_title = response.data.question.title
+    question_difficulty = response.data.question.difficulty
+    question_company_tags = json.loads(
+        response.data.question.company_tag_stats)
+    question_is_premium = response.data.question.is_paid_only
+
+    # Retrieve companies who have asked this question for the following two
+    # company_tag_stat sections:
+    #   1. 0-6 months
+    #   2. 6 months to 1 year
+    companies = construct_company_tag_list(
+        question_company_tags, ["1", "2"])
 
-    question["title"] = title
-    question["difficulty"] = difficulty
+    question["title"] = question_title
+    question["difficulty"] = question_difficulty
     question["companies"] = companies
-    question["premium"] = is_premium
+    question["premium"] = question_is_premium
 
 
 def read_questions(file_name):
@@ -106,21 +118,7 @@ def main(file_name):
 
         response = get_question_metadata(api, title_slug)
 
-        question_title = response.data.question.title
-        question_difficulty = response.data.question.difficulty
-        question_company_tags = json.loads(
-            response.data.question.company_tag_stats)
-        question_is_premium = response.data.question.is_paid_only
-
-        # Retrieve companies who have asked this question within the following two
-        # company_tag_stat sections:
-        #   1. 0-6 months
-        #   2. 6 months to 1 year
-        companies = construct_company_tag_list(
-            question_company_tags, ["1", "2"])
-
-        update_question_metadata(question, question_title, question_difficulty,
-                                 companies, question_is_premium)
+        update_question_metadata(question, response)
 
     write_questions(file_name, questions)