add print_page options

pwncollege · Jan 28, 2025 · b46df8b · b46df8b
1 parent 505eed0
commit b46df8b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pwnshop/templates/base/selenium.py b/pwnshop/templates/base/selenium.py
@@ -62,6 +62,11 @@
 browser.find_element(By.NAME, "submit").submit()
 {% endif %}
 
+{% if challenge.print_page %}
+print("Retrieved the following HTML:")
+print(browser.page_source)
+{% endif %}
+
 {% if challenge.reward_alert %}
 try:
     WebDriverWait(browser, 1).until(EC.alert_is_present())