matu
/
Heroku
mirror of https://github.com/coddrago/Heroku
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update heroku_info.py

pull/133/head
Who? 2025-04-20 19:16:55 +07:00 committed by GitHub
parent 18ca3a8944
commit 9a674284fa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
hikka/modules/heroku_info.py
View File

@ -182,7 +182,7 @@ class HerokuInfoMod(loader.Module):
imgform = self.config['bannerUrl'].split('.')[-1] imgform = self.config['bannerUrl'].split('.')[-1]
imgset = self.config['imgSettings'] imgset = self.config['imgSettings']
if imgform in ['jpg', 'jpeg', 'png', 'bmp', 'webp']: if imgform in ['jpg', 'jpeg', 'png', 'bmp', 'webp']:
response = requests.get(self.config['bannerUrl']) response = requests.get(self.config['bannerUrl'], stream=True)
img = Image.open(BytesIO(response.content)) img = Image.open(BytesIO(response.content))
font = ImageFont.truetype( font = ImageFont.truetype(
glob.glob(f'{os.getcwd()}/assets/font.*')[0], glob.glob(f'{os.getcwd()}/assets/font.*')[0],
@ -238,23 +238,7 @@ class HerokuInfoMod(loader.Module):
@loader.command() @loader.command()
async def infocmd(self, message: Message): async def infocmd(self, message: Message):
start = time.perf_counter_ns() start = time.perf_counter_ns()
if self.config['switchInfo']:
if self.config["custom_message"] is None:
await utils.answer_file(
message,
self.config["bannerUrl"],
self._render_info(start),
reply_to=getattr(message, "reply_to_msg_id", None),
)
elif '{ping}' in self.config["custom_message"]:
message = await utils.answer(message, self.config["ping_emoji"])
await utils.answer_file(
message,
self.config["bannerUrl"],
self._render_info(start),
reply_to=getattr(message, "reply_to_msg_id", None),
)
elif self.config['switchInfo']:
if self._get_info_photo(start) is None: if self._get_info_photo(start) is None:
await utils.answer( await utils.answer(
message, message,
@ -268,7 +252,18 @@ class HerokuInfoMod(loader.Module):
f'{utils.os.getcwd()}/assets/imginfo.{self.config["bannerUrl"].split(".")[-1]}', f'{utils.os.getcwd()}/assets/imginfo.{self.config["bannerUrl"].split(".")[-1]}',
reply_to=getattr(message, "reply_to_msg_id", None), reply_to=getattr(message, "reply_to_msg_id", None),
) )
elif self.config["custom_message"] is None:
if '{ping}' in self.config["custom_message"]:
message = await utils.answer(message, self.config["ping_emoji"])
await utils.answer_file(
message,
self.config["bannerUrl"],
self._render_info(start),
reply_to=getattr(message, "reply_to_msg_id", None),
)
else: else:
if '{ping}' in self.config["custom_message"]:
message = await utils.answer(message, self.config["ping_emoji"])
await utils.answer_file( await utils.answer_file(
message, message,
self.config["bannerUrl"], self.config["bannerUrl"],