83 lines
4.5 KiB
Python
83 lines
4.5 KiB
Python
import re
|
|
from bs4 import BeautifulSoup
|
|
from common import get_api
|
|
from common import list_append
|
|
from common import list_read
|
|
from common import list_write
|
|
|
|
|
|
# Messages
|
|
mensaje = "Alguien que te aprecia mucho quiere recordarte que eres una persona maravillosa :ablobcatheartsqueeze: ¡Sigue así, "
|
|
mensaje_croqueta = "Alguien que te aprecia mucho quiere enviarte croquetas :croqueta: :croqueta: :croqueta:"
|
|
mensaje_cumple = ["Alguien me ha revelado que hoy es tu día, ", ". ¡Feliz cumpleaños de su parte! :blobcatbirthday:"]
|
|
mensaje_mismo = "La persona más importante que debes apreciar eres tú. ¡Eres increíble! ❤"
|
|
mensaje_nobot = "La cuenta objetivo tiene la etiqueta #nobot en su biografía. ¡No tengo poder aquí!"
|
|
mensaje_aviso = "Has intentado apreciar a alguien pero no has usado un mensaje directo/privado. ¡Tienes que mencionarme en un mensaje directo/privado para que funcione!"
|
|
mensaje_error = "No pude procesar tu apreciación. ¡Asegúrate de no incluir saltos de línea ni otros caracteres extraños! El mensaje debería ser tal que así: \"@apreciabot@masto.es usuario@servidor\n"
|
|
mensaje_no_encontrado = " No se pudo encontrar la cuenta del usuario especificado. \n\nRevisa que has escrito bien la cuenta con el formato \"usuario@servidor\" (por ejemplo, rober@masto.es, excluyendo el primer @ para evitar mencionarlo).\n\nSi lo has mencionado por error, ¡borra el mensaje antes de que se de cuenta!"
|
|
|
|
# Initialization
|
|
bot_name = 'apreciabot'
|
|
api = get_api('masto.es', bot_name)
|
|
last_ids = list_read(bot_name)
|
|
max_notifications=10
|
|
new_last_ids=[]
|
|
notifications = api.notifications(types=["mention"],limit=max_notifications)
|
|
no_unicode_spaces_pattern = r"[\u200B-\u200D\u202A\u202C\uFEFF]"
|
|
mode_croqueta_words=["croqueta", "croquetas"]
|
|
mode_cumple_words=["cumple", "cumpleaños", "felicidades"]
|
|
|
|
def check_mode(mode_words, content):
|
|
for word in mode_words:
|
|
if ( (word in content) or ( ('"' + word +'"') in content) ):
|
|
return True
|
|
return False
|
|
|
|
for n in notifications:
|
|
new_last_ids.append(n['id'])
|
|
|
|
# Some notifications may have been deleted since last fetch
|
|
# Therefore, it is better to check less than the maximum number of notifications
|
|
for i in range(0, max_notifications - 5):
|
|
n = notifications[i]
|
|
if str(n['id']) not in last_ids:
|
|
# Mentions data are HTML paragraphs so we delete everything between <> to clean it up
|
|
rawContent = n['status']['content'].replace("</br >", " ").replace("<br>", " ")
|
|
text = BeautifulSoup(rawContent, "html.parser").get_text()
|
|
text = re.sub(" +", " ", text)
|
|
content = re.sub(no_unicode_spaces_pattern, "", text).split(" ")
|
|
try:
|
|
first_mention = content[0]
|
|
target = "@" + content[1]
|
|
user = "@" + n['account']['acct']
|
|
except:
|
|
api.status_reply(n['status'], mensaje_error)
|
|
continue
|
|
# The bot is meant to be anonymous so only allow directs
|
|
if n['status']['visibility'] == "direct":
|
|
if user == target:
|
|
api.status_reply(n['status'], mensaje_mismo, visibility="unlisted")
|
|
else:
|
|
# Find account if it is not known by the server
|
|
try:
|
|
api.search(target, result_type="accounts")
|
|
bio = api.account_lookup(target)
|
|
except:
|
|
api.status_post(user + mensaje_no_encontrado, in_reply_to_id=n['status']['id'], visibility="direct" )
|
|
else:
|
|
if "nobot" in bio['note']:
|
|
api.status_reply(n['status'], mensaje_nobot)
|
|
else:
|
|
#api.status_post(mensaje + target + "!", in_reply_to_id=n['status']['id'], visibility="unlisted")
|
|
if check_mode(mode_croqueta_words, content):
|
|
new_status = api.status_post(target + " " + mensaje_croqueta, visibility="unlisted")
|
|
elif check_mode(mode_cumple_words, content):
|
|
new_status = api.status_post(mensaje_cumple[0] + target + mensaje_cumple[1], visibility="unlisted")
|
|
else:
|
|
new_status = api.status_post(mensaje + target + "!", visibility="unlisted")
|
|
api.status_reply(n['status'], 'Tu muestra de aprecio ha sido enviada ❤️ ' + new_status['url'], visibility="direct")
|
|
elif first_mention == "@apreciabot" and n['status']['in_reply_to_id'] == None:
|
|
api.status_reply(n['status'], mensaje_aviso, visibility='direct')
|
|
|
|
list_write(bot_name, new_last_ids)
|