45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
import re
|
|
|
|
from discord.ext import commands
|
|
|
|
from tuxbot.cogs.Network.functions.exceptions import (
|
|
InvalidIp,
|
|
InvalidDomain,
|
|
InvalidQueryType,
|
|
)
|
|
|
|
|
|
def _(x):
|
|
return x
|
|
|
|
|
|
class IPConverter(commands.Converter):
|
|
async def convert(self, ctx, argument): # skipcq: PYL-W0613
|
|
argument = argument.replace("http://", "").replace("https://", "")
|
|
argument = argument.rstrip("/")
|
|
|
|
if argument.startswith("`") and argument.endswith("`"):
|
|
argument = argument.lstrip("`").rstrip("`")
|
|
|
|
return argument.lower()
|
|
|
|
|
|
class DomainConverter(commands.Converter):
|
|
async def convert(self, ctx, argument): # skipcq: PYL-W0613
|
|
if not argument.startswith("http"):
|
|
return f"http://{argument}"
|
|
|
|
return argument
|
|
|
|
|
|
class QueryTypeConverter(commands.Converter):
|
|
async def convert(self, ctx, argument): # skipcq: PYL-W0613
|
|
return argument.lower()
|
|
|
|
|
|
class IPVersionConverter(commands.Converter):
|
|
async def convert(self, ctx, argument): # skipcq: PYL-W0613
|
|
if not argument:
|
|
return argument
|
|
|
|
return argument.replace("-", "").replace("ip", "").replace("v", "")
|