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()