def main(argv):
# parse arguments
parser = argparse.ArgumentParser()
parser.add_argument("-t", "--tag", help="tag images", action='store_true')
parser.add_argument("-c", "--color", help="color images", action='store_true')
parser.add_argument("-u", "--usage", help="usage", action='store_true')
args, argv = parser.parse_known_args()
if len(argv) == 1:
imageurl = argv[0]
else:
imageurl = 'http://clarifai-img.s3.amazonaws.com/test/toddler-flowers.jpeg'
api = ClarifaiApi()
if args.usage:
response = api.get_usage()
elif not args.color:
response = tag_images(api, imageurl)
elif args.color and not args.tag:
response = color_images(api, imageurl)
else:
raise Exception('call with --tag or --color for the image')
print(json.dumps(response))
def test_usage(self):
api = ClarifaiApi()
response = api.get_usage()
self.assertTrue(response)
self.assertTrue(len(response['user_throttles']) == 2)
