70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
|
import os
|
||
|
import re
|
||
|
import time
|
||
|
from typing import Optional
|
||
|
|
||
|
from selenium import webdriver
|
||
|
from selenium.webdriver.common.by import By
|
||
|
from selenium.webdriver.support import expected_conditions as EC
|
||
|
from selenium.webdriver.firefox.options import Options
|
||
|
from selenium.webdriver.firefox.service import Service
|
||
|
from selenium.webdriver.support.wait import WebDriverWait
|
||
|
from bs4 import BeautifulSoup
|
||
|
|
||
|
from nebula_rss import NebulaVideo
|
||
|
|
||
|
|
||
|
class NebulaLoader:
|
||
|
def __init__(
|
||
|
self,
|
||
|
username: str,
|
||
|
password: str,
|
||
|
driver_path: Optional[os.PathLike] = None
|
||
|
):
|
||
|
self.username = username
|
||
|
self.password = password
|
||
|
service = None
|
||
|
if driver_path:
|
||
|
service = Service(driver_path())
|
||
|
options = Options()
|
||
|
options.headless = True
|
||
|
self.driver = webdriver.Firefox(service=service, options=options)
|
||
|
self.driver.implicitly_wait(10) # seconds
|
||
|
#NebulaVideo('', '', '', None)
|
||
|
|
||
|
def load(self):
|
||
|
self.driver.get('https://nebula.app/login')
|
||
|
|
||
|
username_input = '//*[@name="email"]'
|
||
|
password_input = '//*[@name="password"]'
|
||
|
login_submit = '//*[@id="NebulaApp"]/div[2]/div[2]/div[1]/div/form/button'
|
||
|
|
||
|
self.driver.find_element(By.XPATH, username_input).send_keys(self.username)
|
||
|
self.driver.find_element(By.XPATH, password_input).send_keys(self.password)
|
||
|
self.driver.find_element(By.XPATH, login_submit).click()
|
||
|
|
||
|
delay = 3
|
||
|
# wait for "My shows" link
|
||
|
wait = WebDriverWait(self.driver, delay)
|
||
|
myshows = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, 'menu a[href|="/myshows"]')))
|
||
|
myshows.click()
|
||
|
myshows = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, 'h2')))
|
||
|
|
||
|
video_links = []
|
||
|
count_remaining = 5
|
||
|
follower_error_re = re.compile("You aren't following any creators yet.*")
|
||
|
while not video_links and count_remaining > 0:
|
||
|
time.sleep(2)
|
||
|
soup = BeautifulSoup(self.driver.page_source)
|
||
|
follower_error = False
|
||
|
follower_error = [p for p in soup.find_all('p') if p.find(string=follower_error_re)]
|
||
|
if follower_error:
|
||
|
print('Error loading videos, reloading page')
|
||
|
self.driver.refresh()
|
||
|
count_remaining = 5
|
||
|
else:
|
||
|
video_links = [a for a in soup.find_all('a') if a.get('href').startswith('/videos/')]
|
||
|
count_remaining -= 1
|
||
|
#v = nebula_video.NebulaVideo('', '', '', None)
|
||
|
print(video_links)
|