Upload 7 files

unzipper/database/__init__.py

@@ -0,0 +1,17 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from config import Config
+from motor.motor_asyncio import AsyncIOMotorClient
+
+mongodb = AsyncIOMotorClient(Config.MONGODB_URL)
+unzipper_db = mongodb["Unzipper_Bot"]

unzipper/database/cloud.py

@@ -0,0 +1,41 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from . import unzipper_db
+
+
+# Gofile.io
+class GofileDB:
+    def __init__(self, id: int) -> None:
+        self.id = id
+        self.db = unzipper_db["gofile_db"]
+
+    async def save_token(self, gtoken: str):
+        is_exist = await self.db.find_one({"_id": self.id})
+        if is_exist:
+            await self.db.update_one({"_id": self.id}, {"$set": {"token": gtoken}})
+        else:
+            await self.db.insert_one({"_id": self.id, "token": gtoken})
+
+    async def get_token(self):
+        gtkn = await self.db.find_one({"_id": self.id})
+        if gtkn:
+            return gtkn["token"]
+        else:
+            return None
+
+    async def del_token(self):
+        is_exist = await self.db.find_one({"_id": self.id})
+        if is_exist:
+            await self.db.delete_one({"_id": self.id})
+        else:
+            return

unzipper/database/language.py

@@ -0,0 +1,40 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from . import unzipper_db
+from unzipper.client.caching import USER_LANG
+
+
+lang_db = unzipper_db["languages_db"]
+
+
+async def set_language(user_id: int, lang: str):
+    exists = await lang_db.find_one({"_id": user_id})
+    if exists:
+        await lang_db.update_one({"_id": user_id}, {"$set": {"lang": lang}})
+    else:
+        await lang_db.insert_one({"_id": user_id, "lang": lang})
+
+
+async def get_language(user_id: int):
+    try:
+        return USER_LANG[user_id]
+    except:
+        exists = await lang_db.find_one({"_id": user_id})
+        if exists:
+            return exists["lang"]
+        else:
+            return "en"
+
+
+async def get_user_languages():
+    return lang_db.find({})

unzipper/database/split_arc.py

@@ -0,0 +1,39 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from . import unzipper_db
+
+spl_db = unzipper_db["splitted_archive_users"]
+
+
+async def add_split_arc_user(uid: int, fn: str, passw: str):
+    is_exist = await spl_db.find_one({"_id": uid})
+    if not is_exist:
+        await spl_db.insert_one({"_id": uid, "file_name": fn, "password": passw})
+    else:
+        raise ValueError("Data already exists!")
+
+
+async def get_split_arc_user(uid: int):
+    gsau = await spl_db.find_one({"_id": uid})
+    if gsau:
+        return True, gsau["file_name"], gsau["password"]
+    else:
+        return False, None, None
+
+
+async def del_split_arc_user(uid: int):
+    is_exist = await spl_db.find_one({"_id": uid})
+    if is_exist:
+        await spl_db.delete_one({"_id": uid})
+    else:
+        return

unzipper/database/thumbnail.py

@@ -0,0 +1,75 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from os import path
+from PIL import Image
+from . import unzipper_db
+from requests import post
+from pyrogram.types import Message
+from unzipper.lib.downloader import Downloader
+from unzipper.helpers_nexa.utils import run_cmds_on_cr
+
+
+thumb_db = unzipper_db["thumbnails_db"]
+
+
+def upload_thumbnail(img: str):
+    with open(img, "rb") as file:
+        rs = post(
+            "https://telegra.ph/upload",
+            files={"file": file}
+        ).json()[0]
+        return f"https://telegra.ph{rs['src']}"
+
+
+def prepare_thumb(ipath):
+    tpath = f"{path.splitext(ipath)[0]}.thumb.jpg"
+    with Image.open(ipath) as im:
+        rim = im.convert("RGB")
+        rim.thumbnail((320, 320))
+        rim.save(tpath, "JPEG")
+    return tpath
+
+
+async def save_thumbnail(uid: int, message: Message):
+    # Download the image
+    ip = await message.download()
+    thumb = await run_cmds_on_cr(prepare_thumb, ip)
+    up_thumb = await run_cmds_on_cr(upload_thumbnail, thumb)
+    is_exist = await thumb_db.find_one({"_id": uid})
+    if is_exist:
+        await thumb_db.update_one({"_id": uid}, {"$set": {"url": up_thumb}})
+    else:
+        await thumb_db.insert_one({"_id": uid, "url": up_thumb})
+
+
+async def get_thumbnail(user_id: int, download: bool = False):
+    gtm = await thumb_db.find_one({"_id": user_id})
+    if gtm:
+        if download:
+            dimg = f"Dump/thumbnail_{path.basename(gtm['url'])}"
+            await Downloader().download(gtm["url"], dimg, cont_type="image/")
+            return dimg
+        return gtm["url"]
+    else:
+        return None
+
+
+async def del_thumbnail(user_id: int):
+    is_exist = await thumb_db.find_one({"_id": user_id})
+    if is_exist:
+        await thumb_db.delete_one({"_id": user_id})
+    else:
+        return
+
+
+

unzipper/database/upload_mode.py

@@ -0,0 +1,31 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from . import unzipper_db
+
+mode_db = unzipper_db["upload_mode_db"]
+
+
+async def set_upload_mode(user_id, mode):
+    is_exist = await mode_db.find_one({"_id": user_id})
+    if is_exist:
+        await mode_db.update_one({"_id": user_id}, {"$set": {"mode": mode}})
+    else:
+        await mode_db.insert_one({"_id": user_id, "mode": mode})
+
+
+async def get_upload_mode(user_id):
+    umode = await mode_db.find_one({"_id": user_id})
+    if umode:
+        return umode["mode"]
+    else:
+        return "doc"

unzipper/database/users.py

@@ -0,0 +1,67 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from . import unzipper_db
+
+
+# Main users database
+user_db = unzipper_db["users_db"]
+
+
+async def add_user(user_id: int):
+    is_exist = await user_db.find_one({"user_id": user_id})
+    if not is_exist:
+        await user_db.insert_one({"user_id": user_id})
+
+
+async def del_user(user_id: int):
+    is_exist = await user_db.find_one({"user_id": user_id})
+    if is_exist:
+        await user_db.delete_one({"user_id": user_id})
+
+
+async def is_user_in_db(user_id: int):
+    is_exist = await user_db.find_one({"user_id": user_id})
+    return True if is_exist else False
+
+
+async def count_users():
+    return await user_db.count_documents({})
+
+
+async def get_users_list():
+    return (user["user_id"] async for user in user_db.find({}))
+
+
+# Banned users database (I don't know why tf i added this, but who cares)
+b_user_db = unzipper_db["banned_users_db"]
+
+
+async def add_banned_user(user_id: int):
+    is_exist = await b_user_db.find_one({"banned_user_id": user_id})
+    if not is_exist:
+        await b_user_db.insert_one({"banned_user_id": user_id})
+
+
+async def del_banned_user(user_id: int):
+    is_exist = await b_user_db.find_one({"banned_user_id": user_id})
+    if is_exist:
+        await b_user_db.delete_one({"banned_user_id": user_id})
+
+
+async def is_user_in_bdb(user_id: int):
+    is_exist = await b_user_db.find_one({"banned_user_id": user_id})
+    return True if is_exist else False
+
+
+async def count_banned_users():
+    return await b_user_db.count_documents({})