1
0
mirror of https://github.com/azlux/botamusique synced 2024-11-23 22:06:09 +00:00

Compare commits

..

No commits in common. "3200c83fd0e9415c3aa4c7cd1d7cd5036f6ca2bd" and "33a9e75ba9d0a382f7a76d23a0ceb626924a8b49" have entirely different histories.

View File

@ -46,7 +46,6 @@ class BasePlaylist(list):
self.pending_items = [] self.pending_items = []
self.log = logging.getLogger("bot") self.log = logging.getLogger("bot")
self.validating_thread_lock = threading.Lock() self.validating_thread_lock = threading.Lock()
self.playlist_lock = threading.RLock()
def is_empty(self): def is_empty(self):
return True if len(self) == 0 else False return True if len(self) == 0 else False
@ -60,82 +59,69 @@ class BasePlaylist(list):
return self return self
def append(self, item: CachedItemWrapper): def append(self, item: CachedItemWrapper):
with self.playlist_lock: self.version += 1
self.version += 1 super().append(item)
super().append(item) self.pending_items.append(item)
self.pending_items.append(item)
self.async_validate() self.async_validate()
return item return item
def insert(self, index, item): def insert(self, index, item):
with self.playlist_lock: self.version += 1
self.version += 1
if index == -1:
index = self.current_index
super().insert(index, item) if index == -1:
index = self.current_index
if index <= self.current_index: super().insert(index, item)
self.current_index += 1
self.pending_items.append(item) if index <= self.current_index:
self.current_index += 1
self.pending_items.append(item)
self.async_validate() self.async_validate()
return item return item
def extend(self, items): def extend(self, items):
print("extend to-enter") self.version += 1
with self.playlist_lock: super().extend(items)
print("extend enter") self.pending_items.extend(items)
self.version += 1
super().extend(items)
self.pending_items.extend(items)
print("extend leave")
self.async_validate() self.async_validate()
return items return items
def next(self): def next(self):
with self.playlist_lock: if len(self) == 0:
if len(self) == 0: return False
return False
if self.current_index < len(self) - 1: if self.current_index < len(self) - 1:
self.current_index += 1 self.current_index += 1
return self[self.current_index] return self[self.current_index]
else: else:
return False return False
def point_to(self, index): def point_to(self, index):
with self.playlist_lock: if -1 <= index < len(self):
if -1 <= index < len(self): self.current_index = index
self.current_index = index
def find(self, id): def find(self, id):
with self.playlist_lock: for index, wrapper in enumerate(self):
for index, wrapper in enumerate(self): if wrapper.item.id == id:
if wrapper.item.id == id: return index
return index
return None return None
def __delitem__(self, key): def __delitem__(self, key):
return self.remove(key) return self.remove(key)
def remove(self, index): def remove(self, index):
with self.playlist_lock: self.version += 1
self.version += 1 if index > len(self) - 1:
if index > len(self) - 1: return False
return False
removed = self[index] removed = self[index]
super().__delitem__(index) super().__delitem__(index)
if self.current_index > index: if self.current_index > index:
self.current_index -= 1 self.current_index -= 1
# reference counter # reference counter
counter = 0 counter = 0
@ -160,54 +146,47 @@ class BasePlaylist(list):
self.remove(index) self.remove(index)
def current_item(self): def current_item(self):
with self.playlist_lock: if len(self) == 0:
if len(self) == 0: return False
return False
return self[self.current_index] return self[self.current_index]
def next_index(self): def next_index(self):
with self.playlist_lock: if self.current_index < len(self) - 1:
if self.current_index < len(self) - 1: return self.current_index + 1
return self.current_index + 1 else:
return False
return False
def next_item(self): def next_item(self):
with self.playlist_lock: if self.current_index < len(self) - 1:
if self.current_index < len(self) - 1: return self[self.current_index + 1]
return self[self.current_index + 1] else:
return False
return False
def randomize(self): def randomize(self):
with self.playlist_lock: # current_index will lose track after shuffling, thus we take current music out before shuffling
# current_index will lose track after shuffling, thus we take current music out before shuffling # current = self.current_item()
# current = self.current_item() # del self[self.current_index]
# del self[self.current_index]
random.shuffle(self) random.shuffle(self)
# self.insert(0, current) # self.insert(0, current)
self.current_index = -1 self.current_index = -1
self.version += 1 self.version += 1
def clear(self): def clear(self):
with self.playlist_lock: self.version += 1
self.version += 1 self.current_index = -1
self.current_index = -1
super().clear()
var.cache.free_all() var.cache.free_all()
super().clear()
def save(self): def save(self):
with self.playlist_lock: var.db.remove_section("playlist_item")
var.db.remove_section("playlist_item") assert self.current_index is not None
assert self.current_index is not None var.db.set("playlist", "current_index", self.current_index)
var.db.set("playlist", "current_index", self.current_index)
for index, music in enumerate(self): for index, music in enumerate(self):
var.db.set("playlist_item", str(index), json.dumps({'id': music.id, 'user': music.user})) var.db.set("playlist_item", str(index), json.dumps({'id': music.id, 'user': music.user}))
def load(self): def load(self):
current_index = var.db.getint("playlist", "current_index", fallback=-1) current_index = var.db.getint("playlist", "current_index", fallback=-1)
@ -264,8 +243,8 @@ class BasePlaylist(list):
self.log.debug("playlist: validating failed.") self.log.debug("playlist: validating failed.")
if var.bot: if var.bot:
var.bot.send_channel_msg(e.msg) var.bot.send_channel_msg(e.msg)
self.remove_by_id(item.id)
var.cache.free_and_delete(item.id) var.cache.free_and_delete(item.id)
self.remove_by_id(item.id)
continue continue
if item.version > ver: if item.version > ver:
@ -282,42 +261,35 @@ class OneshotPlaylist(BasePlaylist):
self.current_index = -1 self.current_index = -1
def current_item(self): def current_item(self):
with self.playlist_lock: if self.current_index == -1:
if len(self) == 0: self.current_index = 0
self.current_index = -1
return False
if self.current_index == -1: return self[self.current_index]
self.current_index = 0
return self[self.current_index]
def from_list(self, _list, current_index): def from_list(self, _list, current_index):
with self.playlist_lock: if len(_list) > 0:
if len(_list) > 0: if current_index > -1:
if current_index > -1: for i in range(current_index):
for i in range(current_index): _list.pop(0)
_list.pop(0) return super().from_list(_list, 0)
return super().from_list(_list, 0) return super().from_list(_list, -1)
return super().from_list(_list, -1) return self
return self
def next(self): def next(self):
with self.playlist_lock: if len(self) > 0:
if len(self) > 0: self.version += 1
self.version += 1
if self.current_index != -1: if self.current_index != -1:
super().__delitem__(self.current_index) super().__delitem__(self.current_index)
if len(self) == 0: if len(self) == 0:
return False return False
else:
self.current_index = 0
return self[0]
else: else:
self.current_index = -1 self.current_index = 0
return False
return self[0]
else:
self.current_index = -1
return False
def next_index(self): def next_index(self):
if len(self) > 1: if len(self) > 1:
@ -332,11 +304,10 @@ class OneshotPlaylist(BasePlaylist):
return False return False
def point_to(self, index): def point_to(self, index):
with self.playlist_lock: self.version += 1
self.version += 1 self.current_index = -1
self.current_index = -1 for i in range(index):
for i in range(index): super().__delitem__(0)
super().__delitem__(0)
class RepeatPlaylist(BasePlaylist): class RepeatPlaylist(BasePlaylist):
@ -345,27 +316,23 @@ class RepeatPlaylist(BasePlaylist):
self.mode = "repeat" self.mode = "repeat"
def next(self): def next(self):
with self.playlist_lock:
if len(self) == 0:
return False
if self.current_index < len(self) - 1:
self.current_index += 1
return self[self.current_index]
else:
self.current_index = 0
return self[0]
def next_index(self):
with self.playlist_lock:
if self.current_index < len(self) - 1:
return self.current_index + 1
else:
return 0
def next_item(self):
if len(self) == 0: if len(self) == 0:
return False return False
if self.current_index < len(self) - 1:
self.current_index += 1
return self[self.current_index]
else:
self.current_index = 0
return self[0]
def next_index(self):
if self.current_index < len(self) - 1:
return self.current_index + 1
else:
return 0
def next_item(self):
return self[self.next_index()] return self[self.next_index()]
@ -380,18 +347,17 @@ class RandomPlaylist(BasePlaylist):
return super().from_list(_list, -1) return super().from_list(_list, -1)
def next(self): def next(self):
with self.playlist_lock: if len(self) == 0:
if len(self) == 0: return False
return False
if self.current_index < len(self) - 1: if self.current_index < len(self) - 1:
self.current_index += 1 self.current_index += 1
return self[self.current_index] return self[self.current_index]
else: else:
self.version += 1 self.version += 1
self.randomize() self.randomize()
self.current_index = 0 self.current_index = 0
return self[0] return self[0]
class AutoPlaylist(OneshotPlaylist): class AutoPlaylist(OneshotPlaylist):