swee/MeowNex
swee/MeowNex
1
0
Fork 1
Code Issues Pull requests Activity Actions

Update sweebot.py

Browse source
This commit is contained in:
Swee 2024-11-11 16:46:44 -08:00
parent a959cdd07d
commit e3426ef2ae
1 changed files with 36 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
sweebot.py
View file

@ -75,7 +75,24 @@ color_map = {
'\033[0m': '\x03', # Reset
}
pattern = re.compile(r'\033\[\d+(;\d+)*m|\033\[\?25[lh]|\033\[\?7[lh]|\033\[\d+C|\033\[\d+A|\033\[\d+D|\033\[\d+B')
def humanbytes(B):
"""Return the given bytes as a human friendly KB, MB, GB, or TB string."""
B = float(B)
KB = float(1024)
MB = float(KB ** 2) # 1,048,576
GB = float(KB ** 3) # 1,073,741,824
TB = float(KB ** 4) # 1,099,511,627,776
if B < KB:
return '{0}{1}'.format(B,'Bytes' if 0 == B > 1 else 'Byte')
elif KB <= B < MB:
return '{0:.2f}KB'.format(B / KB)
elif MB <= B < GB:
return '{0:.2f}MB'.format(B / MB)
elif GB <= B < TB:
return '{0:.2f}GB'.format(B / GB)
elif TB <= B:
return '{0:.2f}TB'.format(B / TB)
def replace_color_codes(text):
def replacer(match):
code = match.group(0)
@ -164,6 +181,7 @@ channel_irc = ["##sweezero"]
botnick_irc = environ.get('SBnick')
botnickpass_irc =environ.get('SBuser')
botpass_irc = environ.get('SBpass')
allowedparse = ["text/html", "application/x-httpd-php"]
irc = bot_irc()
irc2 = bot_irc()
irc3 = bot_irc()
@ -591,27 +609,36 @@ while True:
if i[:8] == "https://":
try:
e = get(i, headers=headers, timeout=10)
if content_type in allowedparse:
if e.ok:
soup = BeautifulSoup(e.text, 'html.parser')
multiline("(" + nick + ") " + (soup.title.string if soup.title != None else "[No title provided]"), channel)
else:
multiline("(" + nick + ") [HTTP " + str(e.status_code) + "]", channel)
else:
multiline("(" + nick + ") [" + humanbytes(content_len) + " " + str(content_type) + "]", channel)
except rex.SSLError as ex:
multiline("(" + nick + ") [SSL Error: " + str(ex.message) + "]", channel)
except Exception as ex:
multiline("(" + nick + ") [Request error: " + str(ex.message) + "]", channel)
elif i[:7] == "http://":
e = get(i, headers=headers, timeout=10)
header = e.headers
content_type = header.get('content-type')
content_len = header.get('Content-length')
if content_type in allowedparse:
if e.ok:
soup = BeautifulSoup(e.text, 'html.parser')
multiline("(" + nick + ") " + (soup.title.string if soup.title != None else "[No title provided]"), channel)
else:
multiline("(" + nick + ") [HTTP " + str(e.status_code) + "]", channel)
else:
multiline("(" + nick + ") [" + humanbytes(content_len) + " " + str(content_type) + "]", channel)
except:
print(traceback.format_exc())
elif "JOIN" in text and "#nixsanctuary" in text:
nick = text.split(":")[1].split("!")[0]
if not "swe" in nick:
if not "Meow" in nick:
irc.send_irc("##hiya", "hiya: " + nick + " has joined #nixsanctuary")
pass