getvideourl.py
· 740 B · Python
Raw
#!/usr/bin/env python3
# pip install selenium, you'll also need firefox installed
import contextlib
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from sys import argv, stderr, stdin
if len(argv) > 1:
if argv[1] == "-":
urls = [line for line in stdin]
else:
urls = argv[1:]
with contextlib.closing(webdriver.Firefox()) as driver:
for i in urls:
driver.get(i)
WebDriverWait(driver, 10).until(lambda driver: driver.find_element_by_tag_name('video'))
print(driver.title, file=stderr)
print(driver.find_element_by_tag_name("video").get_attribute("src"))
| 1 | #!/usr/bin/env python3 |
| 2 | # pip install selenium, you'll also need firefox installed |
| 3 | |
| 4 | import contextlib |
| 5 | from selenium import webdriver |
| 6 | from selenium.webdriver.support.ui import WebDriverWait |
| 7 | from selenium.webdriver.support import expected_conditions as EC |
| 8 | from sys import argv, stderr, stdin |
| 9 | |
| 10 | if len(argv) > 1: |
| 11 | if argv[1] == "-": |
| 12 | urls = [line for line in stdin] |
| 13 | else: |
| 14 | urls = argv[1:] |
| 15 | with contextlib.closing(webdriver.Firefox()) as driver: |
| 16 | for i in urls: |
| 17 | driver.get(i) |
| 18 | WebDriverWait(driver, 10).until(lambda driver: driver.find_element_by_tag_name('video')) |
| 19 | print(driver.title, file=stderr) |
| 20 | print(driver.find_element_by_tag_name("video").get_attribute("src")) |