wsgi.py

# ===============================================================================
# Copyright 2022 ross
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ===============================================================================


from three_d_model import FormationDB
from fastapi import FastAPI, Request
from fastapi.templating import Jinja2Templates
from fastapi.responses import HTMLResponse
from fastapi.staticfiles import StaticFiles

from response_models import Formation
from typing import List

app = FastAPI()
# app.mount("/static", StaticFiles(directory="static"), name="static")

fm = FormationDB()

templates = Jinja2Templates(directory="templates")


@app.get("/", response_class=HTMLResponse)
async def root(request: Request,
               lat=None,
               lon=None,
               easting=None,
               northing=None,
               depth=None):
    formation = None
    if depth:
        if lat and lon:
            formation = fm.get_formation(float(lon), float(lat), float(depth))
        elif easting and northing:
            formation = fm.get_formation(float(easting), float(northing), float(depth), as_utm=True)

    return templates.TemplateResponse("index.html", {'request': request,
                                                     'formation': formation,
                                                     'lat': lat or '',
                                                     'lon': lon or '',
                                                     'northing': northing or '',
                                                     'easting': easting or '',
                                                     'depth': depth or ''})


@app.get("/stratigraphy/{x}/{y}", response_model=List[Formation])
async def read_stratigraphy(x: float, y: float):
    return fm.get_stratigraphy(x, y)


@app.get("/formation/{x}/{y}/{z}", response_model=Formation)
async def read_formation(x: float, y: float, z: float, as_utm: bool = False):
    return fm.get_formation(x, y, z, as_utm=as_utm)

# ============= EOF =============================================