mirrors/streamrip
1
0
Fork 0
mirror of https://github.com/nathom/streamrip.git synced 2025-05-15 15:44:58 -04:00
Code Issues Projects Releases Packages Wiki Activity

Fix compatibility with new databases

Browse source 
Signed-off-by: nathom <nathanthomas707@gmail.com>
This commit is contained in:
nathom 2021-07-03 13:48:29 -07:00
parent 2f3f425687
commit 97318017e0
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
rip/core.py
View file

@ -43,7 +43,7 @@ from streamrip.constants import (
URL_REGEX,
YOUTUBE_URL_REGEX,
)
from .db import MusicDB
from . import db
from streamrip.exceptions import (
AuthenticationError,
MissingCredentials,
@ -106,18 +106,18 @@ class MusicDL(list):
"soundcloud": SoundCloudClient(),
}
self.db: MusicDB
self.db: db.Database
db_settings = self.config.session["database"]
if db_settings["enabled"]:
path = db_settings["path"]
if path:
self.db = MusicDB(path)
self.db = db.Downloads(path)
else:
self.db = MusicDB(DB_PATH)
self.db = db.Downloads(DB_PATH)
self.config.file["database"]["path"] = DB_PATH
self.config.save()
else:
self.db = MusicDB(None, empty=True)
self.db = db.Downloads(None, empty=True)
def handle_urls(self, urls):
"""Download a url.
@ -153,7 +153,7 @@ class MusicDL(list):
raise ParsingError(message)
for source, url_type, item_id in parsed:
if item_id in self.db:
if {"id": item_id} in self.db:
logger.info(
f"ID {item_id} already downloaded, use --no-db to override."
)
@ -262,7 +262,7 @@ class MusicDL(list):
continue
if item.download(**arguments) and hasattr(item, "id"):
self.db.add(item.id)
self.db.add([item.id])
if isinstance(item, Track):
item.tag()