utils.py

import os, sys
import subprocess
from typing import List
import csv

ProjectName = "productions"


class Utils:

    @classmethod
    def get_root_path(cls):
        abs_path = os.path.dirname(os.path.abspath(__file__))
        return abs_path

    @classmethod
    def add_path(cls, root_path, add):
        """
        connect path
        :param root_path:
        :param add:
        :return:
        """
        res = os.path.join(root_path, add)
        return res

    @classmethod
    def test_online(cls, device: str):
        """
        ping device
        :param device: id address
        :return:
        """
        cmd = f"ping -n 1 {device}"
        try:
            result = subprocess.check_output(cmd, shell=True).decode('gbk')
            if 'ttl' in result.lower():
                return True
            else:
                return False
        except subprocess.CalledProcessError as e:
            print(e, f"{device} online")
            return False

    @classmethod
    def test_online2(cls, device):
        """
        test online
        :param device:
        :return:
        """
        import os
        forbidden = ['&', ';', '-', ' ', '||', '|']
        for i in forbidden:
            if i in device:
                print('Catch you')
                exit()
        result = os.popen(f'ping -c 1 -t 1 {device}').read()
        if 'ttl' in result:
            print('IP 在线')
        else:
            print('IP 不在线')

    @classmethod
    def test_devices_online(cls, devices: List[str]):
        """
        show online devices
        :param devices:
        :return:
        """
        print("============list devices=============")
        for device in devices:
            Utils.test_online(device)

    @classmethod
    def is_file_exist(cls, file_path: str):
        return os.path.exists(file_path)

    @classmethod
    def write_to_csv(cls, filename, row: list):
        with open(filename, 'a', newline='') as file:
            writer = csv.writer(file)
            writer.writerow(row)

    @classmethod
    def get_time_string(cls, format=False):
        from datetime import datetime
        # 方法一：获取当前时间并格式化为指定字符串格式
        if not format:
            current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        else:
            current_time = datetime.now().strftime("%Y-%m-%d %H-%M-%S")
        return current_time


if __name__ == '__main__':
    Utils.get_root_path()