#!/usr/bin/python
# -*- coding: utf-8 -*-
""" Version: 1.1 """
""" Author: Christian Schwatke <christian.schwatke@tum.de>"""
""" Last change: 2019-06-26 """

import os
import sys
import json
import pprint
import requests
import datetime

""" Download path were all water level time series will be stored """
download_path = 'downloads/'

args = {}
""" User configuration """
# IMPORTANT! Your DAHITI user account must be assigned to the 'GlobalCDA' group in order 
# to access globalcda parameters such as arc-id, poly-id, basin-id
args['username'] = 'username'
args['password'] = 'password'

""" General search options """
#args['basin'] = 'Amazon' # These basins differ from globalcda basins
#args['continent'] = 'Asia'
#args['country'] = 'de'
#args['min_lon'] = 0
#args['max_lon'] = 10
#args['min_lat'] = 0
#args['max_lat'] = 10
#args['software'] = '3.1'

""" Globalcda search options """
#~ args['basin_id'] = '7309' # Mississippi
args['basin_id'] = '9232' # Amazon
#~ args['basin_id'] = '9719' # Congo
#~ args['basin_id'] = '7756' # Ganges
#~ args['poly_id'] = '527'
#~ args['arc_id'] = '29046'

url = 'https://dahiti.dgfi.tum.de/api/v1/'

""" send request as method POST """
args['action'] = 'list-targets'
response = requests.post(url, data=args)
if response.status_code == 200:
	""" convert json string in python list """
	data = json.loads(response.text)
	print 'Dataset(s) found:',len(data)
	i = 0
	for record in data:
		print 'Downloading ... ',record['id'],'->',record['target_name'].encode("utf8"),'('+os.path.abspath(download_path+'/'+str(record['id'])+'.txt')+')'
		if not record.has_key('globalcda_arc_id'):
			print 'ERROR: The field `globalcda_arc_id` not found!' 
			print 'Possible reason(s):'
			print '    Your account is probably not assigned to the usergroup `GlobalCDA`.'
			print '    Please contact Christian Schwatke in order to update your account.'
			sys.exit(0)

		""" download water level time series """
		args['action'] = 'download'
		args['dahiti_id'] = record['id']

		response_download = requests.post(url, data=args)
		if response_download.status_code == 200:
			""" convert json string in python list """
			data = json.loads(response_download.text)
			output = open(download_path+'/'+record['id']+'.txt','w')
			output.write('# DAHITI-ID       : '+str(record['id'])+'\n')
			output.write('# Target name     : '+record['target_name'].encode("utf8")+'\n')
			if record['location'] == None:
				output.write('# Location        : '+str(record['location'])+'\n')
			else:
				output.write('# Location        : '+record['location'].encode("utf8")+'\n')
			output.write('# Continent       : '+str(record['continent'])+'\n')
			output.write('# Country         : '+str(record['country'])+'\n')
			output.write('# Longitude       : '+str(record['longitude'])+'\n')
			output.write('# Latitude        : '+str(record['latitude'])+'\n')
			output.write('# Points          : '+str(len(data))+'\n')
			output.write('# Software        : '+str(record['water_level']['software'])+'\n')
			output.write('# Download        : '+str(datetime.datetime.now())[0:19]+'\n')
			output.write('# Last-Update     : '+str(record['water_level']['last_update'])+'\n')			
			output.write('# Arc-Id          : '+str(record['globalcda_arc_id'])+'\n')
			output.write('# Poly-Id         : '+str(record['globalcda_poly_id'])+'\n')
			output.write('# Basin-Id        : '+str(record['globalcda_basin_id'])+'\n')
			output.write('# ----------------------------------------\n')
			output.write('# column 1        : date [yyyy-mm-dd]\n')
			output.write('# column 2        : normal heights w.r.t. geoid model [m]\n')
			output.write('# column 3        : error [m] (Kalman Filter, formal errors)\n')
			output.write('# column 4        : dataset (OP:operational, RT:real-time, CB:operational+real-time)\n')
			output.write('# ----------------------------------------\n')
			for entry in data:								
				output.write(entry['date']+' '+entry['height']+' '+entry['error']+' '+entry['data_type']+'\n')
			output.close()
		elif response_download.status_code == 403:
			print 'Status-Code:',response_download.status_code,'- Permission denied!'
		else:
			print 'Status-Code:',response_download.status_code
		
elif response.status_code == 403:
	print 'Status-Code:',response.status_code,'- Permission denied!'
else:
	print 'Status-Code:',response.status_code