High res previews, fix lorentz

Author: kevinlinxc

capture_all_latex_previews.py

@@ -1,19 +1,21 @@
+"""
+Script to capture screenshots of all the formulas using Selenium.
+This shit doesn't work; see https://stackoverflow.com/questions/76553944/
+"""
 import os
 import time
 
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.chrome.service import Service as ChromeService
-from selenium.webdriver.common.keys import Keys
 from webdriver_manager.chrome import ChromeDriverManager
 import pyautogui
-import PIL
-
-dry_run = False
 
+dry_run = False  # if False, don't save anything
+save_to_temp_folder = True  # if True, save to temp folder instead of content/formulas (for previewing)
 driver = webdriver.Chrome(service=ChromeService(ChromeDriverManager().install()))
 
-driver.get('http://localhost:1313/formulas')
+driver.get('http://stemformulas.com/formulas')
 driver.implicitly_wait(10)
 
 def zoom_in(times=1):
@@ -29,11 +31,12 @@ def zoom_in(times=1):
 print(f"Found {page_count} pages of formulas")
 
 # increase zoom to 400% for higher quality screenshots
-# zoom_in(8) # TODO, screenshots become weird when zoomed in, but we want this for higher quality screenshots
+# zoom_in(4)
+
 for i in range(1, page_count + 1): # iterate over pages
     print("Visiting page: ", i)
     # visit formulas page
-    driver.get(f'http://localhost:1313/formulas/page/{i}')
+    driver.get(f'http://stemformulas.com/formulas/page/{i}')
 
     # Find the section with class "grid-container mt-6"
     section = driver.find_element(By.CSS_SELECTOR, 'section.grid-container.mt-6')
@@ -53,16 +56,27 @@ def zoom_in(times=1):
         x_path = f"//*[@id=\"main-content\"]/section[2]/a[{i+1}]/div[1]"
 
         div = anchor.find_element(By.XPATH, x_path)
-
+        div.location_once_scrolled_into_view # scroll into view
         # http://localhost:1313/formulas/<folder_name>/
         folder_name = href.split('/')[-2]
         output_image_path = os.path.join("content", "formulas", folder_name, "preview.png")
 
         if dry_run:
             print(f"Would have screenshot div {div} to {output_image_path}")
+        elif save_to_temp_folder:
+            if not os.path.exists("temp"):
+                os.mkdir("temp")
+            if not os.path.exists(os.path.join("temp", folder_name)):
+                os.mkdir(os.path.join("temp", folder_name))
+            new_path = os.path.join("temp", folder_name, "preview.png")
+            bits = div.screenshot_as_png
+            with open(new_path, 'wb') as f:
+                f.write(bits)
+            print(f"Screenshot div {div} to {new_path}")
+            time.sleep(1)
         else:
             bits = div.screenshot_as_png
             with open(output_image_path, 'wb') as f:
                 f.write(bits)
             print(f"Screenshot div {div} to {output_image_path}")
-            time.sleep(0.7) # if we don't sleep, elements become stale
+            time.sleep(1) # if we don't sleep, elements become stale