bluebubbles-bot/persona/skills/greeter.py

from __future__ import annotations
from typing import List
import persona
from persona import PersonaBaseSkill
from dataclasses import dataclass
import datetime
import logging
import random
import re

BACKOFF_SEC = 30

class PersonaSkill(PersonaBaseSkill) :
	"""A simple test skill that responds to the message 'Hello' with 'Hello!'"""

	def __init__(self) :
		self.last_check = datetime.datetime.now().timestamp() - BACKOFF_SEC
		self.log = logging.getLogger(__name__)
		self.log = logging.LoggerAdapter(self.log,{'log_module':'greeter'})

	def match_intent(self,message: Message) -> Bool :
		# Don't respond if you've responded already recently
		if datetime.datetime.now().timestamp() < (self.last_check + BACKOFF_SEC) :
			self.log.warn('Responding too fast, not responding again.')
			return False
		matches = re.search('^Hello$', message.text)
		if matches :
			return True
		return False

	def respond(self, message: Message) -> Message :
		"""Respond to a message by generating another message."""
		response_options = ['Hello!','Howdy!','Hello there!','What\'s up!','Hi there!']
		response_text = random.choice(response_options)
		return persona.Message(text=response_text,sender_identifier=message.sender_identifier,chat_identifier=message.chat_identifier,attachments=[],timestamp=datetime.datetime.now()+datetime.timedelta(seconds=15), recipients=[message.sender_identifier], identifier=None, meta={})