http_wait_for

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from __future__ import absolute_import
from __future__ import generators
from __future__ import division

import httplib
import socket
import time
import datetime

from ansible.module_utils.basic import *

class HTTPWaitFor(object):
    class HTTPClient(object):
        def __init__(self, module):
            self.params = module.params
        
        def connect(self):
            if self.params['ssl']:
                return httplib.HTTPSConnection(self.params['host'], self.params['port'])
            else:
                return httplib.HTTPConnection(self.params['host'], self.params['port'], True)
    
    class Progress(object):
        def __init__(self):
            self.start = datetime.datetime.now()
        
        def elapsed(self):
            return (datetime.datetime.now() - self.start).seconds
        
        def wait(self, value):
            time.sleep(value)
    
    def __init__(self, AnsibleModule = AnsibleModule, HTTPClient = HTTPClient, Progress = Progress):
        self.module = AnsibleModule(
            argument_spec = dict(
                host     = dict(default='127.0.0.1', type='str'),
                port     = dict(type='int'),
                path     = dict(default='/', type='str'),
                ssl      = dict(default='no', type='bool'),
                status   = dict(default=200, type='int'),
                interval = dict(default=3  , type='int'),
                timeout  = dict(default=60 , type='int'),
            )
        )
        if 'port' not in self.module.params:
            self.module.params['port'] = 443 if self.module.params['ssl'] else 80
        
        self.HTTPClient = HTTPClient
        self.Progress = Progress
    
    def execute(self):
        progress = self.Progress()
        params = self.module.params
        
        while True:
            try:
                connection = self.HTTPClient(self.module).connect()
                connection.request('GET', params['path'])
                response = connection.getresponse()
                connection.close()
                
                if response.status == params['status']:
                    self.module.exit_json(status = 1, elapsed = progress.elapsed())
                    break
                
            except httplib.BadStatusLine:
                connection.close()
                self.module.fail_json(msg = 'Non HTTP server!!', status = 3, elapsed = progress.elapsed())
                break
              
            except socket.error:
                connection.close()
            
            elapsed = progress.elapsed()
            
            if elapsed >= params['timeout']:
                self.module.fail_json(msg = 'Timeout!! sec %d' % elapsed, status = 2, elapsed = elapsed)
                break
            else:
                progress.wait(params['interval'])
                continue

if __name__ == '__main__':
    HTTPWaitFor().execute()