mapfre.py

from mechanize import Browser
import os, socket, thread, random

nice_ids = ["52"]

def get_top(html):
    t = html.split("Mais Votados")[1]
    t = t.split("/div")[0]
    top = []
    for i in [ i.split('"')[0] for i in t.split("id=")[1:] ]:
        if i not in nice_ids:
            top.append(i)
            break
    return top + nice_ids*2

def get(c, d, sid, n):
    brs = []
    b = Browser()
    top = get_top(b.open("http://www.omaiorbarbeirodobrasil.com.br/ranking/ranking.do").read())
    for i in range(n):
        id = random.choice(top)
        print id
        brs.append((Browser(), id))
        try: 
            u = brs[i][0].open("http://www.omaiorbarbeirodobrasil.com.br/home-promocao/historia.do?id=%s" % id)
            img = brs[i][0].open("http://www.omaiorbarbeirodobrasil.com.br/captcha.do")
        except:
            continue
        open("images/%s_captcha%d.png" % (sid, i), 'w').write(img.read())
    d[sid] = brs
    c.send("OK")
    c.close()

def send(c, brs, codes):
    html = "<ul>"
    for i in range(min(len(brs), len(codes))):
        br = brs[i][0]
        id = brs[i][1]
        if id in nice_ids:
            nota = "5"
        else:
            nota = "1"
        data = "id=%s&nota=%s&codigo=%s&nocacheattr=1240840177958" % (id, nota, codes[i])
        try:
            r = br.open('http://www.omaiorbarbeirodobrasil.com.br/home-promocao/votar.do', data=data)
            resp = r.read()
            print resp
            if "sucesso!" in resp:
                html+="<li><b>OK!</b></li>"
            else:
                html+="<li><i>FALHOU!</i></li>"
        except:
            html+="<li><i>FALHOU!</i></li>"

    del brs
    html += "</ul><br/><br/>Total: %d <b>OK!</b> e %d <i>FALHOU!</i>" % (html.count("OK!"), html.count("FALHOU!"))
    html += "<h3><a href='/10'>Voltar</a></h3>"
    c.send(html)
    c.close()

def main():
    dic = dict()
    while True:
        try:
            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
            s.bind(("127.0.0.1", 12345))
            s.listen(1)
            c, d = s.accept()
            s.close()
            print "conectou: ", d
            cmd = c.recv(1024)
            print "cmd: ", cmd
            cmd = cmd.split("/")
            print cmd[1]
            if cmd[0] == "GET":
                thread.start_new_thread(get, (c, dic, cmd[1], int(cmd[2])))
            else:
                thread.start_new_thread(send, (c, dic[cmd[1]], cmd[2].split(";")))
                del dic[cmd[1]]
        except KeyboardInterrupt:
            import sys
            sys.exit()
        except Exception, e:
            print e
            c.send("FAIL")
            c.close()
            pass

main()