download_gedi_granules.py

import os

import click
from maap.maap import MAAP
maap = MAAP(maap_host='api.maap-project.org')
from maap.Result import Granule

from access_gedi import download_gedi

@click.command()
@click.argument('l1b_ur', type=str)
@click.argument('l2a_ur', type=str)
@click.argument('l4a_ur', type=str)
@click.argument('output_dir', type=str)
def main(l1b_ur, l2a_ur, l4a_ur, output_dir):
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)

    # Add .h5 extension if not present
    l1b_ur, l2a_ur, l4a_ur = (
        map(lambda x: f"{x}.h5" if not x.endswith('.h5') else x,
            [l1b_ur, l2a_ur, l4a_ur])
    )

    # Download the GEDI data
    download_gedi.get_gedi_data(filename=l1b_ur,
                                target_dir=output_dir)
    download_gedi.get_gedi_data(filename=l2a_ur,
                                target_dir=output_dir)
    download_gedi.get_gedi_data(filename=l4a_ur,
                                target_dir=output_dir)

    print(f"Downloaded L1B file: {l1b_ur}")
    print(f"Downloaded L2A file: {l2a_ur}")
    print(f"Downloaded L4A file: {l4a_ur}")
    print(f"Files saved in: {output_dir}")

if __name__ == '__main__':
    main()