markov.py

#!/usr/bin/python2
# coding: utf-8

import re, datetime, random, time
CHAIN_LENGTH=3
START = '\1'
END = '\0'
ansi_escape = re.compile(r'\x1b[^m]+m')

data = {}
data_b = {}
def split(msg, length):
    m = msg.split()
    m = [START] + m + [END]
    for i in range(len(m) - length):
        yield m[i:i + length + 1]

def process(m, length=CHAIN_LENGTH, d=data):
    m = m.lower()
    m.replace("(", "")
    m.replace(")", "")
    m.replace("[", "")
    m.replace("]", "")
    for y in re.split("[.!?;\n]", m):
        for x in split(y, length):
            key = tuple(z.replace(",", "") for z in x[:length])
            value = x[length]
            d[key] = d.get(key, []) + [value.replace(",", "")]
            #key2 = tuple(z.replace(",", "") for z in x[1:length+1])
            #data_b[key2] = data_b.get(key2, []) + [x[0]]

def say(m):
    r = [ w for w in m.strip().split() if len(w) > 2 ]
    if not r:
        return ""
    out = random.choice(r)
    new = random.choice(data.get((START, out), [END]))
    if new == END:
        return say(m.replace(out, ""))
    c = random.choice(data.get((START, out, new), [END]))
    out = "%s %s" % (out, new)
    while c != END:
        out = "%s %s" % (out, c)
        o = out.split()
        c = random.choice(data.get((o[-3], o[-2], o[-1]), [END]))
    return out
try:
    for line in open("input.txt"):
        process(line)
        process(line, 2, data)
except IOError:
    pass
if __name__ == "__main__":
    while True:
        print say(raw_input("> "))

shirley.py

#!/usr/bin/python2
# coding: utf-8

from subprocess import Popen, PIPE, STDOUT
from functools import partial
import re, datetime, random, time
from glob import glob
from markov import say, process
ansi_escape = re.compile(r'\x1b[^m]+m')

molejo = """Acorda criancada ta na hora da gente brincar (oba)
Brincar de pique-esconde,pique-cola e de pique-ta, ta, ta, ta
Nessa brincadeira tambam tem pique-bandeira e amarelinha pra quem gosta de pular
E aquela brincadeira de beijar.
Eh essa ? nao,
Eh essa ? eh
Pera, uva, maca ou salada mista ?
Salada mista (beija beija)
Brincadeira de crianca
Como e bom, como e bom
Guardo ainda na lembranca
Paz, amor e esperanca
Diga aonde voce vai
Que eu vou varrendo
Vou varrendo, vou varrendo, vou varrendo, vou varrendo
Oh menininha eu sou seu fan
Danco contigo ate de manha
Na danca da bruxinha
Danca preta, danca loura
Agora todo mundo
Na dancinha da vassoura
Varre pra esquerda
Varre pra direita
Levanta poeira
Que essa danca eh porreta...
Piti pi piti pi piti pau!
Mas tome cuidado
Com o cabo da vassoura
Eh pior do que cenoura
Voce pode se dar mal..."""

funk = """é o pet é o pet
Na madruga boladona sentada na esquina
se ela dança eu danço
pra dançar créu tem que ter disposição
pra dançar créu tem que ter habilidade
traição é traição, romance é romance, amor é amor e um lance é um lance
Agora eu tô solteira e ninguém vai me segurar
Vem que vem que vem kicando
bota o dedinho na boca e faz cara de tarada
foge foge mulher maravilha foge com o super man
Eu vou pro baile procurar o meu negão
ela balança mas não pára
pra te enlouquecer todas que provaram não conseguem esquecer
Mas não se esqueça que eu sou vagabundo depois que a putaria começou rolar no mundo
Sou foda, Eu sou sinistro
Eu sou sinistro melhor que seu marido esculacho seu amigo no escuro eu sou um perigo
Faz quadradinho de 4
Papai jogo mel , e mamãe deu carinho , essa mulher , e pedaço de mal caminho
Ela desce ela kika , ela sobe ela empina
Olha os amigos lá em cima , jogando dinheiro do camarote
Bate o pé , treme a bunda e faz meu bonde passar mal
Ostentação fora do normal , quem tem moto faz amor e quem não tem passa mal
Garoto sem limite, se amarra em Red Bull com Whisky
Ela só pensa em beijar, beijar, beijar, beijar
Glamurosaaaaaa
Pre-pa-ra
To com o cu pegando fogo!
Beijinho no ombro pro recalque passar longe!"""


process(funk)
process(molejo)

p = Popen(['telegram', '-N'], stdin=PIPE, stdout=PIPE, stderr=STDOUT)

p.stdout.read(255) # read first bytes...
print "started"

line = ""
title = "voltei"

bomdia = """DYA!
Bundinha!!
Bom dia :D
Dia !!! :)"""


perdi = """PERDI!!!!
hm... perdi !!!! :P
hehe, perdi
gente, PERDI!!!!!!!!!
peeeeeeerdi :)
p e r d i  :-X """

def chance(perc):
    return random.randint(0, 99) > (100 - perc)

def wrt(m):
    p.stdin.write("msg " + title.replace(" ", "_") + " " + m + "\n")

def voice(m):
    pp = Popen(["espeak", "-vpt+f2", "'%s'" % m.replace("'", "'"), "-w", "/tmp/x.wav"], stdout=PIPE)
    pp.wait()
    pp = Popen(["rm", "/tmp/x.mp3"])
    pp.wait()
    pp = Popen(["ffmpeg", "-i", "/tmp/x.wav", "/tmp/x.mp3"])
    pp.wait()
    p.stdin.write("send_audio " + title.replace(" ", "_") + " /tmp/x.mp3\n")

def img(i):
    p.stdin.write("send_photo " + title.replace(" ", "_") + " " + i + "\n")

from PIL import Image

def glitch():
    f = open("/tmp/teleglitch.jpg", 'r')

    new = f.read(400)
    for c in f.read():
        if random.random() > random.choice([0.9992, 0.9993, 0.9994, 0.9995, 0.9996, 0.9997, 0.9998, 0.9999]):
            new += chr(int(ord(c) + random.random()*random.choice(range(100)))%256)
        else:
            new += c
    fw = open("/tmp/teleglitched.jpg", 'w')
    fw.write(new)
    fw.close()
    return "/tmp/teleglitched.jpg"

reading_msg = False
waiting_photo_download = False
while True:
    if line:
        print line.strip()
        if ansi_escape.sub("", line)[0].startswith("[") or line.startswith(">"):
            reading_msg = False
    if "changed title" in line:
        reading_msg = False
        title = line.split("changed title to ")[-1].strip()
        title = ansi_escape.sub("", title)
        if "shirleyyy" not in title:
            p.stdin.write("rename_chat %s shirleyyy!!!!\n" % title.replace(" ", "_"))
            print "changing title"

    elif line.startswith("Chat "):
        reading_msg = False
        title = ":".join(line.split("Chat ")[1].split(":")[:-1])
        title = ansi_escape.sub("", title)
        wrt(random.choice(["chegay .... :D", "oie", "voltay gent!!!", "aloooou!!1", "hein??"]))
        #img("disco.png")
    elif waiting_photo_download and "*** Done" in line:
        fname = ansi_escape.sub("", line).split("Done: ")[-1].strip()
        i = Image.open(fname)
        i.save("/tmp/teleglitch.jpg")
        newf = glitch()
        img(newf)
    elif title in line and ">>>" in line or reading_msg:
        reading_msg = True
        msg = line.split(">>> ")[-1]
        msg_id = "not"
        try:
            i = 0
            while 'unread' in msg_id or not msg_id.isdigit():
                msg_id = ansi_escape.sub("", line).strip().split()[i]
                i+=1
        except Exception, e:
            print e
            msg_id = 0
        if title in line:
            user = ansi_escape.sub("", line.split(">>> ")[0].split(title)[-1].strip())
        else:
            user = 'noone'
        if "shirley" in user.lower():
            pass
        else:
            if msg.lower().startswith("qotd"):
                wrt("%s" % (Popen(["fortune"], stdout=PIPE).stdout.read().replace("\n", " ")))
            elif msg.lower().startswith("hora certa do pupi") or msg.lower().startswith("pupi ta chegando?"):
                wrt(random.choice(["atrasado ...", "chegando..", "quase la..", "hm...", "que horas sao?"]))
            elif msg.startswith("hora certa"):
                wrt(datetime.datetime.now().strftime("%H:%M:%S"))
            elif "molej" in msg.lower():
                wrt("%s" % random.choice(molejo.split("\n")))
            elif "funk" in msg.lower():
                wrt("%s" % random.choice(funk.split("\n")))
            elif "piroca" in msg.lower():
                wrt("8%sD" % ("="*random.randint(1,40)))
            elif "xxt" in msg.lower() or "xoxota" in msg.lower():
                img(random.choice(glob("pussy*.jpg")))
            elif msg.lower().startswith("dia ") or msg.lower().startswith("dia!") or msg.lower().startswith("dya") or "bom dia" in msg.lower() or "bom dya" in msg.lower() or "dia!" in msg.lower():
                wrt("%s" % (random.choice(bomdia.split("\n"))))
            elif "perd" in msg.lower():
                wrt(random.choice(perdi.split("\n")))
            elif "[photo]" in msg and 'shirley porter' not in line.lower():
                p.stdin.write("load_photo %s\n" % msg_id)
                waiting_photo_download = True
            else:
                ss = say(msg.lower())
                if ss.strip()[:-2] in msg.lower().strip().replace(",", ""):
                    print "SAME:", ss
                    pass #img(random.choice(glob("*.jpg") + glob("*.png")))
                else:
                    f = open("porter.txt", 'a')
                    f.write("FALA: " + msg)
                    f.write("RESPOSTA: " + ss + "\n\n")
                    f.close()
                    time.sleep(.1)
                    if chance(5):
                        voice(ss)
                    else:
                        wrt(ss)
            mm = msg.lower().strip().replace("\x1b[0m", "")
            process(mm)
            f = open("input.txt", 'a')
            f.writelines([mm + "\n"])
            f.close()
    time.sleep(0.5)
    line_old = line
    line = p.stdout.readline()
    if line == line_old:
        line = ""