From a1ec0b9760c48bca0063a2c6a17376defdc095b6 Mon Sep 17 00:00:00 2001 From: Kevin Veen-Birkenbach Date: Mon, 4 Sep 2023 18:04:41 +0200 Subject: [PATCH] Refactored class --- baserow_api.py | 86 +++++++++++++++++++++++++------------------------- 1 file changed, 43 insertions(+), 43 deletions(-) diff --git a/baserow_api.py b/baserow_api.py index d92e70e..842ac20 100644 --- a/baserow_api.py +++ b/baserow_api.py @@ -6,6 +6,8 @@ class BaserowAPI: self.base_url = base_url self.api_key = api_key self.verbose = verbose + self.headers = create_headers(self) + self.print_verbose_message("Headers:", headers) def create_headers(self): """Create headers for API requests.""" @@ -13,13 +15,16 @@ class BaserowAPI: "Authorization": f"Token {self.api_key}", "Content-Type": "application/json" } + + def print_verbose_message(self, message): + if self.verbose: + print(message) def handle_api_response(self, response): """Handle API response, check for errors and decode JSON.""" - if self.verbose: - print("[INFO] Handling API response...") - print("Response Status Code:", response.status_code) - print("Response Headers:", response.headers) + self.print_verbose_message("[INFO] Handling API response...") + self.print_verbose_message("Response Status Code:", response.status_code) + self.print_verbose_message("Response Headers:", response.headers) if response.status_code != 200: print(f"Error: Received status code {response.status_code} from Baserow API.") print("Response content:", response.content.decode()) @@ -34,15 +39,12 @@ class BaserowAPI: def get_all_rows_from_table(self, table_id): if self.verbose: print(f"[INFO] Fetching all rows from table with ID: {table_id}...") - headers = self.create_headers() rows = [] next_url = f"{self.base_url}database/rows/table/{table_id}/" while next_url: - response = requests.get(next_url, headers=headers) - if self.verbose: - print("Headers:", headers) - print("Requesting:", next_url) + response = requests.get(next_url, headers=self.headers) + self.print_verbose_message("Requesting:", next_url) data = self.handle_api_response(response) if not data: break @@ -52,17 +54,12 @@ class BaserowAPI: return rows def get_all_tables_from_database(self, database_id): - if self.verbose: - print(f"[INFO] Fetching all tables from database with ID: {database_id}...") - headers = self.create_headers() - response = requests.get(f"{self.base_url}database/tables/database/{database_id}/", headers=headers) - if self.verbose: - print("Headers:", headers) + self.print_verbose_message("[INFO] Fetching all tables from database with ID: {database_id}...") + response = requests.get(f"{self.base_url}database/tables/database/{database_id}/", headers=self.headers) return self.handle_api_response(response) or [] def get_all_data_from_database(self, database_id): - if self.verbose: - print(f"[INFO] Fetching all data from database with ID: {database_id}...") + self.print_verbose_message("[INFO] Fetching all data from database with ID: {database_id}...") tables = self.get_all_tables_from_database(database_id) data = {} @@ -75,43 +72,46 @@ class BaserowAPI: def fetch_fields_for_table(self, table_id): """Fetch fields for a given table.""" - headers = self.create_headers() - response = requests.get(f"{self.base_url}database/fields/table/{table_id}/", headers=headers) + response = requests.get(f"{self.base_url}database/fields/table/{table_id}/", headers=self.headers) if response.status_code == 200: return response.json() - else: - raise Exception(f"Failed to fetch fields for table {table_id}. Status code: {response.status_code}") + raise Exception(f"Failed to fetch fields for table {table_id}. Status code: {response.status_code}") def merge_tables_on_reference(self, tables_data): - if self.verbose: - print("Merging tables based on references...") + self.print_verbose_message("Merging tables based on references...") + indexed_data = self.index_tables_by_id(tables_data) + link_fields = self.get_link_fields_for_all_tables(tables_data) + self.embed_referenced_data_into_tables(tables_data, indexed_data, link_fields) + self.print_verbose_message("Merged Tables Data:", tables_data) + return tables_data - # Create a mapping of table names to their rows indexed by ID + def index_tables_by_id(self, tables_data): indexed_data = {table_name: {row['id']: row for row in rows} for table_name, rows in tables_data.items()} - if self.verbose: - print("Indexed Data:", indexed_data) + self.print_verbose_message("Indexed Data: {indexed_data}") + return indexed_data - # Fetch field information for each table and identify link fields + def get_link_fields_for_all_tables(self, tables_data): link_fields = {} for table_name in tables_data: - fields = self.fetch_fields_for_table(table_name) - link_fields_for_table = [field for field in fields if field['type'] == 'link_row'] + link_fields_for_table = self.get_link_fields_for_table(table_name) link_fields[table_name] = link_fields_for_table - if self.verbose: - print("Link Fields:", link_fields) - print("Link Fields For Table:", link_fields_for_table) + self.print_verbose_message("Link Fields For Table: {link_fields_for_table}") + self.print_verbose_message("Link Fields: {link_fields}") + return link_fields - # Embed referenced data into tables + def get_link_fields_for_table(self, table_name): + fields = self.fetch_fields_for_table(table_name) + return [field for field in fields if field['type'] == 'link_row'] + + def embed_referenced_data_into_tables(self, tables_data, indexed_data, link_fields): for table_name, rows in tables_data.items(): for row in rows: - for link_field in link_fields[table_name]: - field_name = link_field['name'] - referenced_table_id = link_field['link_row_table_id'] - if field_name in row and row[field_name] in indexed_data[referenced_table_id]: - if self.verbose: - print(f"Embedding referenced data for field {field_name} in table {table_name}") - row[field_name] = indexed_data[referenced_table_id][row[field_name]] + self.embed_data_for_row(row, table_name, indexed_data, link_fields) - if self.verbose: - print("Merged Tables Data:", tables_data) - return tables_data \ No newline at end of file + def embed_data_for_row(self, row, table_name, indexed_data, link_fields): + for link_field in link_fields[table_name]: + field_name = link_field['name'] + referenced_table_id = link_field['link_row_table_id'] + if field_name in row and row[field_name] in indexed_data[referenced_table_id]: + self.print_verbose_message(f"Embedding referenced data for field {field_name} in table {table_name}") + row[field_name] = indexed_data[referenced_table_id][row[field_name]]