python 小工具合集

测代码执行时间

import time
from functools import wraps
def fn_timer(function):
     @wraps(function)
     def function_timer(*args, **kwargs):
         t0 = time.time()
         result = function(*args, **kwargs)
         t1 = time.time()
         print("Total time running %s: %s seconds" %
             (function.__name__, str(t1-t0))
             )
         return result
     return function_timer

@fn_timer
def my_func():

解压tar包，zip包

def untar(file_name,dirs,id):
    tar = tarfile.open(file_name)
    tar.extractall(dirs)
    tar.close()

def untar(file_name, dirs, id):
    zip_ref = zipfile.ZipFile(file_name)  # 创建zip 对象
    os.mkdir(file_name.replace(".zip", ""))  # 创建同名子文件夹
    zip_ref.extractall(file_name.replace(".zip", ""))  # 解压zip文件内容到子文件夹
    zip_ref.close()

检查文件夹内文件名

def searchDirFile(rootDir, fileType, tar_flag):
    flag = 1
    for dir_or_file in os.listdir(rootDir):
        if dir_or_file == fileType:
            print('find file '+fileType)
            flag = 0
        else:
            continue
    tar_flag += flag
    return tar_flag, flag

停止线程

import inspect
import ctypes
def _async_raise(tid, exctype):
    """raises the exception, performs cleanup if needed"""
    tid = ctypes.c_long(tid)
    if not inspect.isclass(exctype):
        exctype = type(exctype)
    res = ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, ctypes.py_object(exctype))
    if res == 0:
        raise ValueError("invalid thread id")
    elif res != 1:
        # """if it returns a number greater than one, you're in trouble,
        # and you should call it again with exc=NULL to revert the effect"""
        ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, None)
        raise SystemError("PyThreadState_SetAsyncExc failed")
def stop_thread(thread):
    _async_raise(thread.ident, SystemExit)