use selenium to capture picture

smilejay · smilejay · commit 822ceb7089bf · 2016-01-14T23:36:55.000+08:00
from a scrolled-down full web page
diff --git a/py2016/selenium_capture_firefox.py b/py2016/selenium_capture_firefox.py
@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+
+from selenium import webdriver
+import time
+
+
+def capture(url, img_file="test1.png"):
+    firefox = webdriver.Firefox()
+    firefox.set_window_size(1200, 900)
+    firefox.get(url)
+    firefox.execute_script("""
+        (function () {
+            var y = 0;
+            var step = 100;
+            window.scroll(0, 0);
+
+            function f() {
+                if (y < document.body.scrollHeight) {
+                    y += step;
+                    window.scroll(0, y);
+                    setTimeout(f, 50);
+                } else {
+                    window.scroll(0, 0);
+                    document.title += "scroll-done";
+                }
+            }
+
+            setTimeout(f, 1000);
+        })();
+    """)
+
+    for i in xrange(30):
+        if "scroll-done" in firefox.title:
+            break
+        time.sleep(1)
+
+    firefox.save_screenshot(img_file)
+    firefox.close()
+
+
+if __name__ == "__main__":
+    capture("http://www.taobao.com")
