💡 add comments and removed empty lines

Author: officialpm

Scripts/Web_Scrappers/CodeChef_Scraper/codechef_ps.py

@@ -1,39 +1,42 @@
+# Developed By Parth Maniar
+# https://github.com/officialpm
+
+# import libs
+
 import os
 import sys
 
 from bs4 import BeautifulSoup
 from selenium import webdriver
 from webdriver_manager.chrome import ChromeDriverManager
 
-if not os.path.exists("problems"):
+if not os.path.exists("problems"):  # create problems folder if not exists
     os.makedirs("problems")
 
 options = webdriver.ChromeOptions()
-options.add_argument("headless")
+options.add_argument("headless")  # headless chrome option
 
 
 def scrapeCodeChef(code):
     try:
         url = f"https://www.codechef.com/problems/{code}"
-
-        browser = webdriver.Chrome(ChromeDriverManager().install(), options=options)
-
+        browser = webdriver.Chrome(
+            ChromeDriverManager().install(), options=options
+        )  # install and open chrome driver
         print("[OPENING] - ", code)
-        browser.get(url)
-
+        browser.get(url)  # open CodeChef Problem  URL
         print("[SCRAPING] - ", code)
-        soup = BeautifulSoup(browser.page_source, features="html.parser")
-
-        head = soup.find_all(["h1"])
-        check = soup.find_all(["p", "h3", "ul", "pre"])
-
-        f = open("problems/" + url[34:] + ".txt", "a")
-        f.write(head[1].text[:-7] + "\n")
-        for i in range(8, len(check) - 18):
-            f.write(check[i].text + "\n")
-        f.close()
-
-        print(f"[FINISHED] - File saved - problems/{code}.txt")    
+        soup = BeautifulSoup(
+            browser.page_source, features="html.parser"
+        )  # parse page source
+        head = soup.find_all(["h1"])  # find all h1 tags
+        body = soup.find_all(["p", "h3", "ul", "pre"])  # find all p, h3, ul, pre tags
+        f = open("problems/" + url[34:] + ".txt", "a")  # open .txt file
+        f.write(head[1].text[:-7] + "\n")  # write title
+        for i in range(8, len(body) - 18):
+            f.write(body[i].text + "\n")  # write body
+        f.close()  # save file
+        print(f"[FINISHED] - File saved - problems/{code}.txt")
     except Exception:
         print("Cannot Find CodeChef Problem! - " + code)
         exit(0)
@@ -43,7 +46,6 @@ def scrapeCodeChef(code):
     try:
         code = sys.argv[1]
     except Exception:
-        print('Please Enter A CodeChef Problem Code as a',
-              'Command-Line Argument!')
+        print("Please Enter A CodeChef Problem Code as a", "Command-Line Argument!")
         exit(0)
     scrapeCodeChef(code)