Better waiting system

This commit is contained in:
Pablo Ferreiro
2022-03-11 19:44:57 +01:00
parent b9300a8791
commit 912e09882f

View File

@@ -1,24 +1,27 @@
from os import getenv from os import getenv
from time import sleep
from selenium import webdriver from selenium import webdriver
from selenium_stealth import stealth from selenium_stealth import stealth
from selenium.webdriver.chrome.options import Options from selenium.webdriver.chrome.options import Options
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from urllib import parse from urllib import parse
from TikTokSigner.Utils import Utils from TikTokSigner.Utils import Utils
class Signer: class Signer:
DEFAULT_URL = 'https://www.tiktok.com/@tiktok/?lang=en' DEFAULT_URL = 'https://www.tiktok.com/@tiktok/?lang=en'
USER_AGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (Windows NT 10.0; Win64; x64) Chrome/90.0.4430.85 Safari/537.36' USER_AGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (Windows NT 10.0; Win64; x64) Chrome/90.0.4430.85 Safari/537.36'
TIMEOUT = 30
driver: webdriver.Chrome driver: webdriver.Chrome
signature = '' signature = ''
xttparams = '' xttparams = ''
def __init__(self): def __init__(self):
print('Starting Browser')
options = Options() options = Options()
path = getenv('GOOGLE_CHROME_SHIM', '') path = getenv('GOOGLE_CHROME_SHIM', '')
options._binary_location = path options._binary_location = path
options.add_argument("start-maximized")
options.add_argument("--headless") options.add_argument("--headless")
options.add_argument("--disable-gpu") options.add_argument("--disable-gpu")
options.add_argument("--no-sandbox") options.add_argument("--no-sandbox")
@@ -38,7 +41,7 @@ class Signer:
) )
self.driver.get(self.DEFAULT_URL) self.driver.get(self.DEFAULT_URL)
sleep(2) WebDriverWait(self.driver, self.TIMEOUT).until(EC.presence_of_element_located((By.ID, 'app')))
# Load scripts # Load scripts
with open('./js/signature.js') as f: with open('./js/signature.js') as f: