Commited backup

D4n13l3k00/FTG-Modules/S3.py

@@ -0,0 +1,72 @@
+# .------.------.------.------.------.------.------.------.------.------.
+# |D.--. |4.--. |N.--. |1.--. |3.--. |L.--. |3.--. |K.--. |0.--. |0.--. |
+# | :/\: | :/\: | :(): | :/\: | :(): | :/\: | :(): | :/\: | :/\: | :/\: |
+# | (__) | :\/: | ()() | (__) | ()() | (__) | ()() | :\/: | :\/: | :\/: |
+# | '--'D| '--'4| '--'N| '--'1| '--'3| '--'L| '--'3| '--'K| '--'0| '--'0|
+# `------`------`------`------`------`------`------`------`------`------'
+#
+#                     Copyright 2023 t.me/D4n13l3k00
+#           Licensed under the Creative Commons CC BY-NC-ND 4.0
+#
+#                    Full license text can be found at:
+#       https://creativecommons.org/licenses/by-nc-nd/4.0/legalcode
+#
+#                           Human-friendly one:
+#            https://creativecommons.org/licenses/by-nc-nd/4.0
+
+# meta developer: @D4n13l3k00
+
+
+from telethon import types
+import rsa
+import aioboto3
+from .. import loader, utils  # type: ignore
+
+
+@loader.tds
+class S3Mod(loader.Module):
+    """S3 file manager"""
+
+    strings = {"name": "S3 file manager", "pref": "<b>[S3]</b> {}"}
+
+    async def client_ready(self, client, db):
+        self._db = db
+        self._db_name = "S3"
+
+    async def set_auth_data(
+        self, endpoint: str, region: str, username: str, password: str, bucket: str
+    ) -> None:
+        self._db.set(self._db_name, "endpoint", endpoint)
+        self._db.set(self._db_name, "region", region)
+        self._db.set(self._db_name, "username", username)
+        self._db.set(self._db_name, "password", password)
+        self._db.set(self._db_name, "bucket", bucket)
+
+    async def get_client(self) -> aioboto3.Session.client:
+        session = aioboto3.Session(
+            aws_access_key_id=self._db.get(self._db_name, "username"),
+            aws_secret_access_key=self._db.get(self._db_name, "password"),
+            region_name=self._db.get(self._db_name, "region"),
+        )
+        return session.client(
+            service_name="s3",
+            endpoint_url=f"https://{self._db.get(self._db_name, 'endpoint')}",
+        )
+
+    @loader.owner
+    async def s3upcmd(self, m: types.Message):
+        reply = await m.get_reply_message()
+        if not reply or not reply.document:
+            return await utils.answer(m, self.strings("s3").format("Reply to a document"))
+        session = await self.generate_session()
+        client = await self.get_client()
+        async with client as s3:
+            await s3.upload_fileobj(
+                await bot.download_file_by_id(video.file_id),
+                os.getenv("S3_BUCKET_NAME"),
+                video_s3_path,
+            )
+        
+    # @loader.owner
+    # async def s3authcmd(self, m: types.Message):
+    #     pass