#!/usr/bin/env python

import io
import socket
import struct
import time
import picamera

TCP_IP = '10.0.1.16'
TCP_PORT = 9001
BUFFER_SIZE = 1024

client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client_socket.connect((TCP_IP, TCP_PORT))

# Make a file-like object out of the connection
connection = client_socket.makefile('wb')
try:
    with picamera.PiCamera() as camera:
        camera.resolution = (640, 480)
        # Start a preview and let the camera warm up for 2 seconds
        camera.start_preview()
        time.sleep(2)

        # Note the start time and construct a stream to hold image data
        # temporarily (we could write it directly to connection but in this
        # case we want to find out the size of each capture first to keep
        # our protocol simple)
        start = time.time()
        stream = io.BytesIO()
        #for foo in camera.capture_continuous(stream, 'jpeg'):
            # Write the length of the capture to the stream and flush to
            # ensure it actually gets sent
        camera.capture(stream, 'jpeg')
        connection.write(struct.pack('<L', stream.tell()))
        connection.flush()
            # Rewind the stream and send the image data over the wire
        stream.seek(0)
        connection.write(stream.read())
            # If we've been capturing for more than 30 seconds, quit
        	#if time.time() - start > 30:
                #break
            # Reset the stream for the next capture
            #stream.seek(0)
            #stream.truncate()
    # Write a length of zero to the stream to signal we're done
    connection.write(struct.pack('<L', 0))
finally:
    connection.close()
    client_socket.close()