[F] Fix mteam login
This commit is contained in:
+115
-51
@@ -12,22 +12,18 @@ USER_DATA_DIR = BASE_DIR / "data/browser_profile"
|
||||
CONFIG_FILE = BASE_DIR / "config.toml"
|
||||
|
||||
def load_config():
|
||||
if not CONFIG_FILE.exists():
|
||||
print(f"Error: Configuration file '{CONFIG_FILE}' not found.")
|
||||
sys.exit(1)
|
||||
return tomllib.loads(CONFIG_FILE.read_text())["m-team"]
|
||||
|
||||
# with CONFIG_FILE.open("rb") as f:
|
||||
# config = tomllib.load(f)
|
||||
config = tomllib.loads(CONFIG_FILE.read_text())
|
||||
def get_browser_context(p, headless=False):
|
||||
return p.chromium.launch_persistent_context(user_data_dir=USER_DATA_DIR, headless=headless, channel="chrome")
|
||||
|
||||
if "m-team" not in config:
|
||||
print("Error: '[m-team]' section not found in config.toml")
|
||||
sys.exit(1)
|
||||
def ensure_logged_in(page):
|
||||
# 1. url is /login, 2. login form is visible
|
||||
if page.url.startswith(LOGIN_URL) or page.is_visible("input#username"):
|
||||
login(page, load_config())
|
||||
return True
|
||||
|
||||
return config["m-team"]
|
||||
|
||||
def run():
|
||||
config = load_config()
|
||||
def login(page, config):
|
||||
username = config.get("username")
|
||||
password = config.get("password")
|
||||
otp_key = config.get("otp_key")
|
||||
@@ -36,18 +32,6 @@ def run():
|
||||
print("Error: Missing username, password, or otp_key in config.toml")
|
||||
sys.exit(1)
|
||||
|
||||
print(f"Launching browser with persistent profile at: {USER_DATA_DIR}")
|
||||
|
||||
with sync_playwright() as p:
|
||||
# Launch a persistent context to save cookies
|
||||
browser = p.chromium.launch_persistent_context(
|
||||
user_data_dir=USER_DATA_DIR,
|
||||
headless=False, # Set to True if you don't want to see the browser
|
||||
channel="chrome", # Optional: Use 'msedge' or remove to use bundled chromium
|
||||
)
|
||||
|
||||
page = browser.new_page()
|
||||
|
||||
print(f"Navigating to {LOGIN_URL}...")
|
||||
page.goto(LOGIN_URL)
|
||||
page.wait_for_load_state("networkidle")
|
||||
@@ -57,40 +41,25 @@ def run():
|
||||
print("Login form not found. You might already be logged in.")
|
||||
print("Checking page title...")
|
||||
print(f"Current Title: {page.title()}")
|
||||
else:
|
||||
return
|
||||
|
||||
print("Login form detected. Attempting to log in...")
|
||||
|
||||
# 1. Fill Username
|
||||
page.fill("input#username", username)
|
||||
|
||||
# 2. Fill Password
|
||||
page.fill("input#password", password)
|
||||
|
||||
# 3. Click Submit
|
||||
submit_selector = 'button[type="submit"]'
|
||||
page.click(submit_selector)
|
||||
|
||||
print("Credentials submitted. Waiting for OTP field...")
|
||||
|
||||
# 4. Handle OTP
|
||||
try:
|
||||
# Wait up to 10 seconds for the OTP input to appear
|
||||
page.wait_for_selector("input#otpCode", timeout=10000)
|
||||
|
||||
page.wait_for_selector("input#otp-code", timeout=10000)
|
||||
print("Generating OTP code from provided key...")
|
||||
# Generate TOTP code using the secret key
|
||||
totp = pyotp.TOTP(otp_key.replace(" ", "")) # Sanitize spaces just in case
|
||||
|
||||
totp = pyotp.TOTP(otp_key.replace(" ", ""))
|
||||
current_otp = totp.now()
|
||||
print(f"Generated Code: {current_otp}")
|
||||
|
||||
# Fill the OTP
|
||||
page.fill("input#otpCode", current_otp)
|
||||
|
||||
# Press Enter to submit
|
||||
page.press("input#otpCode", "Enter")
|
||||
|
||||
page.fill("input#otp-code", current_otp)
|
||||
page.press("input#otp-code", "Enter")
|
||||
print("OTP Submitted.")
|
||||
|
||||
except Exception as e:
|
||||
print(f"OTP field did not appear or an error occurred: {e}")
|
||||
print("Maybe login failed or OTP wasn't required?")
|
||||
@@ -99,10 +68,105 @@ def run():
|
||||
page.wait_for_timeout(5000)
|
||||
|
||||
print(f"Final URL: {page.url}")
|
||||
print("Script finished. Cookies are saved in the profile folder.")
|
||||
print("Login process finished.")
|
||||
|
||||
# Close the browser
|
||||
browser.close()
|
||||
def get_torrents(page, imdb: str):
|
||||
ensure_logged_in(page)
|
||||
|
||||
if not imdb.startswith("https://www.imdb.com/title/"):
|
||||
imdb = f"https://www.imdb.com/title/{imdb}"
|
||||
|
||||
url = f"https://kp.m-team.cc/mdb/title?source=imdb&imdb={urllib.parse.quote(imdb)}"
|
||||
print(f"Navigating to {url}...")
|
||||
page.goto(url)
|
||||
page.wait_for_load_state("networkidle")
|
||||
|
||||
|
||||
|
||||
|
||||
def download(page, tid):
|
||||
url = f"https://kp.m-team.cc/detail/{tid}"
|
||||
print(f"Navigating to {url}...")
|
||||
page.goto(url)
|
||||
page.wait_for_load_state("networkidle")
|
||||
|
||||
# Check if we are logged in (if we see the login form, we are not)
|
||||
if page.is_visible("input#username"):
|
||||
print("Error: Not logged in. Please run 'login' command first.")
|
||||
return
|
||||
|
||||
try:
|
||||
print("Looking for download button...")
|
||||
# Button selector based on user request: <button ...><span>下載</span></button>
|
||||
# We use a role selector combined with name for robustness
|
||||
download_button = page.get_by_role("button", name="下載")
|
||||
|
||||
if not download_button.is_visible():
|
||||
# Fallback to specific class if role text fails, though "下載" should work.
|
||||
# The user provided class: ant-btn css-fjnik7 ant-btn-primary ant-btn-color-primary ant-btn-variant-solid
|
||||
# But classes like css-fjnik7 might be dynamic.
|
||||
print("Download button not found by role/name. Trying generic selector...")
|
||||
# Try a looser selector
|
||||
download_button = page.locator("button:has-text('下載')")
|
||||
|
||||
if not download_button.is_visible():
|
||||
print("Error: Download button not found on page.")
|
||||
return
|
||||
|
||||
print("Clicking download button...")
|
||||
with page.expect_download() as download_info:
|
||||
download_button.click()
|
||||
|
||||
download = download_info.value
|
||||
print(f"Download started: {download.suggested_filename}")
|
||||
|
||||
# Save to current directory
|
||||
save_path = Path.cwd() / download.suggested_filename
|
||||
download.save_as(save_path)
|
||||
print(f"Successfully saved to: {save_path}")
|
||||
|
||||
except Exception as e:
|
||||
print(f"An error occurred during download: {e}")
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="M-Team Automation Tool")
|
||||
subparsers = parser.add_subparsers(dest="command", help="Command to execute")
|
||||
|
||||
# Login command
|
||||
login_parser = subparsers.add_parser("login", help="Perform login")
|
||||
|
||||
# Download command
|
||||
dl_parser = subparsers.add_parser("download", help="Download a torrent by TID")
|
||||
dl_parser.add_argument("tid", help="Torrent ID")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
# Default to login if no command provided (backward compatibility behavior)
|
||||
if not args.command:
|
||||
print("No command specified, defaulting to 'login'.")
|
||||
command = "login"
|
||||
else:
|
||||
command = args.command
|
||||
|
||||
config = load_config()
|
||||
|
||||
print(f"Launching browser with persistent profile at: {USER_DATA_DIR}")
|
||||
|
||||
with sync_playwright() as p:
|
||||
context = get_browser_context(p, headless=False)
|
||||
|
||||
# Persistent context might have an existing page or we create one
|
||||
if len(context.pages) > 0:
|
||||
page = context.pages[0]
|
||||
else:
|
||||
page = context.new_page()
|
||||
|
||||
if command == "login":
|
||||
login(page, config)
|
||||
elif command == "download":
|
||||
download(page, args.tid)
|
||||
|
||||
context.close()
|
||||
|
||||
if __name__ == "__main__":
|
||||
run()
|
||||
main()
|
||||
Reference in New Issue
Block a user