def update_key_columns()

in nl2sql/datasets/custom.py [0:0]

• 16 lines of code
• 4 McCabe index (conditional complexity)

    def update_key_columns(self, dataset_id):
        """
        Update Key columns of tables present in the dataset.
        """
        try:
            pkey = pd.read_excel(self.filepath, sheet_name="Primary Keys")
            fkey = pd.read_excel(self.filepath, sheet_name="Foreign Keys")

            pkey["Query"] = generate_pk_query(
                dataset_id, pkey["Table"], pkey["Primary Key"]
            )
            fkey["Query"] = generate_fk_query(
                dataset_id, fkey["Table"], fkey["Foreign Key"], fkey["References"]
            )

            for pk_query in pkey["Query"].tolist():
                self.client.query(pk_query)

            for fk_query in fkey["Query"].tolist():
                self.client.query(fk_query)
        except ValueError as err:
            logger.error(f"Sheetname value error: {err}")