Spaces:
Build error
Build error
File size: 3,581 Bytes
0cee3b2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
# ===================================================================== #
# 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/> #
# ===================================================================== #
import logging
from os import path, remove
from .pyro_client import UnzipperBot
from unzipper.database.thumbnail import get_thumbnail
from unzipper.helpers_nexa.utils import run_shell_cmds
from unzipper.database.users import add_user, is_user_in_db, is_user_in_bdb
from config import Config
class PatchMethods:
def __init__(self) -> None:
super().__init__()
async def check_user(self: UnzipperBot, message):
"""
Checks database checks of new users
"""
# Checking if user is banned
is_banned = await is_user_in_bdb(int(message.from_user.id))
if is_banned:
await message.reply("**Sorry You're Banned!** \n\nReport this at @Nexa_bots if you think this is a mistake")
raise UserIsBanned
# Chacking if user already in db
is_in_db = await is_user_in_db(int(message.from_user.id))
if not is_in_db:
try:
await add_user(int(message.from_user.id))
await self.send_message(
chat_id=Config.LOGS_CHANNEL,
text=f"**#NEW_USER** 🎙 \n\n**User Profile:** `{message.from_user.mention}` \n**User ID:** `{message.from_user.id}` \n**Profile Url:** [Click here](tg://user?id={message.from_user.id})",
disable_web_page_preview=True
)
except Exception as e:
logging.warn(
f"Unable to add user to the database due to: \n{e}")
async def get_or_gen_thumb(self: UnzipperBot, uid: int, doc_f: str, isvid: bool = False):
"""
Get saved thumbnail from the database. If there isn't any thumbnail saved, None will be returned.
For video files, a thumbnail will be generated using ffmpeg
Parameters:
- `uid` - User id
- `doc_f` - File path
- `isvid` - Pass True if file is a video
"""
dbthumb = await get_thumbnail(int(uid), True)
if dbthumb:
return dbthumb
elif isvid:
thmb_pth = f"Dump/thumbnail_{path.basename(doc_f)}.jpg"
if path.exists(thmb_pth):
remove(thmb_pth)
await run_shell_cmds(f"ffmpeg -ss 00:00:01.00 -i {doc_f} -vf 'scale=320:320:force_original_aspect_ratio=decrease' -vframes 1 {thmb_pth}")
return thmb_pth
else:
return None
class UserIsBanned(Exception):
def __init__(self) -> None:
super().__init__("You're banned from using this bot!")
def init_patch():
"""
Apply custom methods defined in CustomMethods class to pyrogram.Client class
"""
for ckey, cval in PatchMethods.__dict__.items():
if ckey[:2] != "__":
setattr(UnzipperBot, ckey, cval)
|