Implemented linked fields parameter

This commit is contained in:
Kevin Veen-Birkenbach 2023-09-04 19:44:43 +02:00
parent 74e5f3e73f
commit ef8fe33a5d

View File

@ -1,41 +1,42 @@
# controller.py
import argparse import argparse
import json import json
from baserow_api import BaserowAPI from baserow_api import BaserowAPI
import view import view
def handle_output(quiet,data):
if not quiet:
view.print_json_output(data)
def main(): def main():
args = parse_arguments() args = parse_arguments()
api = BaserowAPI(args.base_url, args.api_key, args.verbose) api = BaserowAPI(args.base_url, args.api_key, args.verbose)
if not args.database_id and not args.table_ids:
view.print_error_message("Either database_id or table_ids must be provided.")
exit(1)
if args.table_ids: if args.table_ids:
tables_data = fetch_table_data(api, args.table_ids) tables_data = fetch_table_data(api, args.table_ids)
if args.matrix: if "linked_fields" in args.output:
linked_fields_data = api.get_link_fields_for_all_tables(tables_data)
handle_output(args.quiet, linked_fields_data)
if "matrix" in args.output:
merged_data = api.merge_tables_on_reference(tables_data) merged_data = api.merge_tables_on_reference(tables_data)
if not args.quiet: handle_output(args.quiet, merged_data)
view.print_json_output(merged_data)
else: if "tables" in args.output:
if not args.quiet: handle_output(args.quiet, tables_data)
view.print_json_output(tables_data)
else: if args.database_id:
all_data = api.get_all_data_from_database(args.database_id) all_data = api.get_all_data_from_database(args.database_id)
if not args.quiet: if not args.quiet:
view.print_json_output(all_data) view.print_json_output(all_data)
def parse_arguments(): def parse_arguments():
parser = argparse.ArgumentParser(description="Fetch all data from a Baserow database.") parser = argparse.ArgumentParser(description="Fetch all data from a Baserow database.")
parser.add_argument("base_url", help="Base URL of your Baserow instance, e.g., https://YOUR_BASEROW_INSTANCE_URL/api/") parser.add_argument("base_url", help="Base URL of your Baserow instance, e.g., https://YOUR_BASEROW_INSTANCE_URL/api/")
parser.add_argument("api_key", help="Your Baserow API key.") parser.add_argument("api_key", help="Your Baserow API key.")
parser.add_argument("--database_id", help="ID of the Baserow database you want to fetch data from.", default=None) parser.add_argument("--database_id", help="ID of the Baserow database you want to fetch data from.", default=None)
parser.add_argument("--table_ids", help="IDs of the Baserow tables you want to fetch data from, separated by commas.", default=None) parser.add_argument("--table_ids", help="IDs of the Baserow tables you want to fetch data from, separated by commas.", default=None)
parser.add_argument("--matrix", action="store_true", help="Merge tables based on references.")
parser.add_argument("-v", "--verbose", action="store_true", help="Enable verbose mode for debugging.") parser.add_argument("-v", "--verbose", action="store_true", help="Enable verbose mode for debugging.")
parser.add_argument("--linked_fields", action="store_true", help="Outputs the linked tables") parser.add_argument("--output", choices=["linked_fields", "tables", "matrix"], default=[], nargs='+', help="Specify the type(s) of output: linked_fields, matrix, content or both")
parser.add_argument("--quiet", action="store_true", help="Suppress output of json") parser.add_argument("--quiet", action="store_true", help="Suppress output of json")
return parser.parse_args() return parser.parse_args()