utils/server_status/check_server_status.py

@@ -0,0 +1,76 @@
+import sys
+import socket
+import time
+import ssl
+import paho.mqtt.client as mqtt
+
+# TODO: make port a constant? 80 for http
+# 443 for https
+def servertest(argv):
+    host = argv[1]
+    port = int(argv[2])
+
+    print (host)
+    print (port)
+
+    args = socket.getaddrinfo(host, port, socket.AF_INET, socket.SOCK_STREAM)
+    for family, socktype, proto, canonname, sockaddr in args:
+        s = socket.socket(family, socktype, proto)
+
+        print (sockaddr)
+        try:
+            s.connect_ex(sockaddr)
+        except socket.error:
+            return False
+        else:
+            s.close()
+            return True
+
+def on_log(client, userdata, level, buf):
+    print("log: ", buf)
+
+def on_connect(client, userdata, flags, rc):
+    if rc == 0:
+        client.connected_flag = True
+        print("connected to mqtt server")
+    else:
+        client.bad_connection_flag = True
+        if rc == 5:
+            print("broker requires authentication")
+
+def on_message(client, userdata, message):
+    print("message received " ,str(message.payload.decode("utf-8")))
+    print("message topic=",message.topic)
+    print("message qos=",message.qos)
+    print("message retain flag=",message.retain)
+
+if __name__ == "__main__":
+    if servertest(sys.argv):
+        print("Server is up")
+    else:
+        print("Server is down")
+
+broker_address = "205.185.116.243"
+print ("creating new mqtt client instance")
+client = mqtt.Client("dev-client")
+
+# initialize flags
+client.connected_flag = False
+client.bad_connection_flag = False
+
+
+client.on_log = on_log
+client.on_connect = on_connect
+client.tls_set(ca_certs="/root/mqtt_certs/chain.pem", certfile="/root/mqtt_certs/cert.pem", tls_version=ssl.PROTOCOL_TLSv1_2)
+client.on_message=on_message        #attach function to callback
+print ("connecting to broker")
+
+client.connect(broker_address, 8883)
+client.loop_start()
+#print("Subscribing to topic", "/rider/+/location")
+#client.subscribe("/rider/+/location")
+
+time.sleep(4)
+client.loop_stop()
+
+