Shell Programming and Scripting

./test.py -s 5

#!/usr/bin/python

import socket, sys, getopt, signal, os

def main(argv):
        seconds = ''
        try:
                opts, args = getopt.getopt(argv,"s",[seconds])
        except getopt.GetoptError:
                print 'tcp-server.py -s <seconds>'
                sys.exit(2)

if __name__ == "__main__":
   main(sys.argv[1:])

def handler(signum, frame):
    print 'Signal handler called with signal', signum
    raise IOError("Couldn't open device!")

signal.signal(signal.SIGALRM, handler)
signal.alarm(5)

# This open() may hang indefinitely
fd = os.open('/dev/ttyS0', os.O_RDWR)

signal.alarm(0)

# Create a TCP/IP socket
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# Bind the socket to the port
server_address = ('localhost', 10000)
print >>sys.stderr, 'starting up on %s port %s' % server_address
sock.bind(server_address)

# Listen for incoming connections
sock.listen(1)

while True:
    # Wait for a connection
    print >>sys.stderr, 'waiting for a connection'
    connection, client_address = sock.accept()

    try:
        print >>sys.stderr, 'connection from', client_address

        # Receive the data in small chunks and retransmit it
        while True:
            data = connection.recv(16)
            print >>sys.stderr, 'received "%s"' % data
            if data:
                print >>sys.stderr, 'sending data back to the client'
                connection.sendall(data)
            else:
                print >>sys.stderr, 'no more data from', client_address
                break

    finally:
        # Clean up the connection
        connection.close()

#!/usr/bin/python

import optparse

parser = optparse.OptionParser()
parser.add_option('-s', '--seconds', dest='duration', help='time in seconds', default=0)
opts, args = parser.parse_args()
duration = opts.duration

print "Duration: %s" % duration