27 lines
773 B
Python
27 lines
773 B
Python
from folium import Map, Marker
|
|
from pandas import DataFrame, json_normalize
|
|
|
|
from app.data_request import request_dataset
|
|
from constants import COLUMNS, COORDINATES
|
|
|
|
|
|
def create_dataframe(dataset) -> DataFrame:
|
|
"""
|
|
Creates a DataFrame from a JSON response
|
|
"""
|
|
json = request_dataset(dataset)
|
|
df = json_normalize(data=json, record_path=["records"], errors="ignore",)
|
|
filtered_df = df.filter(items=COLUMNS[dataset])
|
|
return filtered_df
|
|
|
|
|
|
def create_map(df):
|
|
"""
|
|
Creates a Map with markers from the DataFrame
|
|
"""
|
|
m = Map(location=COORDINATES, zoom_start=12)
|
|
for index, row in df.iterrows():
|
|
lng, lat = row["fields.geo_shape.coordinates"]
|
|
Marker(location=[lat, lng]).add_to(m)
|
|
m.save("app/templates/map.html")
|