diff --git a/commands.py b/commands.py index e69de29..4e80cf9 100644 --- a/commands.py +++ b/commands.py @@ -0,0 +1,155 @@ +import random + +user = [] +jokes = [ + ["Testing","Testing you!"], + ["singing pokemon","Jiggalypuff! 🎙️"] +] +#jokes = [] + +def handle_command(state,message): + for command in commands: + if command in message: + result = commands[command](state,message) + print("Bot: "+result["response"]) + return result["state"] + else: + if len(user) > 0: + try: + result = commands[user[0]](state,message) + print("Bot: "+result["response"]) + return result["state"] + except: + print(e) + print("Error saving state:",message) + pass + return state + + """ + Knock-knock implimentation + + when you receive message, read state. + user: knock knock + bot: who's there? -> bot responds and registers state + (user.state=[knock-knock]) + user: {statement} + bot: {statement} who? -> bot responds and stores statement + (user.stage=[knock-knock,{statement}]) + user: {punchline} + bot: Nice one! -> bot saves knock-knock joke + """ + +def nevermind(state,message): + return { + "response":"starting over.", + "state":[] + } + +def knock_knock(state,message): + #return {response:"",state:""} + try: + if state[0] != "knock knock": + #clear the state + state = [] + except: + pass + if len(state) == 0: + state.append("knock knock") + return { + "response":"Who's there?", + "state": state + } + if len(state) == 1: + state.append(message) + return { + "response":message + " who?", + "state":state + } + if len(state) == 2: + state.append(message) + jokes.append([ + state[1],state[2] + ]) + print("joke registered.") + return { + "response":"Hah! very funny!", + "state":[] + } + + return { + "response":"Cleared.", + "state":[] + } + +"""'joke' command + ------------- + user:!joke + bot:Knock knock -> bot initializes state + (user.state=[joke]) + user: Who's there? + bot: {statement} -> bot responds with chosen joke + (user.state=[joke,chosenjoke]) + user: {statement} who? + bot: + + """ +def joke(state,message): + if len(jokes) == 0: + return { + "response":"I don't know any jokes...", + "state":[] + } + try: + if state[0] != "joke": + state = [] + except: + pass + + if len(state) == 0: + return { + "response":"Knock knock", + "state":["joke"] + } + if len(state) == 1: + if message.find("there?") > 0 : + #get random joke index + state.append(random.randint(0,len(jokes)-1)) + return { + "response":jokes[state[1]][0], + "state":state + } + else: + return { + "response":"ask, \"who's there?\"", + "state":state + } + if len(state) == 2: + if message.find("who?") > 0: + return { + "response":jokes[state[1]][1], + "state":[] + } + else: + return { + "response":"ask, \""+ jokes[state[1]][0] +" who?\"", + "state":state + } + + return state + + +commands = { + "nevermind":nevermind, + "knock knock":knock_knock, + "joke":joke +} + + + + +if __name__ == '__main__': + msg = "" + while msg != "exit": + msg = input("user:") + user = handle_command(user,msg) + #print("state:",user) \ No newline at end of file diff --git a/faq.py b/faq.py new file mode 100644 index 0000000..e69de29 diff --git a/knockknock.py b/knockknock.py new file mode 100644 index 0000000..e65871c --- /dev/null +++ b/knockknock.py @@ -0,0 +1 @@ +print("knock-knock functionality loaded") \ No newline at end of file diff --git a/main.py b/main.py index 3b54cb4..0e5ee0b 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,7 @@ import json import nio import simplematrixbotlib as botlib +import commands data = json.loads(open('data.json').read()) dfaq = data["faq"] @@ -20,7 +21,6 @@ def load_faq(): dfaq = json.loads(open('data.json').read())["faq"] return dfaq["header"] + "".join(["\n"+q["question"]+"\n"+q["answer"] for q in dfaq["questions"]]) + "\n\n"+dfaq["footer"] - @bot.listener.on_message_event async def faq(room, message): match = botlib.MessageMatch(room, message, bot, PREFIX) @@ -59,5 +59,8 @@ async def faqresponse(room, message): response + dfaq["footer"] ) + + + bot.run()