get-rekognition-result-recap.py

#! /usr/bin/python
import boto3
import json
import sys
import subprocess
import time
import datetime
foundedNames = []

def actions(name):
	if not (name in foundedNames):
		foundedNames.append(name)
		# subprocess.call(["python", "say_hi.py", name])
		subprocess.call(["python", "sns-publish.py", name])
	else:
		print "Founded %d peoples" % len(foundedNames)

# def putFaces(number):
# 	data = {}
# 	data["faces"] = number
# 	data["timeStamp"] = datetime.datetime.now().isoformat()
# 	payload = json.dumps(data)
# 	response = kinesis.put_record(StreamName=kinesisFaceCountStreamName, Data=payload, PartitionKey="Camera01")
# 	print response

with open('config.json') as json_data_file:
    config = json.load(json_data_file)

region = config['region']
kinesisDataStream = config['kinesisDataStreamName']
kinesisFaceCountStreamName = config['kinesisFaceCountStreamName']

kinesis = boto3.client('kinesis', region_name=region)
stream = kinesis.describe_stream(StreamName=kinesisDataStream)
shardId = stream['StreamDescription']['Shards'][0]['ShardId']
shardIterator = kinesis.get_shard_iterator(StreamName=kinesisDataStream,
									 ShardId=shardId,
									 ShardIteratorType="LATEST")
shard_it = shardIterator["ShardIterator"]
while(True):
	try:
		recs = kinesis.get_records(ShardIterator=shard_it, Limit=1)
		# print recs
		shard_it = recs["NextShardIterator"]
		if len(recs['Records']) > 0:
			data = json.loads(recs['Records'][0]['Data'])
			# print data['FaceSearchResponse']
			if len(data['FaceSearchResponse']) > 0:
				detectedFaces = len(data['FaceSearchResponse'])
				subprocess.Popen(["python", "put-face-count.py", str(detectedFaces)])
				print 'detect faces: %d' % detectedFaces
				for faceSearchResponse in data['FaceSearchResponse']:
					if len(faceSearchResponse['MatchedFaces']) > 0:
						print 'match faces: %d' % len(faceSearchResponse['MatchedFaces'])
						for face in faceSearchResponse['MatchedFaces']:
							name = face['Face']['ExternalImageId']
							confidence = face['Face']['Confidence']
							print 'match face: %s confidence: %d' % (name, confidence)
							actions(name)
	except Exception as e:
		print e.message
		time.sleep(1)
	# avoid ProvisionedThroughputExceededException
	time.sleep(0.2)