dev_stat.py

# !/usr/bin/env python
#统计所有csv文件中设备信息，并导出为汇总表

from path import Path
import pandas as pd

class Stater():
  def __init__(self,base):
    self.base = Path(base)
    self.output_dir = Path('./outdir')
    self.output_dir.mkdir_p()
    self.keys  = ['dev_no','dev','details','owner']

    pass
  def __concat__(self,first_line=False):
    files = self.base.files('*.csv')
    files.sort()
    print("设备统计共计{}人".format(len(files)-1))
    csv_list = []
    for file in files:
      csv_item = pd.read_csv(file)
      # for idx,row in csv_item.iterrows():
      #   pass
      csv_item.loc[:, 'owner'] = file.stem
      csv_list .append(csv_item)

    df_total = pd.concat(csv_list,ignore_index=True)
    return df_total



  def dev_stat(self,isPrint=False):
    cols = ['dev','details']
    dev = ['mm','hdd','ssd','case','sc','gc']

    mm_dict = {'内存'}
    hdd_dict = {'机械硬盘'}
    ssd_dict = {'固态硬盘'}
    case_dict = {'机箱'}
    sc_dict = {'显示器'}
    gc_dict = {'显卡'}


    total = self.__concat__()
    total.sort_values(by = "dev",inplace=True)
    total = total[self.keys]
    if isPrint:
      for idx, row in total.iterrows():
        print("{}\n{} {}".format(idx, row[self.keys[0]],row[self.keys[1]]))
    total.to_csv(self.output_dir/"total.csv",index=False)
    print("汇总:{}".format(self.output_dir/"total.csv"))





    pass


def main():
  stater = Stater(base='./CSVs')
  stater.dev_stat()

  pass

if __name__ == "__main__":
  main()