Network Speed Test

# -*- coding:utf-8 -*-


from TST.FileBox import *
from TST.NetOCR import *

import TLog
import TPeripheral
import Image
import time


# 声明变量
remote = TPeripheral.get_remote()
capture = TPeripheral.get_video_capture()


# 释放资源
def close_peripheral():
    try:
        remote.close()
        capture.close()
    except Exception as e:
        TLog.record_log("close_peripheral()")
        TLog.f("释放外设资源时发生异常(Exception: %s)!" % e)


def only_num(s):
    refer = '0123456789'  # 只显示数字
    for c in s:
        if not c in refer:
            s = s.replace(c, '')

    if s == '':
        TLog.r("NetOCR Works Abnormal")
        return 0
    else:
        return s


def network_speed_check():
    capture.takePicture(FileBox("NetWork Speed Check.bmp"), 1)

    image = Image.open('D:/FileBox/NetWork Speed Check.bmp')
    box = (660, 390, 732, 420)
    region = image.crop(box)
    region.save(FileBox("Speed Check.bmp"))

    net_ocr = NetOCR("10.120.16.35", 3000, FileBox("Speed Check.bmp"), "English", 1)
    ocr_string = net_ocr.getStr()
    TLog.i("识别的字符：%s" % ocr_string)

    return only_num(ocr_string)


def main():
    time1 = time.time()
    time_run = 0
    while time_run < 30:
        time2 = time.time()
        time_run = time2 - time1
        speed_s = network_speed_check()
        speed = int(speed_s)
        TLog.i("Network Speed：%s" % speed)
        if speed == 0:
            continue
        if speed < 10000:
            TLog.i("Network Speed1")
            img = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())
            capture.takePicture("D:\speed1\%s.bmp" % img)
            time.sleep(10)
        if 10000 <= speed < 20000:
            TLog.i("Network Speed2")
            img = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())
            capture.takePicture("D:\speed2\%s.bmp" % img)
            time.sleep(10)
        if 20000 <= speed < 30000:
            TLog.i("Network Speed3")
            img = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())
            capture.takePicture("D:\speed3\%s.bmp" % img)
            time.sleep(10)
        if 30000 <= speed < 40000:
            TLog.i("Network Speed4")
            img = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())
            capture.takePicture("D:\speed4\%s.bmp" % img)
            time.sleep(10)


if __name__ == "__main__":
    try:
        main()
    except Exception as ex:
        TLog.record_log()
        TLog.f("主函数发生异常(Exception: %s)!" % ex)
    else:
        TLog.test_end()
    finally:
        close_peripheral()
        TLog.test_result()