class TestTag(unittest.TestCase):
class TestData:
def __init__(self, name):
self.name = name
self.verbose = True
def setUp(self):
self.recognition = Recognition()
def create_tags(self):
print "Creating some watchlists"
test_data = [
TestTag.TestData(name='Blacklist'),
TestTag.TestData(name='Whitelist'),
TestTag.TestData(name='Office')
]
for this_test in test_data:
print "** Testing {0} **".format(this_test.name)
tag = self.recognition.create_tag(name=this_test.name)
print 'created tag {name} {tag_id}'.format(name=tag['name'], tag_id=tag['watchlist_id'])
self.assertIsNotNone(tag['tag_id'])
def list_tags(self):
print "List all the watchlists"
tags = self.recognition.list_tags()
self.assertIsNotNone(tags)
for tag in tags:
print '{0} has id {1}'.format(tag['name'], tag['tag_id'])
def delete_tags(self):
tags = self.recognition.list_tags()
for tag in tags:
print 'Going to delete {0}'.format(tag['tag_id'])
deleted_id = self.recognition.delete_tag(tag['tag_id'])
self.assertEqual(tag['tag_id'], deleted_id)
parser.add_argument('--tags', dest='tags', action='store_true', help='List all the tags')
parser.add_argument('--tagged', dest='tagged', action='store_true', help='List all the tagged object')
parser.add_argument('--tag', dest='tag', default='Unknown', help='A tag name.')
parser.add_argument('--create', dest='create', action='store_true', help='Create a tag')
parser.add_argument('--colour', dest='colour', default='#95a5a6', help='Create a tag with this colour')
parser.add_argument('--delete', dest='delete', action='store_true', help='Delete a tag(s).')
parser.add_argument('--default', dest='default', action='store_true', help='Create a default set of tags for user')
parser.add_argument('--verbose', dest='verbose', action='store_true', help='Be more verbose')
args = parser.parse_args()
recognition = Recognition(host=args.host, key_file=args.key_file, verbose=args.verbose)
# List all tags
if args.tags:
print "Listing all the tags"
tags = recognition.list_tags()
# List tags
if args.tagged:
print "Listing all the tags"
tags = recognition.list_tagged(args.tag, args.object)
# Generic delete tag
elif args.delete:
try:
recognition.delete_tag(tag_name=args.tag, object_id=args.object)
except:
print 'Unable to delete tag id'
# Generic update tags
elif args.create:
