youtube download

entity/__init__.py

@@ -1,4 +1,7 @@
-from .queue import Queue
+from .album import Album
+from .artist import Artist
 from .entry import Entry
 from .file import File
 from .playlist import Playlist
+from .queue import Queue
+from .title import Title

entity/album.py

@@ -0,0 +1,4 @@
+class Album:
+    def __init__(self, name: str, url: str) -> None:
+        self.name = name
+        self.url = url

entity/artist.py

@@ -0,0 +1,4 @@
+class Artist:
+    def __init__(self, name: str, url: str) -> None:
+        self.name = name
+        self.url = url

entity/entry.py

@@ -1,24 +1,27 @@
-from .playlist import Playlist
+from .album import Album
+from .artist import Artist
 from .file import File
+from .playlist import Playlist
+from .title import Title
 
 
 class Entry:
     def __init__(
         self,
-        title: str,
-        artist: str,
-        album: str,
+        title: Title,
+        artist: Artist,
+        duration: int,
         thumbnail: str,
-        link: str,
         requesterId: int,
+        album: Album | None = None,
         playlist: Playlist | None = None,
-        file: File | None = None,
+        source: File | str | None = None,
     ) -> None:
         self.title = title
         self.artist = artist
         self.album = album
+        self.duration = duration
         self.thumbnail = thumbnail
-        self.link = link
         self.requester = requesterId
         self.playlist = playlist
-        self.file = file
+        self.source = source

entity/file.py

@@ -1,5 +1,4 @@
 class File:
-    def __init__(self, name: str, path: str, size: int) -> None:
+    def __init__(self, name: str, size: int) -> None:
         self.name = name
-        self.path = path
         self.size = size

entity/playlist.py

@@ -1,4 +1,8 @@
+from .artist import Artist
+
+
 class Playlist:
-    def __init__(self, title: str, url: str) -> None:
-        self.title = title
+    def __init__(self, name: str, url: str, owner: Artist) -> None:
+        self.name = name
         self.url = url
+        self.owner = owner

entity/queue.py

@@ -1,5 +1,4 @@
 from .entry import Entry
-from .playlist import Playlist
 
 
 class Queue:
@@ -7,11 +6,8 @@ class Queue:
         self._entries: list[Entry] = []
         self.cursor = 0
 
-    def add(self, entry: Entry) -> None:
-        self._entries.append(entry)
-
-    def addPlalist(self, playlist: Playlist) -> None:
-        for entry in playlist:
+    def add(self, entries: list[Entry]) -> None:
+        for entry in entries:
             self._entries.append(entry)
 
     def remove(self, index: int, recursive: bool) -> None:

entity/title.py

@@ -0,0 +1,4 @@
+class Title:
+    def __init__(self, name: str, url: str) -> None:
+        self.name = name
+        self.url = url