import requests
from requests.auth import HTTPBasicAuth
from urllib.parse import urlencode

import pandas as pd
import matplotlib.pyplot as plt

# set host, like http://localhost:8000, to where your Data Hub is running
HOST = ''
# in case of basic auth set to HTTPBasicAuth('user', 'password')
AUTH = None
# API TOKEN
TOKEN = ""

q = {
    'datalayer_key': 'dhs_waterimprov',

    # Optional filters to narrow down the returned data:
    #'shape_id': '',
    #'shape_type': '',
    #'start_date': '',
    #'end_date': '',
}
response = requests.get(
    f"{HOST}/api/datalayers/data/?{urlencode(q)}",
    headers={"Authorization": f"Bearer {TOKEN}"},
    auth=AUTH,
)
if response.status_code == 200:
    d = response.json()
else:
    raise Exception(f"Request failed ({response.status_code}): {response.text}")

dhs_waterimprov_df = pd.DataFrame(d['data'])
dhs_waterimprov_df['year'] = pd.to_datetime(dhs_waterimprov_df['year'], format="%Y")