irc-scripts/autottm.py

71 lines
1.8 KiB
Python
Raw Permalink Blame History

This file contains invisible Unicode characters

This file contains invisible Unicode characters that are indistinguishable to humans but may be processed differently by a computer. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#-*- coding: utf-8 -*-
# Copyright (c) 2013 Anton Vilhelm Ásgeirsson <anton@antonvilhelm.is>
#
# Everyone is permitted to copy and distribute verbatim or modified
# copies of this license document, and changing it is allowed as long
# as the name is changed.
#
# DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
# TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION
#
# 0. You just DO WHAT THE FUCK YOU WANT TO.
# Greentext:
# ----------
# Modifies any messages starting with '>' to display green text, similar to
# the way popular imageboards do.
import weechat as w
import re
import requests
# Registstration variables.
SCRIPT_NAME = "autottm"
SCRIPT_AUTHOR = "God"
SCRIPT_VERSION = "0.1"
SCRIPT_LICENSE = "WTFPL"
SCRIPT_DESC = "Modify messages starting with 'https://cdn.discordapp.com' to display better link."
# Registration function.
w.register(
SCRIPT_NAME,
SCRIPT_AUTHOR,
SCRIPT_VERSION,
SCRIPT_LICENSE,
SCRIPT_DESC,
"",""
)
# Greentext callback function.
def greentext_cb(data, modifier, modifier_data, string):
try:
nick, msg = string.split('\t')
except ValueError as e: #Get rid of those python error messages.
return string
try:
discordindex = msg.index('https://')
except:
return string
wackindex = len(msg) -1
if msg.find('.png') > -1:
wackindex = msg.index('.png')
elif msg.find('.jpg') > -1:
wackindex = msg.index('.jpg')
else:
return string
msg = msg[discordindex:wackindex+4]
#print(discordindex)
#print(msg[discordindex:])
data = {
'url': msg
}
r = requests.post(url = 'https://ttm.sh', data = data)
print(r.text)
return "%s\t%s%s" % (nick, w.color('green'), r.text)
#return r.text
w.hook_modifier("weechat_print", "greentext_cb", "")