def load_metric()

in aiops/MicroAgents/loader/nezha.py [0:0]

• 28 lines of code
• 13 McCabe index (conditional complexity)

    def load_metric(self, folder_path, date_str, queries, ano_folder):
        if (ano_folder): #Metrics are copied to both ano_folder and normal_data folder
            return
        file_pattern = os.path.join(folder_path, "*.csv")
        for file in glob.glob(file_pattern):
            try:
                system_name = "Error"
                if date_str == "2023-01-29" or date_str =="2023-01-30":
                    system_name = "TrainTicket"
                elif  date_str == "2022-08-23" or date_str =="2022-08-22":
                    system_name = "GShop"

                file_name = os.path.basename(file)
                q = pl.scan_csv(file, has_header=True, infer_schema_length=0, separator=",")
                q = q.with_columns(
                    pl.lit(date_str).alias('date_folder'),  # Folder is date info
                    pl.lit(os.path.basename(file)).alias('metric_file_name'),  # File name storage
                    pl.lit(system_name).alias('system_name')
                )
                if "source_50.csv" in file_name or "destination_50.csv" in file_name:
                    queries['src_dst'].append(q)   
                elif "front_service.csv" in file_name:
                    queries['front_srv'].append(q)
                elif "dependency.csv" in file_name:
                    queries['dep'].append(q)
                else:
                    queries['default'].append(q)
            except pl.exceptions.NoDataError:
                continue