#!/usr/bin/env python
# coding: utf-8

# In[1]:


import os
os.environ['ROOK_URL'] = 'http://cp4cds-cn1.dkrz.de/wps'
# os.environ['ROOK_URL'] = 'http://localhost:5000/wps'


# In[2]:


from rooki import rooki
rooki.url


# ## subsetting

# In[3]:


collection = 'c3s-cmip5.output1.ICHEC.EC-EARTH.historical.day.atmos.day.r1i1p1.tas.latest'
time = '1880-01-01/1900-12-30'


# In[4]:


resp = rooki.subset(collection, time=time)
resp.ok


# In[5]:


resp.download_urls()


# ### provenance for subsetting

# In[6]:


prov_json_url = resp.provenance()
prov_json_url


# In[7]:


import requests
requests.get(prov_json_url).json()


# In[8]:


prov_plot_url = resp.provenance_image()
prov_plot_url


# In[9]:


from IPython.display import Image
Image(prov_plot_url)


# ## workflow

# In[10]:


from rooki import operators as ops


# In[11]:


wf = ops.Subset(
    ops.Subset(
        ops.Input(
            'tas', ['c3s-cmip5.output1.ICHEC.EC-EARTH.historical.day.atmos.day.r1i1p1.tas.latest']
        ),
        time="1860-01-01/1920-12-30",
    ),
    time="1880-01-01/1900-12-30"
)


# In[13]:


resp = wf.orchestrate()
resp


# In[14]:


resp.download_urls()


# ### provenance

# In[15]:


from IPython.display import Image
Image(resp.provenance_image())