mirror of
https://github.com/Aurorastation/Aurora.3.git
synced 2025-12-22 16:12:19 +00:00
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
# nudge.py --channel="nudges|ahelps" --id="Server ID" --key="access key" Message! More message!
|
|
# Credit to the gents at VGStation13/N3XIS for this code.
|
|
|
|
import sys
|
|
import pickle
|
|
import socket
|
|
import argparse
|
|
import html
|
|
|
|
def pack(host, port, key, channel, message):
|
|
|
|
data = {}
|
|
|
|
data['key'] = key
|
|
data['channel'] = channel
|
|
|
|
try:
|
|
d = []
|
|
for in_data in message: # The rest of the arguments is data
|
|
d += [html.unescape(in_data)]
|
|
data['data'] = ' '.join(d)
|
|
|
|
# Buffer overflow prevention.
|
|
if len(data['data']) > 400:
|
|
data['data'] = data['data'][:400]
|
|
except:
|
|
data['data'] = "NO DATA SPECIFIED"
|
|
pickled = pickle.dumps(data)
|
|
nudge(host, port, pickled)
|
|
|
|
def nudge(hostname, port, data):
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
s.connect((hostname, port))
|
|
s.send(data)
|
|
s.close()
|
|
|
|
if __name__ == "__main__" and len(sys.argv) > 1: # If not imported and more than one argument
|
|
argp = argparse.ArgumentParser()
|
|
|
|
argp.add_argument('message', nargs='*', type=str, help='String to send to the server.')
|
|
|
|
argp.add_argument('--host', dest='hostname', default='localhost', help='Hostname expecting a nudge.')
|
|
argp.add_argument('--port', dest='port', type=int, default=5555, help='Port expecting a nudge.')
|
|
argp.add_argument('--channel', dest='channel', default='lobby', help='Channel flag to direct this message to.')
|
|
argp.add_argument('--key', dest='key', default='', help='Access key of the bot or receiving script.')
|
|
|
|
args = argp.parse_args()
|
|
|
|
pack(args.hostname, args.port, args.key, args.channel, args.message)
|