ws support

Dockerfile

@@ -1,5 +1,5 @@
 FROM python:3.10
-RUN python -m pip install glQiwiApi aiohttp
+RUN python -m pip install glQiwiApi aiohttp websocket-client
 ENV PYTHONUNBUFFERED 1
 WORKDIR /app
 COPY ./ ./

git_build/Dockerfile

@@ -1,5 +1,5 @@
 FROM python:3.10
-RUN python -m pip install glQiwiApi aiohttp
+RUN python -m pip install glQiwiApi aiohttp websocket-client
 WORKDIR /app
 
 ENV TZ=Europe/Moscow

paybot.facti13.py

@@ -10,6 +10,7 @@ from glQiwiApi.qiwi.exceptions import QiwiAPIError
 from backend_integration import BackendClient
 from colors import *
 import os, sys, asyncio
+import ws
 
 class Client:
 	wallet: QiwiWallet
@@ -117,12 +118,12 @@ class Client:
 if __name__ == "__main__":
 	print("Build date: "+os.getenv("BUILDDATE", "not setted"))
 
-	backend = BackendClient()
-	asyncio.get_event_loop().run_until_complete(backend.pulse())
-	if not backend.up:
-		print("backend not working down bot")
-		sys.exit(200)
-
+	def run():
 		client = Client()
-	client.backend = backend
+		client.backend = BackendClient()
 		client.run()
+
+	wsc = ws.WS(os.getenv("BACKEND_URL"), os.getenv("SECRET_KEY"))
+	ws.SERVICE_NAME = "qiwipay"
+	ws.START_AFTER_CONNECT = run
+	wsc.run()

ws.py

@@ -0,0 +1,61 @@
+import websocket, sys
+from threading import Thread
+from json import dumps
+from time import sleep
+import os
+
+START_AFTER_CONNECT = None
+SERVICE_NAME = None
+
+def create_pulser(ws: websocket.WebSocketApp):
+    def run(ws):
+        builddate = int(os.getenv("BUILDDATE", "0"))
+        while 1:
+            ws.send_text(dumps({"name":SERVICE_NAME, "builddate":builddate}))
+            sleep(15)
+    thread = Thread(target=run, args=[ws], daemon=True)
+    thread.start()
+
+def start():
+    print("substart")
+
+
+class WS:
+    ws: websocket.WebSocketApp
+    def __init__(self, backend, auth):
+        if (START_AFTER_CONNECT is None or SERVICE_NAME is None):
+            print("Setup START_AFTER_CONNECT and SERVICE_NAME")
+            sys.exit(1)
+        else:
+            print(f"Run after setup {SERVICE_NAME}: {START_AFTER_CONNECT}")
+            
+        addr = backend
+        if not "wss://" in addr:
+            addr = addr.replace("https://", "wss://")
+        if not "/ws/services" in addr:
+            addr += "/ws/services"
+
+        print(f"Connect to {addr}, secret len: {len(auth)}")
+        self.ws = websocket.WebSocketApp(addr, cookie="secretkey="+auth, on_close=self.on_close, on_open=self.on_open)
+
+    @staticmethod
+    def on_close(ws_self, status_code, msg):
+        print(f"WS Close, code:{status_code}, msg: {msg}")
+        sys.exit(0)
+
+    @staticmethod
+    def on_open(ws_self):
+        print("Connected")
+        create_pulser(ws_self)
+        if START_AFTER_CONNECT:
+            START_AFTER_CONNECT()
+
+    def run(self):
+        if self.ws.run_forever():
+            print("fuck")
+
+if __name__ == "__main__":
+    START_AFTER_CONNECT = start
+    SERVICE_NAME = "test"
+    wsc = WS("wss://tf2.pblr-nyk.pro/ws/services", "")
+    wsc.run()