#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
 @Time : 2018/10/24
 @Author : LiuXueWen
 @Site : 
 @File : transfer.py
 @Software: PyCharm
 @Description: 
"""

import os
import traceback
import threading
from multiprocessing import Pool
from multiprocessing.dummy import Pool as ThreadPool


# 兼容python2.7上多線程的bug，不加上下面的反代理程序不能正常執(zhí)行
def proxy(cls_instance, i):
 return cls_instance.multiprocess_thread(i)
def proxy2(cls_instance, i):
 return cls_instance.file_operation(i)

class file2transfer():
 # 多進(jìn)程執(zhí)行程序
 def multiprocessingTransferFiles(self):
  try:
   # 創(chuàng)建進(jìn)程池
   p = Pool()
   //參數(shù)末尾必須加上逗號
   p.apply_async(proxy, args=(self, self.root_path,))
   p.close()
   p.join()
  except Exception as e:
   print(e)

 # 每個(gè)進(jìn)程下的多線程執(zhí)行，線程數(shù)等于當(dāng)前機(jī)器的核數(shù)
 def multiprocess_thread(self, root_path):
  try:
   # 創(chuàng)建線程鎖
   lock = threading.RLock()
   lock.acquire()
   # 獲取每個(gè)文件
   for pfile in os.listdir(root_path):
    # 獲取文件的完整路徑
    full_file_path = os.path.join(root_path, pfile)
    # 多線程讀寫文件
    p = ThreadPool()
    # 執(zhí)行線程
    p.apply_async(proxy2, args=(self, full_file_path,))
    p.close()
    p.join()
  except Exception as e:
   print(e)
  finally:
   # 釋放線程鎖
   lock.release()

 # 對每個(gè)文件夾下的每個(gè)文件進(jìn)行操作
 def file_operation(self, full_file_path):
  try:
   // TODO 真正需要單獨(dú)執(zhí)行的操作
   pass
  except Exception as e:
   print(e)