mirror of
https://github.com/yasirarism/MissKatyPyro.git
synced 2025-12-29 17:44:50 +00:00
a7008c664e
* ci: Update .deepsource.toml
* ci: Update .deepsource.toml
* style: format code with black and isort
Format code with black and isort
This commit fixes the style issues introduced in 0fb651a according to the output
from Black and isort.
Details: https://app.deepsource.com/gh/yasirarism/MissKatyPyro/transform/d8f2f66e-b496-4686-aca6-9830236eda12/
---------
Co-authored-by: deepsource-io[bot] <42547082+deepsource-io[bot]@users.noreply.github.com>
Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
153 lines
4 KiB
Python
153 lines
4 KiB
Python
import logging
|
|
import os
|
|
import random
|
|
import string
|
|
import time
|
|
from http.cookies import SimpleCookie
|
|
from re import match as re_match
|
|
from urllib.parse import urlparse
|
|
|
|
import psutil
|
|
|
|
from misskaty import BOT_NAME, UBOT_NAME, botStartTime
|
|
from misskaty.helper.http import http
|
|
from misskaty.helper.human_read import get_readable_time
|
|
from misskaty.plugins import ALL_MODULES
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
URL_REGEX = r"(http|ftp|https):\/\/([\w_-]+(?:(?:\.[\w_-]+)+))([\w.,@?^=%&:\/~+#-]*[\w@?^=%&\/~+#-])"
|
|
GENRES_EMOJI = {
|
|
"Action": "👊",
|
|
"Adventure": random.choice(["🪂", "🧗♀", "🌋"]),
|
|
"Family": "👨",
|
|
"Musical": "🎸",
|
|
"Comedy": "🤣",
|
|
"Drama": " 🎭",
|
|
"Ecchi": random.choice(["💋", "🥵"]),
|
|
"Fantasy": random.choice(["🧞", "🧞♂", "🧞♀", "🌗"]),
|
|
"Hentai": "🔞",
|
|
"History": "📜",
|
|
"Horror": "☠",
|
|
"Mahou Shoujo": "☯",
|
|
"Mecha": "🤖",
|
|
"Music": "🎸",
|
|
"Mystery": "🔮",
|
|
"Psychological": "♟",
|
|
"Romance": "💞",
|
|
"Sci-Fi": "🛸",
|
|
"Slice of Life": random.choice(["☘", "🍁"]),
|
|
"Sports": "⚽️",
|
|
"Supernatural": "🫧",
|
|
"Thriller": random.choice(["🥶", "🔪", "🤯"]),
|
|
}
|
|
|
|
|
|
def is_url(url):
|
|
url = re_match(URL_REGEX, url)
|
|
return bool(url)
|
|
|
|
|
|
async def bot_sys_stats():
|
|
bot_uptime = int(time.time() - botStartTime)
|
|
cpu = psutil.cpu_percent(interval=0.5)
|
|
mem = psutil.virtual_memory().percent
|
|
disk = psutil.disk_usage("/").percent
|
|
process = psutil.Process(os.getpid())
|
|
return f"""
|
|
{UBOT_NAME}@{BOT_NAME}
|
|
------------------
|
|
UPTIME: {get_readable_time(bot_uptime)}
|
|
BOT: {round(process.memory_info()[0] / 1024**2)} MB
|
|
CPU: {cpu}%
|
|
RAM: {mem}%
|
|
DISK: {disk}%
|
|
|
|
TOTAL PLUGINS: {len(ALL_MODULES)}
|
|
"""
|
|
|
|
|
|
def remove_N(seq):
|
|
i = 1
|
|
while i < len(seq):
|
|
if seq[i] == seq[i - 1]:
|
|
del seq[i]
|
|
i -= 1
|
|
else:
|
|
i += 1
|
|
|
|
|
|
def get_random_string(length: int = 5):
|
|
text_str = "".join(
|
|
random.SystemRandom().choice(string.ascii_letters + string.digits)
|
|
for _ in range(length)
|
|
)
|
|
return text_str.upper()
|
|
|
|
|
|
async def rentry(teks):
|
|
# buat dapetin cookie
|
|
cookie = SimpleCookie()
|
|
kuki = (await http.get("https://rentry.co")).cookies
|
|
cookie.load(kuki)
|
|
kukidict = {key: value.value for key, value in cookie.items()}
|
|
# headernya
|
|
header = {"Referer": "https://rentry.co"}
|
|
payload = {"csrfmiddlewaretoken": kukidict["csrftoken"], "text": teks}
|
|
return (
|
|
(
|
|
await http.post(
|
|
"https://rentry.co/api/new",
|
|
data=payload,
|
|
headers=header,
|
|
cookies=kukidict,
|
|
)
|
|
)
|
|
.json()
|
|
.get("url")
|
|
)
|
|
|
|
|
|
def get_provider(url):
|
|
def pretty(names):
|
|
name = names[1]
|
|
if names[0] == "play":
|
|
name = "Google Play Movies"
|
|
elif names[0] == "hbogoasia":
|
|
name = "HBO Go Asia"
|
|
elif names[0] == "maxstream":
|
|
name = "Max Stream"
|
|
elif names[0] == "klikfilm":
|
|
name = "Klik Film"
|
|
return name.title()
|
|
|
|
netloc = urlparse(url).netloc
|
|
return pretty(netloc.split("."))
|
|
|
|
|
|
async def search_jw(movie_name: str, locale: str):
|
|
m_t_ = ""
|
|
try:
|
|
response = (
|
|
await http.get(
|
|
f"https://yasirapi.eu.org/justwatch?q={movie_name}&locale={locale}"
|
|
)
|
|
).json()
|
|
except:
|
|
return m_t_
|
|
if not response.get("results"):
|
|
LOGGER.error("JustWatch API Error or got Rate Limited.")
|
|
return m_t_
|
|
for item in response.get("results")["items"]:
|
|
if movie_name == item.get("title", ""):
|
|
offers = item.get("offers", [])
|
|
t_m_ = []
|
|
for offer in offers:
|
|
url = offer.get("urls").get("standard_web")
|
|
if url not in t_m_:
|
|
p_o = get_provider(url)
|
|
m_t_ += f"<a href='{url}'>{p_o}</a> | "
|
|
t_m_.append(url)
|
|
if m_t_ != "":
|
|
m_t_ = m_t_[:-2].strip()
|
|
break
|
|
return m_t_
|