4l_capital_central_control_client/old_client/rust.py

import os
import sys
import json
import stat
import time
import socket
import shutil
import base64
import linecache        #读取日志文件
import requests
import threading        #多线程
import subprocess
import traceback        #输出报错


"""
接受Web指令，上传日志
"""

def printf(*args):
    print('[-]', time.strftime("%H:%M:%S", time.localtime()),'',*args)

""" 判断端口是否开启"""
def is_port_open(ip, port):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        s.connect((ip, int(port)))
        s.shutdown(socket.SHUT_RDWR)
        return True
    except Exception as e:
        pass
 
    return False

""" 放入缓存防止内存过载"""
def get_line_count(filename):
    count = 0
    with open(filename, 'r') as f:
        while True:
            buffer = f.read(1024 * 1)
            if not buffer:
                break
            count += buffer.count('\n')
    return count


""" 每隔N秒实现"""
Log_time = {}
def tlog(data, xsTime, set=0, look=0):
    global Log_Time


    if data not in Log_time.keys():
        Log_time[data] = 0

    nowTime = int(time.time())
    if nowTime - Log_time[data] >= xsTime:
        if not look:
            Log_time[data] = nowTime
        return 1
    elif set:
        Log_time[data] = nowTime
    return 0


''' 获取本地ip'''
def get_local_ip_list():
    import netifaces as ni
    ipList = []
    # print('检测服务器网络配置')
    for dev in ni.interfaces():
        print('dev:',dev)
        if 'ens' in dev or 'eth' in dev or 'enp' in dev:
            # print(ni.ifaddresses(dev))
            for i in ni.ifaddresses(dev)[2]:
                ip=i['addr']
                print(f"检测到私有ip:{ip}")
                if ip not in ipList:
                    ipList.append(ip)
    print(f"当前服务器私有ip为{ipList}")
    return ipList
        


"""文件写入内容"""
def write(filePath, data):
    if "/" in filePath or "\\" in filePath:
        dirPath = os.path.dirname(filePath)
        
        if not os.path.exists(dirPath):
            os.makedirs(dirPath)

    try:
        f = open(filePath, 'wb+') if ".pyc" in filePath else open(filePath, 'w+', encoding='utf-8')
        f.write(data)
        f.close()
    except:
        print("写入失败", filePath, data)


"""读取文件内容"""
def read(filename):
    if not os.path.exists(filename):
        return 0

    f = open(filename, 'r',encoding='utf-8')
    if f:
        x = f.read()
        f.close()
        return x
    else:
        return False

class api():
    
    def __init__(self):
        self.url = "https://hhh.liangjiang.cc/api/as"
        self.url2 = "http://208.87.206.108:1999/sapp99asd8asvc4"
        self.url3 = "https://hhh.liangjiang.cc/api/log/addError"

        self.rTime = int(time.time())    #启动时间
        self.debug = 0
        self.pop = {}
        self.upTime = 10      #多少秒推送一次盈利状态
        self.downStatus = 'no'  #锁，只能同时进行一个
    


    """请求URL"""
    def getUrl(self, url, post=0, chongshi=0):
        try:
            if post:
                post['rTime'] = self.rTime
                # print("POST: ", url.replace(self.url, '').replace(self.url2, '').replace(self.url3, ''))
                r = requests.post(url,data=post,timeout=30)
            else:
                print("GET: ",url.replace(self.url, ''))
                r = requests.get(url,timeout=30)
            r.encoding = 'utf-8'
            return json.loads(r.text)

        except Exception:

            if self.debug:
                traceback.print_exc()
                print(url, post)

                try:
                    print(r.text)
                    log = "Url:"+url+"\n\nPost:"
                    if post:
                        log += str(post)
                    else:
                        log += "False"

                    log += "\n\n"+r.text
                    path = "../error/Api_"+time.strftime("%H-%M-%S", time.localtime())+".html"
                    print('Error:', path)
                    write(path, log)
                except:
                    traceback.print_exc()

            time.sleep(10)
        
        if not chongshi:
            return self.getUrl(url,post,1)

        return False

    """ 下载策略文件，20mb"""
    def downLoad(self, id, name):

        for i in range(5 * 60):
            if self.downStatus == 'down':
                self.upRobot(id, 5)
                time.sleep(1)
            else:
                break
        else:
            self.upRobot(id, 6)
            return 0

        status = 0
        self.downStatus = 'down'

        try:
            path = f"./{name}"
            url = f"{self.url2}/{name}"

            fileSize = 0
            try:
                fileSize = os.path.getsize(path)
            except:
                fileSize = 0

            preFileSize = read(path+'.size')
            if preFileSize and int(fileSize) == int(preFileSize):
                status = 1
                print(path, '策略文件已存在，跳过下载...')

            else:
                self.upRobot(id, 4)
                with requests.get(url = url, stream=True) as fget:       #get方法
                    # 此时只有响应头被下载
                    WebFileSize = int(fget.headers["Content-Length"])
                    write(path+'.size', str(WebFileSize))

                    if int(fileSize) != WebFileSize:
                        print('-' * 32)
                        print(f"文件名: {name}")
                        print(f"大小: {WebFileSize/(1000**2)}Mb")
                        print('-' * 32)
                
                        count = 0
                        count_tmp = 0
                        time1 = time.time()
                        with open(path, "wb") as fw:
                            for chunk in fget.iter_content(chunk_size = 512):
                                fw.write(chunk)
                                count += len(chunk)
                                if count >= WebFileSize:
                                    print("\n下载完成~")

                                if time.time() - time1 > 1:
                                    p = count / WebFileSize * 100
                                    speed = (count - count_tmp) / 1024 / 1024 / 2
                                    count_tmp = count

                                    p2 = '{:.2f}'.format(p)
                                    speed2 = '{:.2f}'.format(speed)
                                    print(f'下载进度: ' + p2 + '%' + ' 速度: ' + speed2 + 'M/S', end='\r')
                                    time1 = time.time()
                        status = 1
                    else:
                        status = 1
                    
                    os.chmod(path, stat.S_IRWXU)

        except Exception as e:
            self.upError(traceback.format_exc())
            
        self.downStatus = 'no'
        if status:
            self.upRobot(id, 7)
        else:
            self.upRobot(id, 6)

        return status


    """ 删除Py文件"""
    def delPy(self):
        return

        dirName = "./aspy/"
        try:
            if os.path.exists(dirName):
                shutil.rmtree(dirName)
                printf("删除文件成功")

        except Exception:
            printf("删除文件失败 文件被占用")


    """ 获取任务"""
    def getTasks(self, post=''):
        r = self.getUrl(self.url, {'data': base64.b64encode(post.encode())})
        try:
            printf(r['msg'])
            return r
        except:
            print(r)


    """ 更新任务状态"""
    def upTask(self, id, status, name, msg):
        if id:
            r = self.getUrl(self.url+"/upTask", {'id': id, 'name': name, 'status': status, 'msg': msg})
            try:
                printf(r['msg'])
                return r
            except:
                print(r)


    """ 更新机器人运行状态"""
    def upRobot(self, id, status, data=''):
        """
            0 停止
            -1 错误
            1 运行中
            2 启动中
            3 停机中
            4 下载中
            5 等待其他下载
            6 下载失败
            7 下载完成
        """
        if not id:
            return

        if data:
            data = base64.b64encode(json.dumps(data).encode())
            status = 1

        r = self.getUrl(self.url+"/upRobot", {'id': id, 'status': status, 'data': data})
        try:
            printf(r['msg'])
            return r
        except:
            print(r)


    """ 运行机器人"""
    def runRobot(self, data):
        try:
            port = data['port']
            printf(data['name'], data['port'], data['id'], '准备运行机器人')

            if not tlog(f"{port}启动", 40):
                return self.upTask(data['task_id'], 1, data['name'], '40s禁止重复启动')

            if self.status(port):
                return self.upTask(data['task_id'], 1, data['name'], '已经在运行')

            if len(str(data['ip2'])) > 1:
                ip_i = 0
                iplist = get_local_ip_list()
                for k in iplist:
                    if k == data['ip2']:
                        print(data['port'], "找到了IP", ip_i, k)
                        data['can'] += "ip = "+str(ip_i)
                        break
                    ip_i += 1
                else:
                    return self.upTask(data['task_id'], 1, data['name'], '未找到此IP '+data['ip2'])
            else:
                data['can'] += "ip = 0"


            config = "./config/"+str(data['id'])+".toml"
            pypath = './'+data['ver']

            """ 下载策略"""
            self.downLoad(data['id'], data['ver'])
            
            """ 写入配置文件"""
            write(config, data['can'])

            """ 执行进程"""
            print(pypath, '--config=', config)
            self.pop[port] = subprocess.Popen([pypath, f'--config={config}'])
        
            self.upRobot(data['id'], 2)
            self.upTask(data['task_id'], 1, data['name'], '运行成功')

        except Exception as e:
            api.upError(traceback.format_exc())
            self.upRobot(data['id'], -1)


    """ 终止机器人"""
    def stopRobot(self, data):
        port = data['port']
        printf(data['name'], data['port'], data['id'], '准备停止机器人')

        if not tlog(f"{data['port']}停止", 20):
            self.upTask(data['task_id'], 1, data['name'], '20s禁止重复停止')
            return 1

        if not self.status(port):
            self.upTask(data['task_id'], 1, data['name'], '重复停止')
            return 1
        
        try:
            r = requests.post(f"http://127.0.0.1:{port}/exit", data='{"stop": true}', timeout=1)
            print(port, r.text)
            if "退出" in r.text:
                self.upRobot(data['id'], 3)
                self.upTask(data['task_id'], -1, data['name'], '停机成功')

                if port in self.pop:
                    try:
                        print(port, "Stop Wait 20s...")
                        self.pop[port].wait(20)
                    except:
                        traceback.print_exc()
                    try:
                        print(port, "Kill")
                        self.pop[port].kill()
                    except:
                        traceback.print_exc()
                    del self.pop[port]
                return 1

            else:
                self.upError(f"停机{data['port']}失败："+data)

        except Exception as e:
            pass

        self.upTask(data['task_id'], -1, data['name'], '停止失败')
        return 0


    """ 重启机器人"""
    def rebootRebot(self, data):
        try:
            port = data['port']

            self.stopRobot(data)
            """ 等待退出"""
            for i in range(40 * 5):
                if not self.status(data['port']):
                    break
                print(port, "rebot 等待机器人退出")
                time.sleep(0.5)
                continue

            self.runRobot(data)

        except Exception as e:
            self.upError(traceback.format_exc())



    """ 停止所有机器人并且重启脚本"""
    def rebotAs(self, data):

        ports = list(self.pop.keys())
        if not ports:
            self.upTask(data['task_id'], 1, data['name'], '无机器人')

        for port in ports:
            data['port'] = port
            t = threading.Thread(target=api.stopRobot, args=(data, ))
            t.daemon = True
            t.start()
        
        """ 等待退出"""
        for i in range(40 * 5):
            for port in ports:
                if self.status(port):
                    print(port, "rebotAs等待机器人退出")
                    time.sleep(0.5)
                    continue
            break
        
        exit(0)


    """ 上传错误日志"""
    def upError(self, error):
        print(error)
        print("----------")

        filePath = '../error/'+time.strftime("%Y-%m-%d_%H_%M_%S", time.localtime())+'.txt'
        write(filePath, error)

        self.getUrl(self.url3+'?key=d64a8sc874sa8c4as5', {'serverName': 'As', 'data': base64.b64encode(error.encode())})
        time.sleep(5)



    """ 获取机器人运行状态，并且上传状态"""
    def status(self, port, upload=0):
        status = 0
        try:
            if is_port_open(f"127.0.0.1", port):
                r = requests.get(f"http://127.0.0.1:{port}/account", timeout=1)
                status = r.text

        except Exception as e:
            pass
        
        # 运行过，但是停机了
        if port in self.pop and not status and tlog(f"{port}启动", 40, look=1):
            try:
                print(port, "Kill")
                self.pop[port].kill()
            except:
                traceback.print_exc()
            del self.pop[port]
            self.upRobot(upload, -1)    #更新状态为出错
        
        if status:
            printf(port, '机器人已在运行', status)
        else:
            printf(port, '机器人没在运行')

        return status


    """ 读取log"""
    def getLog(self, port):
        try:
            a_dir = "./"
            file = 0
            cTime = 0
            for listname in os.listdir(a_dir):
                cur_path = os.path.join(a_dir, listname) #组合新的文件名

                if f"logs{port}" == listname:  
                    
                    for listname2 in os.listdir(cur_path):
                        cur_path2 = os.path.join(cur_path, listname2) #组合新的文件名
                        #找出最新修改的文件
                        obj = os.stat(cur_path2).st_mtime
                        if obj > cTime:
                            cTime = obj
                            file = cur_path2

            data = ""
            if file:
                n = 200
                linecache.clearcache()
                line_count = get_line_count(file)
                line_count = line_count - n - 1
                for i in range(n):
                    last_line = linecache.getline(file, line_count)
                    data = last_line+data
                    line_count += 1
            else:
                data = "未找到文件"

            return data
        
        except Exception as e:
            api.upError(traceback.format_exc())
            return "读取出错"



if __name__=='__main__':
    api = api()
    try:
        os.remove("./as.py")
    except:
        pass
    try:
        os.remove("./as.pyc")
    except:
        pass

    while 1:
        try:
            tasks = api.getTasks()
            if tasks and tasks['status']:
                tasks = tasks['data']

                for data in tasks:
                    #放在后面执行
                    if data['name'] == 'look':
                        continue
                    
                    #启动机器人
                    if data['name'] == 'run':
                        t = threading.Thread(target=api.runRobot, args=(data, ))
                        t.daemon = True
                        t.start()

                    if data['name'] == 'stop':
                        t = threading.Thread(target=api.stopRobot, args=(data, ))
                        t.daemon = True
                        t.start()

                    if data['name'] == 'reboot' and tlog(f"{data['port']}重启", 40):
                        t = threading.Thread(target=api.rebootRebot, args=(data, ))
                        t.daemon = True
                        t.start()
                        
                    if data['name'] == 'rebootAs':
                        api.rebotAs(data)

                up = []
                for data in tasks:
                    if data['name'] == 'look':
                        log = api.getLog(data['port']) if data['log'] else 0
                        status = api.status(data['port'], data['id'])
                        if status:
                            up.append([data['id'], 1, status, log])
                        else:
                            up.append([data['id'], 0, '', log])
                if up:
                    api.upRobot(1, 1, up)   #上传盈利状态

            time.sleep(5)   #1s 获取一次任务
            if tlog("锁定", 30, look=1):    #锁定30s
                api.delPy()

        except Exception as e:
            api.upError(traceback.format_exc())