def get_file_size(file_path, KB=False, MB=False):
  """获取文件大小"""
  size = os.path.getsize(file_path)
  if KB:
    size = round(size / 1024, 2)
  elif MB:
    size = round(size / 1024 * 1024, 2)
  else:
    size = size
  return size

def list_files(root_dir):
  """遍历文件"""
  if os.path.isfile(root_dir): # 如果是文件
    size = get_file_size(root_dir, KB=True)
    file_dict[root_dir] = size
  else:
    # 如果是文件夹，则遍历
    for f in os.listdir(root_dir):
      # 拼接路径
      file_path = os.path.join(root_dir, f)
      if os.path.isfile(file_path):
        # 如果是一个文件
        size = get_file_size(file_path, KB=True)
        file_dict[file_path] = size
      else:
        list_files(file_path)

if __name__ == '__main__':
  list_files(root_dir)
  # print( len(file_dict))
  # 计算文件目录大小
  total_size = 0
  # 遍历字典的key
  for file in file_dict:
    total_size += file_dict[file]

  print('total size is : %.2f' % total_size)
  # 找最大最小文件
  max_file = max(file_dict, key=lambda x: file_dict[x])
  min_file = min(file_dict, key=lambda x: file_dict[x])
  print('max file is : ', max_file, '\n file size is :', file_dict[max_file])
  print('min file is : ', min_file, '\n file size is :', file_dict[min_file])

def read_book():
  """读取内容"""
  # 读取一个文件
  file1 = open('book1.txt', 'r', encoding='UTF-8')
  lines1 = file1.readlines()
  file1.close()
  for line in lines1:
    line = line.strip() # 去空白
    content = line.split(',')
    book1[content[0]] = content[1]

  # 另一种方式，读取另一个文件,不需要close，会自动关闭
  with open('book2.txt', 'r', encoding='UTF-8') as file2:
    lines2 = file2.readlines()
  for line in lines2:
    line = line.strip() # 去空白
    content = line.split(',')
    book2[content[0]] = content[1]

def merge_book():
  """合并内容"""
  lines = [] # 定义一个空列表
  header = '姓名\t 电话\t 文本\n'
  lines.append(header)
  # 遍历第一个字典
  for key in book1:
    line = ''
    if key in book2.keys():
      line = line + '\t'.join([key, book1[key], book2[key]])
      line += '\n'
    else:
      line = line + '\t'.join([key, book1[key], ' *****'])
      line += '\n'
    lines.append(line)
  # 遍历第2个，将不包含在第1个里面的写入
  for key in book2:
    line = ''
    if key not in book1.keys():
      line = line + '\t'.join([key, ' *****', book2[key]])
      line += '\n'
      lines.append(line)
  # 写入book3
  with open('book3.txt', 'w', encoding='UTF-8') as f:
    f.writelines(lines)

if __name__ == '__main__':
  # 读取内容
  read_book()
  # 合并内容
  merge_book()

  # print(book1)
  # print(book2)