import requests
import numpy as np
import pandas as pd

import cmocean
import matplotlib.pylab as plt
from scipy.interpolate import griddata
from scipy import interpolate
from datetime import datetime
import pdb
import os
import csv

from datetime import datetime, timedelta
import calendar

import matplotlib
matplotlib.font_manager._rebuild()

#used for map projections
from cartopy import config
import cartopy.crs as ccrs
import matplotlib.patches as mpatches

%matplotlib inline

#sets plot styles
import seaborn as sns
from matplotlib import rc
from matplotlib import rcParams
import matplotlib.ticker as mtick
rc('text', usetex=False)
rcStyle = {"font.size": 10,
           "axes.titlesize": 20,
           "axes.labelsize": 20,
           'xtick.labelsize': 16,
           'ytick.labelsize': 16}
sns.set_context("paper", rc=rcStyle)
sns.set_style("whitegrid", {'axes.grid' : False})
myColors = ["windows blue", "amber", "dusty rose", "prussian blue", "faded green", "dusty purple", "gold", "dark pink", "green", "red", "brown"]
colorsBW = ["black", "grey"]
sns.set_palette(sns.xkcd_palette(myColors))

curDir = os.getcwd()
dataDir = os.path.join(curDir, 'data')

if not os.path.exists(dataDir):
    os.mkdir(dataDir)
    
import warnings
warnings.filterwarnings('ignore')

def get_profile(profile_number):
    url = 'https://argovis.colorado.edu/catalog/profiles/{}'.format(profile_number)
    resp = requests.get(url)
    # Consider any status other than 2xx an error
    if not resp.status_code // 100 == 2:
        return "Error: Unexpected response {}".format(resp)
    profile = resp.json()
    return profile

def json2dataframe(profiles, measKey='measurements'):
    """ convert json data to Pandas DataFrame """
    # Make sure we deal with a list
    if isinstance(profiles, list):
        data = profiles
    else:
        data = [profiles]
    # Transform
    rows = []
    for profile in data:
        keys = [x for x in profile.keys() if x not in ['measurements', 'bgcMeas']]
        meta_row = dict((key, profile[key]) for key in keys)
        for row in profile[measKey]:
            row.update(meta_row)
            rows.append(row)
    df = pd.DataFrame(rows)
    return df

def get_platform_profile_metadata(platform_number):
    url = f'https://argovis.colorado.edu/catalog/platform_profile_metadata/{platform_number}'
    print(url)
    resp = requests.get(url)
    # Consider any status other than 2xx an error
    if not resp.status_code // 100 == 2:
        return "Error: Unexpected response {}".format(resp)
    platformMetadata = resp.json()
    return platformMetadata

def get_platform_profile_data(platform_number, xaxis='doxy', yaxis='pres'):
    url = 'https://argovis.colorado.edu/catalog/bgc_platform_data/{0}/?xaxis={1}&yaxis={2}'.format(platform_number, xaxis, yaxis)
    print(url)
    resp = requests.get(url)
    # Consider any status other than 2xx an error
    if not resp.status_code // 100 == 2:
        return "Error: Unexpected response {}".format(resp)
    platformData = resp.json()
    return platformData

def join_platform_data(platformMetadata, platformData):
    platforms = []
    for idx, platform in enumerate(platformMetadata):
        metadata_id = platform['_id']
        data_id = platformData[idx]['_id']
        if (metadata_id == data_id) and ('bgcMeas' in platformData[idx].keys()) and isinstance(platformData[idx]['bgcMeas'], list):
            platform['bgcMeas'] = platformData[idx]['bgcMeas']
            platforms.append(platform)
    return platforms

def get_bgc_selection_profiles(startDate, endDate, shape, xaxis, yaxis, presRange=None, printUrl=True):
    url = 'https://argovis.colorado.edu/selection/bgc_data_selection'
    url += '?startDate={}'.format(startDate)
    url += '&endDate={}'.format(endDate)
    url += '&shape={}'.format(shape)
    url += '&xaxis={}'.format(xaxis)
    url += '&yaxis={}'.format(yaxis)
    if presRange:
        pressRangeQuery = '&presRange='.format(presRange)
        url += pressRangeQuery
    url = url.replace(' ', '')
    if printUrl:
        print(url)
    resp = requests.get(url)
    # Consider any status other than 2xx an error
    if not resp.status_code // 100 == 2:
        return "Error: Unexpected response {}".format(resp)
    selectionProfiles = resp.json()
    return selectionProfiles

df = json2dataframe(profiles, 'bgcMeas')