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

# # Advanced Rooki Usage

# ## Use enviroment to change rooki config

# In[ ]:


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


# In[ ]:


# change default download folder
os.environ['ROOKI_OUTPUT_DIR'] = '/tmp/rooki'


# In[ ]:


# HINT: restart to re-init rooki!
from rooki import rooki
rooki.url


# In[ ]:


rooki.output_dir


# In[ ]:


resp = rooki.subset(
    collection='c3s-cmip5.output1.ICHEC.EC-EARTH.historical.day.atmos.day.r1i1p1.tas.latest', 
    time='1880-01-01/1900-12-30')
resp.ok


# In[ ]:


# number of files to download
resp.num_files


# In[ ]:


# total size of all files in bytes
resp.size


# In[ ]:


resp.size_in_mb


# In[ ]:


resp.download_urls()


# In[ ]:


files = resp.download()


# ## Use Rooki client

# In[ ]:


from rooki.client import Rooki
url='http://cp4cds-cn1.dkrz.de/wps'
# url='http://localhost:5000/wps'

rooki = Rooki(url, mode='async', output_dir='/tmp/rooki')
rooki.url


# In[ ]:


resp = rooki.subset(
    collection='c3s-cmip5.output1.ICHEC.EC-EARTH.historical.day.atmos.day.r1i1p1.tas.latest', 
    time='1850-01-01/2008-12-30')
resp.ok


# In[ ]:


# total size
resp.size_in_mb


# In[ ]:


# download files
files = resp.download()


# In[ ]:


files[0]


# In[ ]:


# open as xarray dataset
dsets = resp.datasets()


# In[ ]:


ds = dsets[0]
ds