discordbot_message.py

Credit to N3XIS over at VGStation13 for the majority of this script!

scripts/discordbot_message.py

@@ -0,0 +1,49 @@
+# 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)