catspace-dev/unicheckbot
1
1
Fork 0
mirror of https://github.com/catspace-dev/unicheckbot.git synced 2024-11-25 07:13:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix #10

Browse Source
This commit is contained in:
kiriharu 2021-01-06 00:35:25 +03:00
parent 714dc2e4a2
commit b33b938b95
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
apps/tgbot/tgbot/handlers/default/whois.py
View File

@ -19,6 +19,7 @@ no_domain_text = """
localhost_exception = "❗Локальные адреса запрещены!" localhost_exception = "❗Локальные адреса запрещены!"
# TODO: Very shitty code. I should rewrite this.
def create_whois_message(domain: str) -> str: def create_whois_message(domain: str) -> str:
domain_info = whois.whois(domain) domain_info = whois.whois(domain)
domain_name = domain_info.get("domain_name") domain_name = domain_info.get("domain_name")
@ -44,13 +45,22 @@ def create_whois_message(domain: str) -> str:
message += f"\n📅 Дата окончания:: {expiration_date}\n" message += f"\n📅 Дата окончания:: {expiration_date}\n"
if address := domain_info.get("address"): if address := domain_info.get("address"):
message += f"\n📖 Адрес: {address}" if isinstance(address, list):
message += "\n📖 Адрес: \n" + str.join("\n", [f" * <code>{address_obj}</code>" for address_obj in address])
else:
message += f"\n📖 Адрес: {address}"
if city := domain_info.get("city"): if city := domain_info.get("city"):
message += f"\n🏘 Город: {city}" if isinstance(city, list):
message += "\n🏘 Город: \n" + str.join("\n", [f" * <code>{city_obj}</code>" for city_obj in city])
else:
message += f"\n🏘 Город: {city}"
if country := domain_info.get("country"): if country := domain_info.get("country"):
message += f"\n🏳️ Страна: {country}" message += f"\n🏳️ Страна: {country}"
if name := domain_info.get("name"): if name := domain_info.get("name"):
message += f"\n💬 Имя: {name}" if isinstance(name, list):
message += "\n🏘 💬 Имя: \n" + str.join("\n", [f" * <code>{name_obj}</code>" for name_obj in name])
else:
message += f"\n💬 Имя: {name}"
if org := domain_info.get("org"): if org := domain_info.get("org"):
message += f"\n💼 Организация: {org}" message += f"\n💼 Организация: {org}"
if zipcode := domain_info.get("zipcode"): if zipcode := domain_info.get("zipcode"):
@ -65,6 +75,7 @@ def create_whois_message(domain: str) -> str:
message += f"\n🔐 DNSSec: {dnssec}" message += f"\n🔐 DNSSec: {dnssec}"
return message return message
@rate_limit @rate_limit
async def whois_cmd(msg: Message): async def whois_cmd(msg: Message):
args = msg.text.split(" ") args = msg.text.split(" ")