def _connect()

in src/databao_context_engine/plugins/resources/parquet_introspector.py [0:0]

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

    def _connect(self, file_config: ParquetConfigFile):
        duckdb_secret = file_config.duckdb_secret
        with duckdb.connect() as conn:
            if duckdb_secret is not None:
                if duckdb_secret.type == "s3":
                    conn.execute("INSTALL httpfs;")
                    conn.execute("LOAD httpfs;")
                    conn.execute("INSTALL s3;")
                    conn.execute("LOAD s3;")
                with _create_secret(conn, duckdb_secret):
                    yield conn
            else:
                yield conn