from selenium import webdriver import sys import time import dbhelper from video import * from selenium.common.exceptions import NoSuchElementException driver = webdriver.Chrome("chromedriver.exe") # 数据库连接 conn, c = dbhelper.connect_db() if len(sys.argv) < 3: start_id = 12872143 end_id = 12972143 else: start_id = int(sys.argv[1]) end_id = int(sys.argv[2]) print("Start_ID:",sys.argv[1]) print("End_ID:",sys.argv[2]) base_url = "URL:Hidden" i = 0 while (start_id+i)<end_id: driver.get(base_url + str(start_id + i)) try: video = driver.find_element_by_id("post_video") except NoSuchElementException as msg: print("ID:",str(start_id + i)," | 非视频页面!")
def connect_to_db(): conn, c = dbhelper.connect_db() return conn, c