import csv


class HandleCsv:
  '''
  csv文件處理類
  '''
  def __init__(self, filename):
    '''
    構(gòu)造器
    :param filename: csv文件名
    '''
    self.filename = filename

  def get_data(self):
    '''
    獲取csv中所有數(shù)據(jù)
    :return: 嵌套字典的列表
    '''
    with open(self.filename, mode='r', encoding='utf-8') as f:
      cb = csv.reader(f) # 實例化reader對象
      header = next(cb) # 獲取表頭，并將指針轉(zhuǎn)向下一行
      list_dict = []
      for row in cb:
        list_dict.append(dict(zip(header, row)))
    return list_dict

  def get_one_row(self, row):
    '''
    獲取單行數(shù)據(jù)
    :param row: 指定的行號
    :return: 對應(yīng)行號的數(shù)據(jù)
    '''
    return self.get_data()[row - 1]

  def write_csv(self, headers, values, data_type, mode='w'):
    '''
    寫入數(shù)據(jù)到csv到文件中
    :param headers: 表頭：列表類型
    :param values: 表數(shù)據(jù)：1.嵌套元組的列表；2.嵌套字典的列表
    :param data_type: 傳入的數(shù)據(jù)類型：1.'tuple'；2.'dict'
    :param mode: 寫入方式，默認(rèn)寫入“w”
    :return:
    '''
    with open(file=self.filename, mode=mode, encoding='utf-8', newline='') as f:
      if data_type == 'tuple':
        writer = csv.writer(f) # 實例化writer對象
        writer.writerow(headers) # 寫入表頭
        writer.writerows(values) # 寫入數(shù)據(jù)
      elif data_type == 'dict':
        writer = csv.DictWriter(f, headers) # 實例化DictWriter對象
        writer.writeheader() # 寫入表頭
        writer.writerows(values) # 寫入數(shù)據(jù)
      else:
        print("數(shù)據(jù)類型錯誤，請確認(rèn)！")