mirror of
https://github.com/nexy7574/LCC-bot.git
synced 2024-09-19 18:16:34 +01:00
Properly use custom servers.
This commit is contained in:
parent
9276d7b62a
commit
8070fe497e
1 changed files with 26 additions and 0 deletions
|
@ -3,6 +3,7 @@ import base64
|
||||||
import functools
|
import functools
|
||||||
import glob
|
import glob
|
||||||
import io
|
import io
|
||||||
|
import ipaddress
|
||||||
import json
|
import json
|
||||||
import typing
|
import typing
|
||||||
import zlib
|
import zlib
|
||||||
|
@ -1897,6 +1898,9 @@ class OtherCog(commands.Cog):
|
||||||
context = self.context_cache[context]
|
context = self.context_cache[context]
|
||||||
|
|
||||||
content = None
|
content = None
|
||||||
|
RESTRICTED_SERVERS = (
|
||||||
|
"100.106.34.86:11434"
|
||||||
|
)
|
||||||
try_hosts = {
|
try_hosts = {
|
||||||
"127.0.0.1:11434": "localhost",
|
"127.0.0.1:11434": "localhost",
|
||||||
"100.106.34.86:11434": "NexTop",
|
"100.106.34.86:11434": "NexTop",
|
||||||
|
@ -1929,6 +1933,28 @@ class OtherCog(commands.Cog):
|
||||||
else:
|
else:
|
||||||
return await ctx.respond(":x: No servers available.")
|
return await ctx.respond(":x: No servers available.")
|
||||||
else:
|
else:
|
||||||
|
try:
|
||||||
|
try:
|
||||||
|
server, port = server.split(":", 1)
|
||||||
|
port = int(port)
|
||||||
|
except ValueError:
|
||||||
|
port = 11434
|
||||||
|
server = ipaddress.ip_address(server)
|
||||||
|
if not isinstance(server, ipaddress.IPv4Address):
|
||||||
|
raise ValueError
|
||||||
|
server = "%s:%s" % (server, port)
|
||||||
|
except ValueError:
|
||||||
|
try:
|
||||||
|
if not server.startswith("http"):
|
||||||
|
server = "http://" + server
|
||||||
|
server = urlparse(server)
|
||||||
|
if not server.netloc:
|
||||||
|
raise ValueError
|
||||||
|
except ValueError:
|
||||||
|
return await ctx.respond(f":x: Failed to parse {server!r} as a domain/IPv4.")
|
||||||
|
else:
|
||||||
|
server = server.netloc
|
||||||
|
|
||||||
host = server
|
host = server
|
||||||
|
|
||||||
embed = discord.Embed(
|
embed = discord.Embed(
|
||||||
|
|
Loading…
Reference in a new issue