interactive_options.py

import datetime
import logging
import os
import subprocess
import uuid

import shapely
from ipyleaflet import Rectangle

LOGGER = logging.getLogger(os.path.basename(__file__))
BUCKET = 'jupyter-app-temp-storage'


def leaflet_rectangle_from_bbox(bbox):
    shapely_geom = shapely.geometry.shape(bbox)
    minx, miny, maxx, maxy = shapely_geom.bounds
    bbox_layer = Rectangle(
        name="Bounding box",
        bounds=[(miny, minx), (maxy, maxx)],
        color='black',
        fill=False,
        weight=1,  # default=5
    )
    return bbox_layer


def compute(source_raster_path, aoi_geom, target_epsg, target_raster_path,
            target_pixel_size):
    try:
        os.environ['PROJ_LIB']
    except KeyError:
        os.environ['PROJ_LIB'] = '/var/www/prototype-jupyter-layer-clipping-app/env/share/proj/'

    # Do the expensive imports here in the compute function
    LOGGER.info("Importing pygeoprocessing")
    import pygeoprocessing
    LOGGER.info("Importing GDAL")
    from osgeo import gdal
    from osgeo import osr

    WGS84_SRS = osr.SpatialReference()
    WGS84_SRS.ImportFromEPSG(4326)
    gdal.DontUseExceptions()

    LOGGER.info(f"Starting to clip {source_raster_path}")
    LOGGER.info(f"Using target EPSG:{target_epsg}")
    # Clip the source raster to the target bounding box in WGS84
    target_srs = osr.SpatialReference()
    target_srs.ImportFromEPSG(int(target_epsg))
    target_srs_wkt = target_srs.ExportToWkt()
    LOGGER.info(f"Source SRS WKT: {WGS84_SRS.ExportToWkt()}")
    LOGGER.info(f"Target SRS WKT: {target_srs_wkt}")
    LOGGER.info(f"Source geometry: {aoi_geom}")
    minx, maxx, miny, maxy = shapely.geometry.shape(aoi_geom).bounds
    source_bbox = [minx, miny, maxx, maxy]
    LOGGER.info(f"Source bbox: {source_bbox}")
    target_bbox = pygeoprocessing.transform_bounding_box(
        bounding_box=source_bbox,
        base_projection_wkt=WGS84_SRS.ExportToWkt(),
        target_projection_wkt=target_srs_wkt,
    )
    LOGGER.info(f"Bounding box converted to EPSG:{target_epsg} is "
                f"{target_bbox}")

    LOGGER.info("Warping to the target bounding box")
    if not target_pixel_size:
        LOGGER.info("Using the source raster's pixel size")
        target_pixel_size = pygeoprocessing.get_raster_info(
            source_raster_path)['pixel_size']
    LOGGER.info(f"Target pixel size:{target_pixel_size}")
    LOGGER.info(f"Target bounding box: {target_bbox}")
    LOGGER.info(f"Starting to warp {target_raster_path}")
    pygeoprocessing.warp_raster(
        base_raster_path=source_raster_path,
        target_pixel_size=target_pixel_size,
        target_raster_path=target_raster_path,
        resample_method='near',
        target_projection_wkt=target_srs_wkt,
        target_bb=target_bbox,
    )
    LOGGER.info('Finished warping')
    LOGGER.info(f"Clipped raster available at {target_raster_path}")


def upload_to_gcs(source_raster_path, layer_slug):
    from google.cloud import storage

    today = datetime.datetime.now().strftime('%Y-%m-%d')
    target_filename = f'{layer_slug}-{today}-{uuid.uuid4()}'
    try:
        storage_client = storage.Client()
        bucket = storage_client.bucket(BUCKET)
        blob = bucket.blob(target_filename)
        blob.upload_from_filename(source_raster_path)
        return f"https://storage.googleapis.com/{BUCKET}/{target_filename}"
    except:
        LOGGER.info("Falling back to cmdline gsutil")
        subprocess.run(["gsutil", "cp", source_raster_path, f'gs://{BUCKET}/target_filename'])