make queue.remove take Track instead of int

This commit is contained in:
cloudwithax 2022-11-13 22:01:15 -05:00
parent 8dfdf9f200
commit ec2f2d734f
1 changed files with 3 additions and 4 deletions

View File

@ -118,15 +118,14 @@ class Queue(Iterable[Track]):
def _index(self, item: Track) -> int: def _index(self, item: Track) -> int:
return self._queue.index(item) return self._queue.index(item)
def _put(self, item: Track) -> None: def _put(self, item: Track) -> None:
self._queue.append(item) self._queue.append(item)
def _insert(self, index: int, item: Track) -> None: def _insert(self, index: int, item: Track) -> None:
self._queue.insert(index, item) self._queue.insert(index, item)
def _remove(self, index: int) -> None: def _remove(self, item: Track) -> None:
self._queue.remove(index) self._queue.remove(item)
def _get_random_float(self) -> float: def _get_random_float(self) -> float:
return random.random() return random.random()
@ -233,7 +232,7 @@ class Queue(Iterable[Track]):
Removes a item within the queue. Removes a item within the queue.
Raises ValueError if item is not in queue. Raises ValueError if item is not in queue.
""" """
return self._remove(self._index(self._check_track(item))) return self._remove(self._check_track(item))
def find_position(self, item: Track) -> int: def find_position(self, item: Track) -> int: