fix error

This commit is contained in:
vvzvlad 2024-11-24 04:31:42 +03:00
parent 586193e60f
commit e072fc2082
4 changed files with 14 additions and 8 deletions

View File

@ -12,7 +12,7 @@ import random
import sys import sys
import pkg_resources import pkg_resources
import requests import requests
import json
required_packages = ['grist-api', 'colorama'] required_packages = ['grist-api', 'colorama']
installed_packages = [pkg.key for pkg in pkg_resources.working_set] installed_packages = [pkg.key for pkg in pkg_resources.working_set]
@ -130,7 +130,6 @@ def check_logs(log_handler):
return data return data
if __name__ == "__main__": if __name__ == "__main__":
print("Checker started")
colorama.init(autoreset=True) colorama.init(autoreset=True)
logger = logging.getLogger("Checker") logger = logging.getLogger("Checker")
logger.setLevel(logging.INFO) logger.setLevel(logging.INFO)
@ -139,16 +138,18 @@ if __name__ == "__main__":
ch.setFormatter(formatter) ch.setFormatter(formatter)
logger.addHandler(ch) logger.addHandler(ch)
logger.info("Checker started")
random_sleep = random.randint(1, 600) random_sleep = random.randint(1, 600)
logger.info(f"Sleeping for {random_sleep} seconds") logger.info(f"Sleeping for {random_sleep} seconds")
time.sleep(random_sleep) time.sleep(random_sleep)
GRIST_SERVER = "###GRIST_SERVER###" grist_data = {}
GRIST_DOC_ID = "###GRIST_DOC_ID###" with open('/root/node/grist.json', 'r', encoding='utf-8') as f:
GRIST_API_KEY = "###GRIST_API_KEY###" grist_data = json.loads(f.read())
GRIST_ROW_NAME = socket.gethostname() GRIST_ROW_NAME = socket.gethostname()
NODES_TABLE = "Nodes" 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] 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) def grist_callback(msg): grist.update(current_vm.id, msg, NODES_TABLE)

View File

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

5
grist.json Normal file
View File

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

View File

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