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

# Copyright (C) 2014 Enrico Rossi
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import urllib3
import json
import Cookie
from time import sleep

user = 'myusername'
passwd = 'mypassword'
headers = {'content-type': 'application/json'}

# Prepare the connection with TLSv1 as SSL support for the HTTPS.
conn = urllib3.connectionpool.HTTPSConnectionPool('myhostname.mydomain', ssl_version='TLSv1')

# The 1st post will be a json with username and password.
payload = json.dumps({'username':user, 'password':passwd})

# Do the connection and retrieve the result.
r=conn.urlopen('POST', '/login', body=payload, headers=headers)

# Debug the result.
jsonr = json.loads(r.data)
print 'HDR:', r.headers
print 'DATA:', r.data
print 'JSON:',jsonr.keys()

# get the cookie from the headers
cookie=Cookie.SimpleCookie(r.headers['set-cookie'])
# and set the next header to send with the cookie in it.
headers['Cookie']=cookie.output(attrs=[], header='').strip()

# delay some seconds.
print 'Wait ...'
sleep(10)

# re-connect the server, sending back the cookie in order to get the
# password-protected page.
r=conn.urlopen('GET', '/getsomejsonthing', headers=headers)

# debug
jsonr = json.loads(r.data)
print 'HDR:', r.headers
print 'DATA:', r.data
print 'JSON:',jsonr.keys()

# update the new header with the (optionally) re-newed cookie from the server to be ready for the next request.
cookie.load(r.headers['set-cookie'])
headers['Cookie']=cookie.output(attrs=[], header='').strip()

# debug the new cookie. Normally now you will get a new protected-page
# as before.
print cookie.output(attrs=[], header='').strip()

# end
r.close()

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4