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

cover many domain zones with whois_vu package

Browse Source
This commit is contained in:
kiriharu 2021-01-13 23:49:23 +03:00
parent f7a8622875
commit 8ff4d8c16c
2 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
apps/tgbot/pyproject.toml
View File

@ -12,6 +12,7 @@ python-whois = "^0.7.3"
core = {path = "../core"} core = {path = "../core"}
aioinflux = "^0.9.0" aioinflux = "^0.9.0"
loguru = "^0.5.3" loguru = "^0.5.3"
whois-vu = "^0.2.0"
[tool.poetry.dev-dependencies] [tool.poetry.dev-dependencies]

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

@ -3,7 +3,7 @@ from typing import Optional
from whois import whois, parser from whois import whois, parser
from aiogram.types import Message from aiogram.types import Message
from dataclasses import dataclass from dataclasses import dataclass
from aiogram.utils.markdown import quote_html from whois_vu.api import WhoisSource
from tgbot.handlers.base import SimpleCommandHandler from tgbot.handlers.base import SimpleCommandHandler
from tgbot.handlers.errors import NotEnoughArgs, LocalhostForbidden from tgbot.handlers.errors import NotEnoughArgs, LocalhostForbidden
@ -29,7 +29,7 @@ class DomainAttrClass:
attr: str attr: str
# DOMAIN_ATTR_CLASSES order have matter! # DOMAIN_ATTR_CLASSES order matters!
DOMAIN_ATTR_CLASSES = [ DOMAIN_ATTR_CLASSES = [
DomainAttrClass("👤", "Регистратор", "registrar"), DomainAttrClass("👤", "Регистратор", "registrar"),
DomainAttrClass("📅", "Дата создания", "creation_date"), DomainAttrClass("📅", "Дата создания", "creation_date"),
@ -46,21 +46,26 @@ DOMAIN_ATTR_CLASSES = [
] ]
def whois_request(domain: str) -> parser.WhoisEntry:
domain_info = whois(domain)
if domain_info.get("domain_name") is None:
ws = WhoisSource().get(domain)
domain_info = parser.WhoisEntry.load(domain, ws.whois)
return domain_info
def create_whois_message(domain: str) -> str: def create_whois_message(domain: str) -> str:
try: try:
domain_info = whois(domain) domain_info = whois_request(domain)
except parser.PywhoisError as e: except parser.PywhoisError as e:
return f"❗ Домен {domain} свободен или не был найден." return f"❗ Домен {domain} свободен или не был найден."
domain_name = domain_info.get("domain_name") domain_name = domain_info.get("domain_name")
if domain_name is None:
return no_domain_text
if isinstance(domain_name, list): if isinstance(domain_name, list):
domain_name = domain_name[0] domain_name = domain_name[0]
message = f"\n📝 Информация о домене {domain_name.lower()}:" message = f"\n📝 Информация о домене {domain_name.lower()}:"
for i, domain_attr in enumerate(DOMAIN_ATTR_CLASSES): for i, domain_attr in enumerate(DOMAIN_ATTR_CLASSES):
# for pretty printing, DOMAIN_ATTR_CLASSES order have matter! # for pretty printing, DOMAIN_ATTR_CLASSES order matters!
if i in [2, 10]: if i in [2, 10]:
message += "\n" message += "\n"
resp = format_domain_item( resp = format_domain_item(
@ -76,6 +81,7 @@ def format_domain_item(icon, item_name, items) -> Optional[str]:
if not items: if not items:
return return
if isinstance(items, list): if isinstance(items, list):
items = map(str, items) # fix datetime bug
message = f"\n{icon} {item_name}:\n" message = f"\n{icon} {item_name}:\n"
message += str.join("\n", [f" * <code>{ns}</code>" for ns in list(set(map(str.lower, items)))]) message += str.join("\n", [f" * <code>{ns}</code>" for ns in list(set(map(str.lower, items)))])
else: else: