59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
import asyncio
|
|
import sys
|
|
import time
|
|
from pathlib import Path
|
|
from bilibili_api.login_v2 import QrCodeLogin, QrCodeLoginEvents
|
|
|
|
QR_PATH = Path("/home/bilibili/qrcode.png")
|
|
COOKIE_FILE = Path("/home/bilibili/cookies.txt")
|
|
CF_QR_PATH = "/mnt/E/temp/bilibili_qrcode.png"
|
|
|
|
async def main():
|
|
login = QrCodeLogin()
|
|
await login.generate_qrcode()
|
|
pic = login.get_qrcode_picture()
|
|
QR_PATH.write_bytes(pic.content)
|
|
print(f"QR saved to {QR_PATH}")
|
|
print("Please scan the QR code with Bilibili App")
|
|
|
|
# copy to cf
|
|
import subprocess
|
|
try:
|
|
subprocess.run([
|
|
"sshpass", "-p", "ftA406u@all#2309",
|
|
"scp", "-P", "10526", str(QR_PATH), "root@rrrosmontis.icu:/mnt/E/temp/bilibili_qrcode.png"
|
|
], check=True, timeout=30)
|
|
print(f"QR copied to cf:{CF_QR_PATH}")
|
|
except Exception as e:
|
|
print(f"Failed to copy QR to cf: {e}")
|
|
|
|
# wait up to 5 minutes
|
|
for i in range(60):
|
|
await asyncio.sleep(5)
|
|
state = await login.check_state()
|
|
print(f"[{i*5}s] state: {state}")
|
|
if state == QrCodeLoginEvents.DONE:
|
|
cred = login.get_credential()
|
|
lines = []
|
|
if cred.sessdata:
|
|
lines.append(f"SESSDATA={cred.sessdata}")
|
|
if cred.bili_jct:
|
|
lines.append(f"bili_jct={cred.bili_jct}")
|
|
if cred.buvid3:
|
|
lines.append(f"buvid3={cred.buvid3}")
|
|
if cred.dedeuserid:
|
|
lines.append(f"dedeuserid={cred.dedeuserid}")
|
|
if cred.ac_time_value:
|
|
lines.append(f"ac_time_value={cred.ac_time_value}")
|
|
COOKIE_FILE.write_text("\n".join(lines) + "\n")
|
|
print(f"Login success. Cookie saved to {COOKIE_FILE}")
|
|
print(f"ac_time_value present: {bool(cred.ac_time_value)}")
|
|
return
|
|
elif state == QrCodeLoginEvents.TIMEOUT:
|
|
print("QR code timeout")
|
|
sys.exit(1)
|
|
print("QR login timeout after 5 minutes")
|
|
sys.exit(1)
|
|
|
|
asyncio.run(main())
|