NeoNode/watch_neo_node.py at master · trinity-project/NeoNode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
import time
import requests
import os
from project_log import setup_mylogger

NEOSCAN_GETBLOCKHEIGHT_URL = "https://api.neoscan.io/api/main_net/v1/get_height"
LOCAL_NEONODE_URL = "http://127.0.0.1:10332"
# LOCAL_NEONODE_URL = "http://seed3.ngd.network:10332"

logger = setup_mylogger(logfile="log/watch_neo_node.log")

ENVIRON=os.environ
NO_REPLY_EMAIL_PAWD = ENVIRON.get("EMAIL_PASSWORD")

def execute_shell_command(command):
    import os
    os.system(command)


def request_block_height_from_neoscan(url):
    res = requests.get(url)
    if res.status_code == 200:
        return res.json().get("height")
    return None

def request_block_height_from_local(url):
    data = {
      "jsonrpc": "2.0",
      "method": "getblockcount",
      "params": [],
      "id": 1
}
    res = requests.post(url,json=data)
    if res.status_code == 200:
        return res.json().get("result")
    return None

def send_email(toAddr, local_block_number, infura_block_number):
    from email.header import Header
    from email.mime.text import MIMEText
    import smtplib

    msg = MIMEText(
        ''''
        <!DOCTYPE html>
        <html>
        <head>
        </head>
        <body>
        <div id='preview-contents' class='note-content'>

        <h3 >title: restart neo node </h3>

        <p>local block number: {},neoscan block number:{}</p>

        </div>
        </body>
        </html>'''.format(local_block_number, infura_block_number), 'html', 'utf-8')

    msg['From'] = "no-reply@trinity.tech"
    msg['To'] = toAddr
    msg['Subject'] = Header('notification from WATCH NEO NODE.....')

    server = smtplib.SMTP_SSL("smtp.mxhichina.com", 465)  # SMTP协议默认端口是25
    server.login("no-reply@trinity.tech",NO_REPLY_EMAIL_PAWD )
    server.sendmail("no-reply@trinity.tech", [toAddr], msg.as_string())
    server.quit()


def compare_block_number():
    while True:
        try:
            neoscan_block_number = request_block_height_from_neoscan(NEOSCAN_GETBLOCKHEIGHT_URL)
            local_block_number = request_block_height_from_local(LOCAL_NEONODE_URL)
            logger.info("local_block_number:{},neoscan_block_number:{}".format(local_block_number, neoscan_block_number))
            if neoscan_block_number - local_block_number >= 20:
                execute_shell_command("supervisorctl restart NeoNode:dotnet")
                logger.warning("restart neo-cli")
                send_email("m17379352738@163.com", local_block_number, neoscan_block_number)
                time.sleep(10* 60)
        except:
            pass

        finally:
            time.sleep(120)


if __name__ == "__main__":
    compare_block_number()