ida-modbus-reader/app.py at main · tawanmeow/ida-modbus-reader · 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
# Tawannnnnnnn :)
# modbus_ida - app.py
# Webapp update checker.

import os
import time
import stat
import shutil


CURRENT_DIRECTORY = os.path.dirname(os.path.abspath(__file__))

VERSION = os.path.join(CURRENT_DIRECTORY, "version.txt")
WEBAPP_FOLDER = os.path.join(CURRENT_DIRECTORY, "webapp", "")
MODBUS_IDA = os.path.join(CURRENT_DIRECTORY, "ida-modbus-reader", "")
APPFILE = os.path.join(CURRENT_DIRECTORY, "webapp", "webapp.py")
GIT_FOLDER = os.path.join(CURRENT_DIRECTORY, "ida-modbus-reader", ".git", "")
WEBAPP_TEMP = os.path.join(CURRENT_DIRECTORY, "ida-modbus-reader", "webapp", "")
REQ = os.path.join(CURRENT_DIRECTORY, "webapp" ,"req", "requirements.txt")

# pip install.
def pipUpdate():
    try:
        os.system('pip install -r ' + REQ)
    except:
        pass

# Test connection between device and github using ICMP.
def githubChecker():
    import pyping
    githubLoopChecker = True
    while githubLoopChecker == True:
        try:
            r = pyping.ping('www.github.com')
            if r.ret_code == 0:
                githubLoopChecker = False
            else:
                pass

        except:
            time.sleep(5)

# Check for update fucntion.
def updateChecker():
    import git
    import requests
    try:
        #response = requests.get('https://raw.githubusercontent.com/idaplatform/modbus_ida/master/version.txt')
        response = requests.get('https://raw.githubusercontent.com/tawanmeow/ida-modbus-reader/master/version.txt')
        # Note: data = version.txt on github | txt = version.txt on your device.
        data = response.text
        data = str(data)
        data = data[:-1]
        f = open(VERSION, "r")
        txt = f.read()
        f.close()
        txt = txt[:-1]
        if data != txt:
            # chmod 7xx
            for root, dirs, files in os.walk(GIT_FOLDER):
                for dir in dirs:
                    os.chmod(os.path.join(root, dir), stat.S_IRWXU)
                for file in files:
                    os.chmod(os.path.join(root, file), stat.S_IRWXU)
            # Remove old directories then clone repo from github.
            shutil.rmtree(MODBUS_IDA, ignore_errors=True)
            shutil.rmtree(WEBAPP_FOLDER, ignore_errors=True)
            os.mkdir(WEBAPP_FOLDER)
            git.Git(CURRENT_DIRECTORY).clone("https://github.com/tawanmeow/ida-modbus-reader.git")
            # Replace old version.txt with new version.txt
            os.remove(VERSION)
            shutil.move(os.path.join(MODBUS_IDA, "version.txt"), CURRENT_DIRECTORY)
            for file in os.listdir(WEBAPP_TEMP):
                shutil.move(os.path.join(WEBAPP_TEMP, file), WEBAPP_FOLDER)
    except:
        pass

if __name__ == '__main__':
    #pipUpdate()
    #githubChecker()
    #updateChecker()
    os.system('python '+ APPFILE)