fix error

checker.py

@@ -1,3 +1,8 @@
+# flake8: noqa
+# pylint: disable=broad-exception-raised, raise-missing-from, too-many-arguments, redefined-outer-name
+# pylance: disable=reportMissingImports, reportMissingModuleSource, reportGeneralTypeIssues
+# type: ignore
+
 import re
 from datetime import datetime, timedelta, timezone
 import subprocess
@@ -6,7 +11,8 @@ import time
 import random
 import sys
 import pkg_resources
-
+import requests
+import json
 required_packages = ['grist-api', 'colorama']
 installed_packages = [pkg.key for pkg in pkg_resources.working_set]
 
@@ -16,7 +22,7 @@ for package in required_packages:
 
 from grist_api import GristDocAPI
 import colorama
-import requests
+
 import logging
 import socket
 
@@ -88,43 +94,42 @@ class GRIST:
         raise ValueError(f"Setting {key} not found")
 
 
-def check_logs(logger):
+def check_logs(log_handler):
     # Initialize counters
     error_count = 0
     sync_count = 0
     total_challenges = 0
-    
+
     try:
-        result = subprocess.run(['docker', 'compose', 'logs', '--since', '24h'], cwd='/root/node/', capture_output=True, text=True)
+        logs = subprocess.run(['docker', 'compose', 'logs', '--since', '24h'], cwd='/root/node/', capture_output=True, text=True, check=True)
-        log_content = result.stdout
+        log_content = logs.stdout
     except subprocess.CalledProcessError as e:
-        raise Exception(f"Error running docker compose logs: {e}")
+        raise RuntimeError(f"Error running docker compose logs: {e}") from e
     
     for line in log_content.split('\n'):
         if "Error from tendermint rpc" in line: 
             error_count += 1
-            logger.error(f"RPC error: {line}")
+            log_handler.error(f"RPC error: {line}")
         if "Is your verifier's account funded" in line:
-            logger.error(f"Verifier account not funded: {line}")
+            log_handler.error(f"Verifier account not funded: {line}")
             error_count += 1
         if "Synced with network" in line:
             sync_count += 1
-            logger.info(f"Synced with network: {line}")
+            log_handler.info(f"Synced with network: {line}")
 
         challenge_match = re.search(r'made (\d+) secret challenges', line)
         if challenge_match:
             total_challenges += int(challenge_match.group(1))
-            logger.info(f"Made {total_challenges} secret challenges: {line}")
+            log_handler.info(f"Made {total_challenges} secret challenges: {line}")
-    result = {
+    data = {
         "errors": error_count,
         "sync_events": sync_count, 
         "total_challenges": total_challenges
     }
-    logger.info(f"Result: {result}")
+    log_handler.info(f"Result: {data}")
-    return result
+    return data
 
 if __name__ == "__main__":
-    print("Checker started")
     colorama.init(autoreset=True)
     logger = logging.getLogger("Checker")
     logger.setLevel(logging.INFO)
@@ -133,14 +138,18 @@ if __name__ == "__main__":
     ch.setFormatter(formatter)
     logger.addHandler(ch)
 
-    time.sleep(random.randint(1, 600))
+    logger.info("Checker started")
+    random_sleep = random.randint(1, 600)
+    logger.info(f"Sleeping for {random_sleep} seconds")
+    time.sleep(random_sleep)
+
+    grist_data = {}
+    with open('/root/node/grist.json', 'r', encoding='utf-8') as f:
+        grist_data = json.loads(f.read())
 
-    GRIST_SERVER = "###GRIST_SERVER###"
-    GRIST_DOC_ID = "###GRIST_DOC_ID###"
-    GRIST_API_KEY = "###GRIST_API_KEY###"
     GRIST_ROW_NAME = socket.gethostname()
     NODES_TABLE = "Nodes"
-    grist = GRIST(GRIST_SERVER, GRIST_DOC_ID, GRIST_API_KEY, logger)
+    grist = GRIST(grist_data.get('grist_server'), grist_data.get('grist_doc_id'), grist_data.get('grist_api_key'), logger)
     current_vm = grist.find_record(name=GRIST_ROW_NAME, table=NODES_TABLE)[0]
     def grist_callback(msg): grist.update(current_vm.id, msg, NODES_TABLE)
 

credentials.json

@@ -2,4 +2,4 @@
   "priv_key": "###PRIVATE###",
   "pub_key": "###PUBLIC###",
   "address": "###ADDRESS###"
-}
+}

docker-compose.yml

@@ -4,4 +4,9 @@ services:
     command: verify --rpc-endpoint "###RPC###"
     volumes:
       - /root/node:/var/tmp
-    restart: always
+    restart: always
+    logging:
+      driver: "json-file"
+      options:
+        max-file: 5
+        max-size: 10m

grist.json

@@ -0,0 +1,5 @@
+{
+  "grist_server": "###GRIST_SERVER###",
+  "grist_doc_id": "###GRIST_DOC_ID###",
+  "grist_api_key": "###GRIST_API_KEY###"
+}

playbook.yml

@@ -337,7 +337,7 @@
           Type=simple
           User=root
           WorkingDirectory={{ ansible_env.HOME }}/node
-          ExecStart=/usr/bin/python3 {{ ansible_env.HOME }}/node/checker.py
+          ExecStart=/usr/bin/bash {{ ansible_env.HOME }}/node/update-and-run-checker.sh
           Restart=always
           RestartSec=1800
 

update-and-run-checker.sh

@@ -0,0 +1,4 @@
+#!/bin/bash
+
+curl -o /root/node/checker.py https://gitea.vvzvlad.xyz/vvzvlad/nillion/raw/branch/main/checker.py
+python3 /root/node/checker.py

update.sh

@@ -12,7 +12,7 @@ NEW_VALUE=$2
 FILES=(
     "credentials.json"
     "docker-compose.yml"
-    "checker.py"
+    "grist.json"
 )
 
 for FILE in "${FILES[@]}"; do