Update app.py
Browse files
app.py
CHANGED
@@ -41,6 +41,7 @@ ARL_TOKEN = os.getenv('ARL')
|
|
41 |
class DownloadRequest(BaseModel):
|
42 |
url: str
|
43 |
quality: str
|
|
|
44 |
|
45 |
@app.get("/")
|
46 |
def read_root():
|
@@ -72,9 +73,13 @@ def get_track(track_id: str):
|
|
72 |
@app.post("/download/track")
|
73 |
def download_track(request: DownloadRequest):
|
74 |
try:
|
75 |
-
|
|
|
|
|
|
|
76 |
url = request.url
|
77 |
quality = request.quality
|
|
|
78 |
|
79 |
if quality not in ["MP3_320", "MP3_128", "FLAC"]:
|
80 |
raise HTTPException(status_code=400, detail="Invalid quality specified")
|
|
|
41 |
class DownloadRequest(BaseModel):
|
42 |
url: str
|
43 |
quality: str
|
44 |
+
arl: str
|
45 |
|
46 |
@app.get("/")
|
47 |
def read_root():
|
|
|
73 |
@app.post("/download/track")
|
74 |
def download_track(request: DownloadRequest):
|
75 |
try:
|
76 |
+
if request.arl is None or request.arl.strip() == "":
|
77 |
+
ARL = ARL_TOKEN
|
78 |
+
else:
|
79 |
+
ARL = request.arl
|
80 |
url = request.url
|
81 |
quality = request.quality
|
82 |
+
dl = DeeLogin(arl=ARL)
|
83 |
|
84 |
if quality not in ["MP3_320", "MP3_128", "FLAC"]:
|
85 |
raise HTTPException(status_code=400, detail="Invalid quality specified")
|