githubmirror/vgstation13

scripts/ircbot_message.py

31 строка883 B

Mloc-Argent

Another tiny change to IRC nudge.

06 авг 2013, 18:55

06 авг 2013, 18:55f10c7a9

#!/usr/bin/env python2

# Four arguments, password host channel and message.
# EG: "ircbot_message.py hunter2 example.com #adminchannel ADMINHELP, people are killing me!"

import sys,cPickle,socket,HTMLParser

def pack():
    ht = HTMLParser.HTMLParser()
    
    passwd = sys.argv[1]
    ip = sys.argv[3]
    try:
        data = []
        for in_data in sys.argv[4:]: #The rest of the arguments is data
            data += {ht.unescape(in_data)}
    except:
        data = "NO DATA SPECIFIED"
    dictionary = {"ip":ip,"data":[passwd] + data}
    pickled = cPickle.dumps(dictionary)
    nudge(pickled)
def nudge(data):
    HOST = sys.argv[2]
    PORT = 45678
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.connect((HOST,PORT))
    s.send(data)
    s.close()

if __name__ == "__main__" and len(sys.argv) > 1: # If not imported and more than one argument
    pack()