Merge pull request #66 from Rilliat/v1.6.9

fix: ping in info
pull/68/head
Who? 2025-01-16 01:37:11 +10:00 committed by GitHub
commit 93691f0c1f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 21 deletions

View File

@ -10,8 +10,6 @@ import time
import psutil import psutil
from hikkatl.tl.types import Message from hikkatl.tl.types import Message
from hikkatl.utils import get_display_name from hikkatl.utils import get_display_name
import requests
import os
from .. import loader, utils, version from .. import loader, utils, version
from ..inline.types import InlineQuery from ..inline.types import InlineQuery
import subprocess import subprocess
@ -49,7 +47,7 @@ class HerokuInfoMod(loader.Module):
), ),
) )
def _render_info(self, inline: bool) -> str: def _render_info(self, inline: bool, start: float) -> str:
try: try:
repo = git.Repo(search_parent_directories=True) repo = git.Repo(search_parent_directories=True)
diff = repo.git.log([f"HEAD..origin/{version.branch}", "--oneline"]) diff = repo.git.log([f"HEAD..origin/{version.branch}", "--oneline"])
@ -89,6 +87,10 @@ class HerokuInfoMod(loader.Module):
("🐧", "<emoji document_id=5361541227604878624>🐧</emoji>") ("🐧", "<emoji document_id=5361541227604878624>🐧</emoji>")
]: ]:
platform = platform.replace(emoji, icon) platform = platform.replace(emoji, icon)
try: os = lib_platform.freedesktop_os_release()["PRETTY_NAME"]
except Exception: os = self.strings('non_detectable')
return ( return (
( (
"<b>🪐 Heroku</b>\n" "<b>🪐 Heroku</b>\n"
@ -108,9 +110,10 @@ class HerokuInfoMod(loader.Module):
branch=version.branch, branch=version.branch,
hostname=lib_platform.node(), hostname=lib_platform.node(),
user=subprocess.run(['whoami'], stdout=subprocess.PIPE).stdout.decode().strip(), user=subprocess.run(['whoami'], stdout=subprocess.PIPE).stdout.decode().strip(),
os=lib_platform.freedesktop_os_release()["PRETTY_NAME"] or self.strings('non_detectable'), os=os,
kernel=lib_platform.release(), kernel=lib_platform.release(),
cpu=f"{psutil.cpu_count(logical=False)} ({psutil.cpu_count()}) core(-s); {psutil.cpu_percent()}%", cpu=f"{psutil.cpu_count(logical=False)} ({psutil.cpu_count()}) core(-s); {psutil.cpu_percent()}%",
ping=round((time.perf_counter_ns() - start) / 10**6, 3)
) )
if self.config["custom_message"] if self.config["custom_message"]
else ( else (
@ -147,22 +150,6 @@ class HerokuInfoMod(loader.Module):
) )
) )
async def info(self, _: InlineQuery) -> dict:
"""Send userbot info"""
return {
"title": self.strings("send_info"),
"description": self.strings("description"),
**(
{"photo": self.config["banner_url"], "caption": self._render_info(True)}
if self.config["banner_url"]
else {"message": self._render_info(True)}
),
"thumb": (
"https://github.com/hikariatama/Hikka/raw/master/assets/hikka_pfp.png"
),
"reply_markup": self._get_mark(),
}
@loader.command() @loader.command()
async def infocmd(self, message: Message): async def infocmd(self, message: Message):
@ -181,7 +168,7 @@ class HerokuInfoMod(loader.Module):
await utils.answer_file( await utils.answer_file(
message, message,
self.config["banner_url"], self.config["banner_url"],
self._render_info(False).format(ping=round((time.perf_counter_ns() - start) / 10**6, 3)), self._render_info(False, start),
) )
@loader.command() @loader.command()