78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
import asyncio
|
|
|
|
import aiohttp
|
|
from bs4 import BeautifulSoup
|
|
|
|
from tuxbot.cogs.Linux.functions.exceptions import CNFException
|
|
|
|
|
|
def _(x):
|
|
return x
|
|
|
|
|
|
class CNF:
|
|
_url = "https://command-not-found.com/{}"
|
|
_content: BeautifulSoup
|
|
|
|
command: str
|
|
|
|
description: str = ""
|
|
meta: dict = {}
|
|
distro: dict = {}
|
|
|
|
def __init__(self, command: str):
|
|
self.command = command
|
|
|
|
# =========================================================================
|
|
# =========================================================================
|
|
|
|
async def fetch(self):
|
|
try:
|
|
async with aiohttp.ClientSession() as cs:
|
|
async with cs.get(self._url.format(self.command)) as s:
|
|
if s.status == 200:
|
|
self._content = BeautifulSoup(
|
|
await s.text(), "html.parser"
|
|
)
|
|
return self.parse()
|
|
|
|
except (aiohttp.ClientError, asyncio.exceptions.TimeoutError):
|
|
pass
|
|
|
|
raise CNFException(_("Something went wrong ..."))
|
|
|
|
def parse(self):
|
|
info = self._content.find("div", class_="row-command-info")
|
|
distro = self._content.find_all("div", class_="command-install")
|
|
|
|
try:
|
|
self.description = info.find("p", class_="my-0").text.strip()
|
|
except AttributeError:
|
|
self.description = "N/A"
|
|
|
|
try:
|
|
for m in info.find("ul", class_="list-group").find_all("li"):
|
|
row = m.text.strip().split("\n")
|
|
|
|
self.meta[row[0].lower()[:-1]] = row[1]
|
|
except AttributeError:
|
|
self.meta = {}
|
|
|
|
try:
|
|
del distro[0] # unused row
|
|
|
|
for d in distro:
|
|
self.distro[
|
|
d.find("dt").text.strip().split("\n")[-1].strip()
|
|
] = d.find("code").text
|
|
except (AttributeError, IndexError):
|
|
self.distro = {}
|
|
|
|
def to_dict(self):
|
|
return {
|
|
"command": self.command,
|
|
"description": self.description,
|
|
"meta": self.meta,
|
|
"distro": self.distro,
|
|
}
|