def setUp(self):
service_connection = MockConnection()
self.contents = '0123456789'
bucket = MockBucket(service_connection, 'mybucket')
key = bucket.new_key('mykey')
key.set_contents_from_string(self.contents)
self.keyfile = KeyFile(key)
parser.add_argument('-b',
'--bucket',
help='Bucket name on AWS S3',
required=True)
args = parser.parse_args()
bucket = BucketWrapper(args.bucket).load()
classifier = ImageClassifier(config)
error_file = open('errors.log', 'w')
while bucket.keys:
try:
key = bucket.pop()
keyfile = KeyFile(key)
print('Processing: {}'.format(keyfile.name))
result = classifier.identify_image(keyfile)
with open(result['filename'] + '.json', 'w') as f:
res = json.dump(result,
f,
sort_keys=True,
indent=4,
separators=(',', ': '))
except Exception as ex:
try:
error_file.write('Key: {} failed with error: {} \n'.format(
key.name, str(ex)))
except Exception as ex:
error_file.write('Miserable failure: {} \n'.format(str(ex)))
error_file.close()
