zoukankan      html  css  js  c++  java
  • python 多进程练习 调用 os.system命令

    import sys
    import getopt
    import os
    import multiprocessing
    
    
    def list_all_file(path):
        """
        list all files of a directory
        :param path:
        :return:
        """
        file_list = []
        for (path, _, files) in os.walk(path):
            for filename in files:
                file_list.append(os.path.join(path, filename))
        return file_list
    
    
    def process_file(filename, is_black):
        suffix = os.path.splitext(filename)[-1][1:]
        if suffix != "pcap":
            return
    
        if is_black:
            cmd = "python extract_tls_flow4.py -vr {} -o black/{}.txt  >logs/black/{}.log".format(filename, os.path.basename(filename), os.path.basename(filename))
        else:
            cmd = "python extract_tls_flow4.py -vr {} -o white/{}.txt  >logs/white/{}.log".format(filename, os.path.basename(filename), os.path.basename(filename))
        os.system(cmd)
    
    
    def process_black_file(filename):
        process_file(filename, 1)
    
    
    def process_white_file(filename):
        process_file(filename, 0)
    
    
    def process_dir(sample_dir, is_black):
        file_list = list_all_file(sample_dir)
    
        process_num = 30
        pool = multiprocessing.Pool(processes=process_num)
        if is_black:
            pool.map(process_black_file, file_list)
        else:
            pool.map(process_white_file, file_list)
    
        pool.close()
        pool.join()
        print("End...........")
    
    
    black_sample_dir = "/opt/data/samples/black_pcap"
    white_sample_dir = "/opt/data/samples/white_pcap"
    
    process_dir(black_sample_dir, 1)
    process_dir(white_sample_dir, 0)
    

      

  • 相关阅读:
    WordPress网站绑定多个域名的方法
    htpasswd 命令使用
    在Windows下用OpenSSL生成证书步骤
    WCF中关于List和数据的转换问题
    NET2.0的配置文件
    C# Attribute
    c#自定义属性
    VS2005中读写配置文件(方法二)
    c#的反射
    Asp.NET 操作配置文件 Steven Pei 博客园
  • 原文地址:https://www.cnblogs.com/bonelee/p/10409271.html
Copyright © 2011-2022 走看看