Refactor Myjdapi usage in app.py
This commit is contained in:
@@ -12,7 +12,7 @@ import time
|
|||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from typing import Any, Dict, List, Optional, Tuple
|
from typing import Any, Dict, List, Optional, Tuple
|
||||||
|
|
||||||
import myjdapi
|
from myjdapi import Myjdapi
|
||||||
import paramiko
|
import paramiko
|
||||||
from fastapi import FastAPI, Form, Request
|
from fastapi import FastAPI, Form, Request
|
||||||
from fastapi.responses import HTMLResponse, RedirectResponse
|
from fastapi.responses import HTMLResponse, RedirectResponse
|
||||||
@@ -133,10 +133,10 @@ def ensure_env():
|
|||||||
raise RuntimeError("Missing env vars: " + ", ".join(missing))
|
raise RuntimeError("Missing env vars: " + ", ".join(missing))
|
||||||
|
|
||||||
def get_device():
|
def get_device():
|
||||||
jd = myjdapi.myjdapi()
|
jd = Myjdapi()
|
||||||
jd.connect(MYJD_EMAIL, MYJD_PASSWORD)
|
jd.connect(MYJD_EMAIL, MYJD_PASSWORD)
|
||||||
jd.getDevices()
|
jd.update_devices()
|
||||||
dev = jd.getDevice(name=MYJD_DEVICE)
|
dev = jd.get_device(MYJD_DEVICE)
|
||||||
if dev is None:
|
if dev is None:
|
||||||
raise RuntimeError(f"MyJDownloader device not found: {MYJD_DEVICE}")
|
raise RuntimeError(f"MyJDownloader device not found: {MYJD_DEVICE}")
|
||||||
return dev
|
return dev
|
||||||
|
|||||||
Reference in New Issue
Block a user