diff --git a/__pycache__/chatlog.cpython-36.pyc b/__pycache__/chatlog.cpython-36.pyc index 2783f7e..123c799 100644 Binary files a/__pycache__/chatlog.cpython-36.pyc and b/__pycache__/chatlog.cpython-36.pyc differ diff --git a/__pycache__/markov.cpython-36.pyc b/__pycache__/markov.cpython-36.pyc index 84c0666..9644ffb 100644 Binary files a/__pycache__/markov.cpython-36.pyc and b/__pycache__/markov.cpython-36.pyc differ diff --git a/__pycache__/velasco.cpython-36.pyc b/__pycache__/velasco.cpython-36.pyc new file mode 100644 index 0000000..8a24ab3 Binary files /dev/null and b/__pycache__/velasco.cpython-36.pyc differ diff --git a/chatlog.py b/chatlog.py index a1eed2a..7da5994 100644 --- a/chatlog.py +++ b/chatlog.py @@ -59,3 +59,7 @@ class Chatlog(object): return new_log else: return Chatlog(lines[0], lines[1], lines[2], lines[4:], int(lines[3])) + + def fuse_with(chatlog): + self.count += chatlog.count + self.gen.fuse_with(chatlog.gen) diff --git a/markov.py b/markov.py index 3e39383..f8b42aa 100644 --- a/markov.py +++ b/markov.py @@ -30,8 +30,8 @@ def triples(wordlist): class Markov(object): def __init__(self, text=None, from_json=False): + self.cache = {} if not from_json: - self.cache = {} if text is not None: for line in text: self.add_text(line) @@ -73,3 +73,11 @@ class Markov(object): else: w1, w2 = w2, random.choice(self.cache[getkey(w1, w2)]) return ' '.join(gen_words) + + def fuse_with(self, gen): + d = gen.cache + for key in gen.cache: + if key in self.cache: + self.cache[key].extend(d[key]) + else: + self.cache[key] = list(d[key]) diff --git a/ocsalev.py b/ocsalev.py new file mode 100644 index 0000000..7d9e80b --- /dev/null +++ b/ocsalev.py @@ -0,0 +1,180 @@ +#!/usr/bin/env python3 + +# FAILED ATTEMPT TO MAKE BOT THAT USES VELASCO MEMORY IN ALL GROUPS SIMULTANEOUSLY + +import sys, os +from telegram.ext import Updater, CommandHandler, MessageHandler, Filters +from telegram.error import * +from markov import * +from velasco import GUILLERMO_ID, LOG_DIR, LOG_EXT +import logging +import argparse + +# Enable logging +logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', + level=logging.INFO) + +logger = logging.getLogger(__name__) + +generator = Markov() +chatfreqs = {} + +DEFAULT_FREQ = 10 + +def parse_file(text): + lines = text.splitlines() + if lines[1] != "private": + ident = lines[0] + freq = int(lines[3]) + chatfreqs[ident] = (freq, 0) + if lines[4] == "dict:": + cache = '\n'.join(lines[5:]) + gen = Markov.from_json(cache) + return gen + else: + return Markov(lines[4:]) + else: + return Markov() + +def load_gen(path): + open_file = open(path, 'r') + gen = parse_file(open_file.read()) + open_file.close() + return gen + +def wake(bot): + global generator + directory = os.fsencode(LOG_DIR) + + for file in os.listdir(directory): + filename = os.fsdecode(file) + if filename.endswith(LOG_EXT): + gen = load_gen(LOG_DIR + filename) + generator.fuse_with(gen) + +def start(bot, update): + update.message.reply_text('WHADDUP NERD') + +def help(bot, update): + update.message.reply_text("""I ANSWER TO + +/start - HELLO +/about - MY BIOGRAPHY +/help - THIS +/freq - HOW LONG I WAIT TO SPEAK +/speak - I SPEAK + """) + +def about(bot, update): + update.message.reply_text('I AM LIKE @velascobot BUT STRONGER. THE TRUE SELF') + +def echo(bot, update): + text = update.message.text.split(None, 2) + if len(text) > 1: + text = text[1] + update.message.reply_text(text) + +def error(bot, update, error): + logger.warn('Update "%s" caused error "%s"' % (update, error)) + +def read(bot, update): + global generator + if not "group" in update.message.chat.type: + update.message.reply_text("I ONLY TALK IN GROUPS") + return + generator.add_text(update.message.text + TAIL) + chat = update.message.chat + ident = str(chat.id) + if not ident in chatfreqs: + chatfreqs[ident] = (DEFAULT_FREQ, 0) + freq, count = chatfreqs[ident] + if count%freq == 0: + msg = generator.generate_markov_text() + try: + bot.sendMessage(ident, msg) + count = 0 + except TimedOut: + chatfreqs[ident] = (freq + CHAT_INC, count) + print("Increased freq for chat " + chat.title + " [" + ident + "]") + chatfreqs[ident] = (freq, count+1) + +def speak(bot, update): + global generator + if not "group" in update.message.chat.type: + update.message.reply_text("I ONLY TALK IN GROUPS") + return + chat = update.message.chat + ident = str(chat.id) + if not ident in chatfreqs: + chatfreqs[ident] = (DEFAULT_FREQ, 0) + msg = generator.generate_markov_text() + update.message.reply_text(msg) + +def get_chatlogs(bot, update): + if str(update.message.chat.id) == GUILLERMO_ID: + bot.sendMessage(GUILLERMO_ID, "HECK YOU") + +def set_freq(bot, update): + ident = str(update.message.chat.id) + if not ident in chatfreqs: + chatfreqs[ident] = (DEFAULT_FREQ, 0) + freq, count = chatfreqs[ident] + if not len(update.message.text.split()) > 1: + reply = "I WAIT FOR " + str(freq) + " MESSAGES" + else: + try: + value = update.message.text.split()[1] + value = int(value) + chatfreqs[ident] = (value, count) + reply = "I NOW WAIT FOR " + str(value) + " MESSAGES" + if value > freq: + reply += "\nYOU WILL NOT SILENCE ME" + except: + reply = "WHAT THE HECK. IMMA STILL WAIT FOR " + str(freq) + " MESSAGES" + update.message.reply_text(reply) + +def stop(bot, update): + chat = update.message.chat + ident = str(chat.id) + del chatfreqs[ident] + +def main(): + parser = argparse.ArgumentParser(description='A Telegram markovbot.') + parser.add_argument('token', metavar='TOKEN', help='The Bot Token to work with the Telegram Bot API') + + args = parser.parse_args() + + # Create the EventHandler and pass it your bot's token. + updater = Updater(args.token) + + # Get the dispatcher to register handlers + dp = updater.dispatcher + + # on different commands - answer in Telegram + dp.add_handler(CommandHandler("start", start)) + dp.add_handler(CommandHandler("about", about)) + dp.add_handler(CommandHandler("help", help)) + dp.add_handler(CommandHandler("freq", set_freq)) + dp.add_handler(CommandHandler("list", get_chatlogs)) + dp.add_handler(CommandHandler("stop", stop)) + dp.add_handler(CommandHandler("speak", speak)) + + # on noncommand i.e message - echo the message on Telegram + # dp.add_handler(MessageHandler(Filters.text, echo)) + dp.add_handler(MessageHandler(Filters.text, read)) + + # log all errors + dp.add_error_handler(error) + + wake(updater.bot) + + # Start the Bot + updater.start_polling() + + # Run the bot until you press Ctrl-C or the process receives SIGINT, + # SIGTERM or SIGABRT. This should be used most of the time, since + # start_polling() is non-blocking and will stop the bot gracefully. + updater.idle() + +if __name__ == '__main__': + main() diff --git a/velasco.py b/velasco.py index 8d6ab73..7a935ac 100755 --- a/velasco.py +++ b/velasco.py @@ -19,14 +19,16 @@ disabled = {} GUILLERMO_ID = "8379173" CHAT_INC = 5 CHAT_SAVE = 15 +LOG_DIR = "chatlogs/" +LOG_EXT = ".txt" def wake(bot): - directory = os.fsencode("chatlogs/") + directory = os.fsencode(LOG_DIR) for file in os.listdir(directory): filename = os.fsdecode(file) - if filename.endswith(".txt"): - chat = loadchat("chatlogs/" + filename) + if filename.endswith(LOG_EXT): + chat = loadchat(LOG_DIR + filename) chatlogs[chat.id] = chat print("loaded chat " + chat.title + " [" + chat.id + "]") continue @@ -45,7 +47,7 @@ def start(bot, update): update.message.reply_text('cowabunga') def savechat(chatlog): - open_file = open('chatlogs/' + chatlog.id + '.txt', 'w') + open_file = open(LOG_DIR + chatlog.id + LOG_EXT, 'w') open_file.write(chatlog.to_txt()) open_file.close() @@ -70,7 +72,11 @@ def about(bot, update): update.message.reply_text('I am yet another Markov Bot experiment. I read everything you type to me and then spit back nonsensical messages that look like yours') def echo(bot, update): - update.message.reply_text(update.message.text) + text = update.message.text.split(None, 2) + if len(text) > 1: + text = text[1] + chatlog.add_msg(text) + update.message.reply_text(text) def error(bot, update, error): logger.warn('Update "%s" caused error "%s"' % (update, error)) @@ -168,7 +174,7 @@ def set_freq(bot, update): def stop(bot, update): chatlog = chatlogs[update.message.chat.id] del chatlogs[chatlog.id] - os.remove("chatlogs/" + chatlog.id + ".txt") + os.remove(LOG_DIR + chatlog.id + LOG_EXT) print("I got blocked. Removed user " + chatlog.id) def main(): @@ -197,7 +203,7 @@ def main(): # dp.add_handler(MessageHandler(Filters.text, echo)) dp.add_handler(MessageHandler(Filters.text, read)) - # chatlog all errors + # log all errors dp.add_error_handler(error) wake(updater.bot)