from multiprocessing import Pool # 進(jìn)程池，用于多進(jìn)程
import os # 用于獲取當(dāng)前執(zhí)行的文件名
import time
import traceback # 用于捕獲異常
import sys # 用于捕獲異常


def main_func(i): # 多進(jìn)程運(yùn)行的函數(shù)
  try: # 多進(jìn)程中發(fā)生異常是不會打印錯誤信息，并且當(dāng)前進(jìn)程會直接跳過，所以異常需要自行捕獲
    rand_time = random.randint(1, 3) # 隨機(jī)產(chǎn)生1~3的整數(shù)
    time.sleep(rand_time )
    print(i)
  except:
    current_filename = str(os.path.basename(sys.argv[0]))[:-3] # 獲取當(dāng)前文件名稱
    cur_err_filname = current_filename + '_error.txt'
    error_info = sys.exc_info() # 打印異常
    with open(f'{cur_err_filname }', 'a') as f:
      error_str = f'{i}:ERROR OCCURRED，{time.strftime("%Y-%m-%d %H:%M:%S")}：\n {error_info[0]}: {error_info[1]}' # 記錄當(dāng)前進(jìn)程特征值，錯誤發(fā)生時間 ，錯誤類型，錯誤概述
      print(error_str, file=f) # 通過打印方式寫入文件
      traceback.print_tb(error_info[2], file=f) # 錯誤細(xì)節(jié)描述（包括bug的代碼位置）
      f.write(f"{'=' * 50}\n") # 分行


if __name__ == '__main__': # 必須在此語句下面運(yùn)行，不然會報錯
  pool = Pool(5) # 定義進(jìn)程數(shù)量
  for i in range(20):
    pool.apply_async(main_func, (i, )) # 調(diào)用函數(shù)執(zhí)行多進(jìn)程
  pool.close() # 關(guān)閉進(jìn)程池
  pool.join() # 阻塞進(jìn)程，此兩部不能少，保證多進(jìn)程正常運(yùn)行