SHOW:
|
|
- or go back to the newest paste.
| 1 | - | #!/usr/bin/env python |
| 1 | + | #!/usr/bin/env python |
| 2 | - | # |
| 2 | + | # |
| 3 | - | # Let me know when the server comes back online |
| 3 | + | # Let me know when the server comes back online |
| 4 | - | |
| 4 | + | |
| 5 | - | |
| 5 | + | |
| 6 | - | import argparse |
| 6 | + | import argparse |
| 7 | - | import socket |
| 7 | + | import socket |
| 8 | - | import errno |
| 8 | + | import errno |
| 9 | - | from time import time as now |
| 9 | + | from time import time as now |
| 10 | - | |
| 10 | + | |
| 11 | - | DEFAULT_TIMEOUT = 120 |
| 11 | + | DEFAULT_TIMEOUT = 120 |
| 12 | - | DEFAULT_SERVER_HOST = 'localhost' |
| 12 | + | DEFAULT_SERVER_HOST = 'http://pastebin.com/' |
| 13 | - | DEFAULT_SERVER_PORT = 80 |
| 13 | + | DEFAULT_SERVER_PORT = 80 |
| 14 | - | |
| 14 | + | |
| 15 | - | |
| 15 | + | |
| 16 | - | class NetServiceChecker(object): |
| 16 | + | class NetServiceChecker(object): |
| 17 | - | """ waiting for service to come back on line """ |
| 17 | + | """ waiting for service to come back on line """ |
| 18 | - | def __init__(self, host, port, timeout=DEFAULT_TIMEOUT): |
| 18 | + | def __init__(self, host, port, timeout=DEFAULT_TIMEOUT): |
| 19 | - | self.host = host |
| 19 | + | self.host = host |
| 20 | - | self.port = port |
| 20 | + | self.port = port |
| 21 | - | self.timeout = timeout |
| 21 | + | self.timeout = timeout |
| 22 | - | self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
| 22 | + | self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
| 23 | - | |
| 23 | + | |
| 24 | - | def end_wait(self): |
| 24 | + | def end_wait(self): |
| 25 | - | self.sock.close() |
| 25 | + | self.sock.close() |
| 26 | - | |
| 26 | + | |
| 27 | - | def check(self): |
| 27 | + | def check(self): |
| 28 | - | """ Check the service """ |
| 28 | + | """ Check the service """ |
| 29 | - | if self.timeout: |
| 29 | + | if self.timeout: |
| 30 | - | end_time = now() + self.timeout |
| 30 | + | end_time = now() + self.timeout |
| 31 | - | |
| 31 | + | |
| 32 | - | while True: |
| 32 | + | while True: |
| 33 | - | try: |
| 33 | + | try: |
| 34 | - | if self.timeout: |
| 34 | + | if self.timeout: |
| 35 | - | next_timeout = end_time - now() |
| 35 | + | next_timeout = end_time - now() |
| 36 | - | if next_timeout < 0: |
| 36 | + | if next_timeout < 0: |
| 37 | - | return False |
| 37 | + | return False |
| 38 | - | else: |
| 38 | + | else: |
| 39 | - | print "Setting socket next timeout %ss"\ |
| 39 | + | print "Setting socket next timeout %ss"\ |
| 40 | - | %round(next_timeout) |
| 40 | + | %round(next_timeout) |
| 41 | - | self.sock.settimeout(next_timeout) |
| 41 | + | self.sock.settimeout(next_timeout) |
| 42 | - | self.sock.connect((self.host, self.port)) |
| 42 | + | self.sock.connect((self.host, self.port)) |
| 43 | - | # handle exceptions |
| 43 | + | # handle exceptions |
| 44 | - | except socket.timeout, err: |
| 44 | + | except socket.timeout, err: |
| 45 | - | if self.timeout: |
| 45 | + | if self.timeout: |
| 46 | - | return False |
| 46 | + | return False |
| 47 | - | except socket.error, err: |
| 47 | + | except socket.error, err: |
| 48 | - | print "Exception: %s" %err |
| 48 | + | print "Exception: %s" %err |
| 49 | - | else: |
| 49 | + | else: |
| 50 | - | self.end_wait() |
| 50 | + | self.end_wait() |
| 51 | - | return True |
| 51 | + | return True |
| 52 | - | |
| 52 | + | |
| 53 | - | |
| 53 | + | |
| 54 | - | if __name__ == '__main__': |
| 54 | + | if __name__ == '__main__': |
| 55 | - | parser = argparse.ArgumentParser(description='Wait for Network Service') |
| 55 | + | parser = argparse.ArgumentParser(description='Wait for Network Service') |
| 56 | - | parser.add_argument('--host', action="store", dest="host", default=DEFAULT_SERVER_HOST) |
| 56 | + | parser.add_argument('--host', action="store", dest="host", default=DEFAULT_SERVER_HOST)
|
| 57 | - | parser.add_argument('--port', action="store", dest="port", type=int,\ |
| 57 | + | parser.add_argument('--port', action="store", dest="port", type=int,\
|
| 58 | - | default=DEFAULT_SERVER_PORT) |
| 58 | + | default=DEFAULT_SERVER_PORT) |
| 59 | - | parser.add_argument('--timeout', action="store", dest="timeout", type=int,\ |
| 59 | + | parser.add_argument('--timeout', action="store", dest="timeout", type=int,\
|
| 60 | - | default=DEFAULT_TIMEOUT) |
| 60 | + | default=DEFAULT_TIMEOUT) |
| 61 | - | given_args = parser.parse_args() |
| 61 | + | given_args = parser.parse_args() |
| 62 | - | host, port, timeout = given_args.host, given_args.port, given_args.timeout |
| 62 | + | host, port, timeout = given_args.host, given_args.port, given_args.timeout |
| 63 | - | service_checker = NetServiceChecker(host, port, timeout=timeout) |
| 63 | + | service_checker = NetServiceChecker(host, port, timeout=timeout) |
| 64 | - | print "Checking for net service %s:%s ..." %(host, port) |
| 64 | + | print "Checking for net service %s:%s ..." %(host, port) |
| 65 | - | if service_checker.check(): |
| 65 | + | if service_checker.check(): |
| 66 | print "Service is available again!" |
