Change to .pkl

SantiiRepair · Aug 7, 2023 · fbfbac1 · fbfbac1
1 parent c57e84c
commit fbfbac1
Show file tree

Hide file tree

Showing 8 changed files with 39 additions and 53 deletions.
diff --git a/.codesandbox/Dockerfile b/.codesandbox/Dockerfile
diff --git a/.codesandbox/tasks.json b/.codesandbox/tasks.json
diff --git a/.github/images/gitpod.png b/.github/images/gitpod.png
diff --git a/bot/dbs/actives.pkl b/bot/dbs/actives.pkl
@@ -1 +0,0 @@
-[]

diff --git a/bot/dbs/vcs.pkl b/bot/dbs/vcs.pkl
@@ -1 +0,0 @@
-[]

diff --git a/bot/helpers/queues.py b/bot/helpers/queues.py
@@ -1,33 +1,41 @@
 import os
 import pickle
 
-active = []
 dir = os.path.dirname(os.path.abspath(__file__))
 queues = os.path.join(dir, "../dbs/queues.pkl")
 actives = os.path.join(dir, "../dbs/actives.pkl")
 
 
 async def get_active_chats() -> list:
-    return active
+    with open(actives, "r") as a:
+        ACTIVE = pickle.load(a)
+    return ACTIVE
 
 
 def add_to_queue(chat, name, url, ref, type):
-    with open(queues, "r+") as f:
-        QUEUE = pickle.load(f)
+    with open(queues, "r") as q:
+        QUEUE = pickle.load(q)
+    with open(actives, "r") as a:
+        ACTIVE = pickle.load(a)
     try:
         if chat.id in QUEUE:
             QUEUE[chat.id].append([name, url, ref, type])
-            pickle.dump(QUEUE[chat.id], f)
+            with open(queues, "w") as q:
+                pickle.dump(QUEUE, q)
             return int(len(QUEUE[chat.id]) - 1)
         if chat.id not in active:
             active.append(chat.id)
+            with open(actives, "w") as a:
+                pickle.dump(ACTIVE, a)
         QUEUE[chat.id] = [[name, url, ref, type]]
+        with open(queues, "w") as q:
+            pickle.dump(QUEUE, q)
     except Exception as e:
         raise e
 
 
 def get_queue(chat):
-    with open(queues, "r") as f:
+    with open(queues, "r") as q:
         QUEUE = pickle.load(f)
     try:
         if chat.id in QUEUE:
@@ -38,28 +46,34 @@ def get_queue(chat):
 
 
 def pop_an_item(chat):
-    with open(queues, "r+") as f:
-        QUEUE = pickle.load(f)
+    with open(queues, "r") as q:
+        QUEUE = pickle.load(q)
     try:
         if chat.id not in QUEUE:
             return 0
         QUEUE[chat.id].pop(0)
-        pickle.dump(QUEUE)
+        with open(queues, "w") as q:
+            pickle.dump(QUEUE, q)
         return 1
     except Exception as e:
         raise e
 
 
 def clear_queue(chat):
-    with open(queues, "r+") as f:
-        QUEUE = pickle.load(f)
+    with open(queues, "r") as q:
+        QUEUE = pickle.load(q)
+    with open(actives, "r") as a:
+        ACTIVE = pickle.load(a)
     try:
         if chat.id not in QUEUE:
             return 0
         QUEUE.pop(chat.id)
-        pickle.dump(QUEUE)
-        if chat.id in active:
-            active.remove(chat.id)
+        with open(queues, "w") as q:
+            pickle.dump(QUEUE, q)
+        if chat.id in ACTIVE:
+            ACTIVE.remove(chat.id)
+            with open(actives, "r") as a:
+                pickle.dump(ACTIVE, a)
         return 1
     except Exception as e:
         raise e
diff --git a/bot/dicts/dicts.py → bot/instance/of_every_vc.py b/bot/dicts/dicts.py → bot/instance/of_every_vc.py
diff --git a/bot/plugins/callbacks.py b/bot/plugins/callbacks.py
@@ -1,20 +1,21 @@
+import pickle
 from asyncio import sleep
 from bot import config, kreacher
-from bot.dicts.dicts import QUEUE, VOICE_CHATS
+from bot.dicts.dicts import VOICE_CHATS
 from telethon import events, Button
 from bot.helpers.queues_handler import next_item, skip_current
 
 thumb = "https://telegra.ph/file/3e14128ad5c9ec47801bd.jpg"
 
+dir = os.path.dirname(os.path.abspath(__file__))
+queues = os.path.join(dir, "../dbs/queues.pkl")
 
 @kreacher.on(events.callbackquery.CallbackQuery(data="cls"))
 async def _(event):
     await event.delete()
 
 
-@kreacher.on(
-    events.callbackquery.CallbackQuery(data="pause_or_resume_callback")
-)
+@kreacher.on(events.callbackquery.CallbackQuery(data="pause_or_resume_callback"))
 async def _(event):
     chat = await event.get_chat()
     if VOICE_CHATS[chat.id].is_video_paused:
@@ -58,6 +59,8 @@ async def _(event):
 
 @kreacher.on(events.callbackquery.CallbackQuery(data="next_callback"))
 async def _(event):
+    with open(queues, "r") as q:
+        QUEUE = pickle.load(q)
     chat = await event.get_chat()
     if len(event.text.split()) < 2:
         op = await skip_current(chat)
@@ -88,8 +91,12 @@ async def _(event):
 
 @kreacher.on(events.callbackquery.CallbackQuery(data="end_callback"))
 async def _(event):
+    with open(queues, "r") as q:
+        QUEUE = pickle.load(q)
     chat = await event.get_chat()
     QUEUE.pop(chat.id)
+    with open(queues, "w") as q:
+        pickle.dump(QUEUE,q)
     await VOICE_CHATS[chat.id].stop_media()
     await VOICE_CHATS[chat.id].stop()
     VOICE_CHATS.pop(chat.id)