From eadb8149ffb016474f8dfb7eb0207acad466b827 Mon Sep 17 00:00:00 2001
From: ALIHAN DIKEL <alihan.dikel@turkcell.com.tr>
Date: Tue, 29 Oct 2024 23:34:37 +0300
Subject: [PATCH] autocommit

---
 scripts/manage_server.sh          | 2 +-
 src/server/routers/router_main.py | 2 ++
 src/server/workers.py             | 4 ++--
 3 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/scripts/manage_server.sh b/scripts/manage_server.sh
index 4bf7132..f908f00 100755
--- a/scripts/manage_server.sh
+++ b/scripts/manage_server.sh
@@ -34,7 +34,7 @@ case "$COMMAND" in
         ;;
     logs)
         echo -e "\033[94m[INFO]\033[0m Reading $SERVICE_NAME logs..."
-        ssh ${REMOTE_USER}@${REMOTE_HOST} "tail -f -n 100 /home/uad/tello-commander/src/server/supervisor/tello-server.out.log"
+        ssh ${REMOTE_USER}@${REMOTE_HOST} "tail -f -n 100 /home/uad/tello-commander/src/server/supervisor/tello-server.out.log | sed -e 's/ERROR/\x1b[31m&\x1b[0m/g' -e 's/WARN/\x1b[33m&\x1b[0m/g' -e 's/INFO/\x1b[94m&\x1b[0m/g'"
         ;;
     *)
         echo -e "\033[31m[ERROR]\033[0m Invalid command!"
diff --git a/src/server/routers/router_main.py b/src/server/routers/router_main.py
index bf9ec73..d90a284 100644
--- a/src/server/routers/router_main.py
+++ b/src/server/routers/router_main.py
@@ -2,6 +2,7 @@ import os
 
 from fastapi import Request, APIRouter
 from fastapi.responses import HTMLResponse
+from loguru import logger
 from starlette.templating import Jinja2Templates
 
 
@@ -26,6 +27,7 @@ async def webui(request: Request):
 
 @router.get("/reconnect")
 async def reconnect(request: Request):
+    logger.debug("Reconnecting to tello")
     request.app.state.shared_state["connection"] = "NOK"
     return {"message": "connection state set to NOK"}
 
diff --git a/src/server/workers.py b/src/server/workers.py
index 15e1f0f..1d17b64 100644
--- a/src/server/workers.py
+++ b/src/server/workers.py
@@ -54,7 +54,7 @@ async def collect_flight_stats(state):
             state["command"] = True
         # Continuously collect stats and update shared state
         if state["connection"] not in ("genesis", "NOK"):
-            logger.debug(f"Connection state: {state['connection']}")
+            #logger.debug(f"Connection state: {state['connection']}")
             await asyncio.sleep(0.2)
             try:
                 bat = await asyncio.to_thread(tello.get_battery)
@@ -67,7 +67,7 @@ async def collect_flight_stats(state):
                 logger.error(te)
         # search for any response if disconnected
         else:
-            if counter % 10 == 0:
+            if counter % 30 == 0:
                 logger.warning("skipping stats collection")
         counter += 1
         await asyncio.sleep(1)