def load_label()

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

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

    def load_label(self, file_path, date_str):
        label_data = pl.DataFrame()
        try:
            with open(file_path, 'r') as file:
                data = json.load(file) #Polars JSON reader cannot handel the non-standard json so using json.load
            # Iterate over each key in the JSON and extract the records
            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"

            for key in data:
                records = data[key]
                if records:  # Check if the list is not empty
                    df = pl.DataFrame(records)
                    df = df.with_columns(
                        pl.lit(date_str).alias('date_folder'),
                        pl.lit(os.path.basename(file_path)).alias('label_file_name'),
                        pl.lit(system_name).alias('system_name')
                    )
                    label_data = label_data.vstack(df)
        except json.JSONDecodeError as e:
            print(f"JSON decoding error in file: {file_path}")
            print(f"Error: {e}")
        except Exception as e:
            print(f"Error processing file: {file_path}")
            print(f"Error: {e}")
            traceback.print_exc()
        return label_data