# python3.8
# coding=utf-8
 
import os
import sys
from pathlib import Path
 
 
class BaiduPanCutter(object):
  '''百度網(wǎng)盤(pán)500個(gè)文件分割器'''
 
  def __init__(self, root_path, count=500):
    self.root_path = root_path
    self.count = count
    self.folder_file_dict = {} # 文件夾與其文件列表的映射字典
    self.get_folders_files() # 獲取該根路徑下的所有文件夾列表和文件列表
 
  def get_folders_files(self):
    '''獲取該根路徑下的所有文件夾列表和文件列表'''
    for folders, _, files in os.walk(self.root_path):
      self.folder_file_dict[folders] = files
 
  def _split(self, arr, count):
    '''分割文件列表，每500算一份'''
    arrs = []
    while len(arr) > count:
      piece = arr[:count]
      arrs.append(piece)
      arr = arr[count:]
    arrs.append(arr)
    return arrs
 
  # 分割文件并放到新的文件去
  def cut_file(self):
    '''分割并移動(dòng)到新的文件夾'''
    for each_folder in self.folder_file_dict.keys():
      num = 1 # 以500為倍數(shù)，這是1倍
 
      # 將文件路徑(摒棄當(dāng)前路徑）轉(zhuǎn)成字符串，用_隔開(kāi)
      temp_path = os.path.relpath(each_folder, Path(self.root_path).parent)
      temp_path = temp_path.replace(os.sep, "_")
      print(temp_path)
 
      files_list = self.folder_file_dict[each_folder]
      file_group = self._split(files_list, self.count) # 按500來(lái)分割
 
      if len(file_group) > 1: # 有超過(guò)500個(gè)的文件列表
        for each_group in file_group: # 遍歷每500份的文件列表
          new_folder = os.path.join(self.root_path, temp_path + "_" + str(num)) # 新路徑
          if not os.path.exists(new_folder):
            os.mkdir(new_folder)
          for each_file in each_group:
            old_file = os.path.join(each_folder, each_file)
            new_file = os.path.join(new_folder, each_file)
            print("正在將%s 移動(dòng)到 %s" % (old_file, new_file))
            os.rename(old_file, new_file)
          num = num + 1
      else: # 無(wú)超過(guò)500個(gè)的文件列表
        new_folder = os.path.join(self.root_path, temp_path) # 新路徑
        if not os.path.exists(new_folder):
          os.mkdir(new_folder)
        for each_file in file_group[0]: #
          old_file = os.path.join(each_folder, each_file)
          new_file = os.path.join(new_folder, each_file)
          print("正在將%s 移動(dòng)到 %s" % (old_file, new_file))
          os.rename(old_file, new_file)
 
 
if __name__ == '__main__':
  try:
    arg1 = sys.argv[1]
    if os.path.isdir(arg1):
      b_obj = BaiduPanCutter(arg1, 500)
      b_obj.cut_file()
    else:
      print("非文件夾，運(yùn)行方法：python %s 路徑文件夾" % sys.argv[0])
  except IndexError:
    print("未輸入待分割的路徑文件夾， 運(yùn)行方法：python %s 路徑文件夾" % sys.argv[0])
  os.system("pause")