selenium执行js滚动到底部

import time
from selenium import webdriver

# 设置等待时间
TIME_WAIT = 3
# 滚动到浏览器顶部
js_top = "var q=document.documentElement.scrollTop=0"

# 滚动到浏览器底部
js_bottom = "window.scrollTo(0,document.body.scrollHeight)"

driver = webdriver.Chrome()
driver.get(url地址)
driver.execute_script(js_bottom)
time.sleep(TIME_WAIT)

driver.execute_script(js_top)
time.sleep(TIME_WAIT)
driver.quit()