From 533b6ee20b5aef335c51b9947e68375d8bea94d8 Mon Sep 17 00:00:00 2001 From: yakamok <38737288+yakamok@users.noreply.github.com> Date: Tue, 3 Jul 2018 00:38:12 +0200 Subject: [PATCH] option for external admin api --- send-view.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/send-view.py b/send-view.py index 085c72e..7586194 100644 --- a/send-view.py +++ b/send-view.py @@ -1,10 +1,17 @@ import socket import json +import sys GETDHT = '{"request":"getDHT", "keepalive":true}' GETSWITCHPEERS = '{"request":"getSwitchPeers"}' SERVER = "y.yakamo.org" +#gives the option to get data from an external server instead and send that +#if no options given it will default to localhost instead +if len(sys.argv) == 4: + host_port = (sys.argv[1], sys.argv[2]) +elif len(sys.argv) == 1: + host_port = ('localhost', 9001) def send_view_to_server(tosend): if tosend: @@ -21,10 +28,10 @@ def send_view_to_server(tosend): attempts -= 1 -def collect_dht_getswitchpeers(): +def collect_dht_getswitchpeers(serport): try: ygg = socket.socket(socket.AF_INET, socket.SOCK_STREAM) - ygg.connect(('localhost', 9001)) + ygg.connect(host_port) ygg.send(GETDHT) dhtdata = json.loads(ygg.recv(1024 * 15))