mirror of
https://github.com/kevinveenbirkenbach/baserow-ifto.git
synced 2024-11-01 01:23:10 +01:00
52 lines
2.1 KiB
Python
52 lines
2.1 KiB
Python
|
# controller.py
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import json
|
||
|
|
from baserow_api import BaserowAPI
|
||
|
|
import view
|
||
|
|
|
||
|
|
def main():
|
||
|
|
args = parse_arguments()
|
||
|
|
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:
|
||
|
|
tables_data = fetch_table_data(api, args.table_ids)
|
||
|
|
if args.matrix:
|
||
|
|
merged_data = api.merge_tables_on_reference(tables_data)
|
||
|
|
if not args.quiet:
|
||
|
|
view.print_json_output(merged_data)
|
||
|
|
else:
|
||
|
|
if not args.quiet:
|
||
|
|
view.print_json_output(tables_data)
|
||
|
|
else:
|
||
|
|
all_data = api.get_all_data_from_database(args.database_id)
|
||
|
|
if not args.quiet:
|
||
|
|
view.print_json_output(all_data)
|
||
|
|
|
||
|
|
def parse_arguments():
|
||
|
|
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("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("--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("--linked_fields", action="store_true", help="Outputs the linked tables")
|
||
|
|
parser.add_argument("--quiet", action="store_true", help="Suppress output of json")
|
||
|
|
return parser.parse_args()
|
||
|
|
|
||
|
|
def fetch_table_data(api, table_ids_str):
|
||
|
|
table_ids = table_ids_str.split(',')
|
||
|
|
tables_data = {}
|
||
|
|
for table_id in table_ids:
|
||
|
|
table_data = api.get_all_rows_from_table(table_id.strip())
|
||
|
|
tables_data[table_id] = table_data
|
||
|
|
return tables_data
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|