added imagetagging call

lineback · lineback · commit 98d4894077f1 · 2014-05-14T15:28:02.000-06:00
diff --git a/alchemyapi.py b/alchemyapi.py
@@ -16,6 +16,8 @@
 
 from __future__ import print_function
 
+import requests
+
 try:
 	from urllib.request import urlopen
 	from urllib.parse import urlparse
@@ -117,6 +119,9 @@ class AlchemyAPI:
 	ENDPOINTS['combined']['text'] = '/text/TextGetCombinedData'
 	ENDPOINTS['image'] = {}
 	ENDPOINTS['image']['url'] = '/url/URLGetImage'
+	ENDPOINTS['imagetagging'] = {}
+	ENDPOINTS['imagetagging']['url'] = '/url/URLGetRankedImageKeywords'
+	ENDPOINTS['imagetagging']['image'] = '/image/ImageGetRankedImageKeywords'
 	ENDPOINTS['taxonomy'] = {}
 	ENDPOINTS['taxonomy']['url'] = '/url/URLGetRankedTaxonomy'
 	ENDPOINTS['taxonomy']['html'] = '/html/HTMLGetRankedTaxonomy'
@@ -718,7 +723,26 @@ def combined(self, flavor, data, options={}):
 		options[flavor] = data
 		return self.__analyze(AlchemyAPI.ENDPOINTS['combined'][flavor], options)
 
-	def __analyze(self, endpoint, params):
+	def imageTagging(self, flavor, data, options={}):
+		"""
+		
+		INPUT:
+		flavor -> which version of the call only url or image.
+		data -> the data to analyze, either the the url or path to image.
+		options -> various parameters that can be used to adjust how the API works, see below for more info on the available options.
+		"""
+		if flavor not in AlchemyAPI.ENDPOINTS['imagetagging']:
+			return { 'status':'ERROR', 'statusInfo':'imagetagging for ' + flavor + ' not available' }
+		elif 'image' == flavor:
+			image = open(data, 'rb').read()
+			options['imagePostMode'] = 'raw'
+			return self.__analyze(AlchemyAPI.ENDPOINTS['imagetagging'][flavor], options, image)
+	
+		options[flavor] = data
+		return self.__analyze(AlchemyAPI.ENDPOINTS['imagetagging'][flavor], options)
+		
+	
+	def __analyze(self, endpoint, params, post_data=bytearray()):
 		"""
 		HTTP Request wrapper that is called by the endpoint functions. This function is not intended to be called through an external interface. 
 		It makes the call, then converts the returned JSON string into a Python object. 
@@ -730,17 +754,15 @@ def __analyze(self, endpoint, params):
 		The response, already converted from JSON to a Python object. 
 		"""
 
-		#Insert the base url
-		url = AlchemyAPI.BASE_URL + endpoint;
-
 		#Add the API Key and set the output mode to JSON
 		params['apikey'] = self.apikey;
 		params['outputMode'] = 'json';
+		#Insert the base url
+		
+		post_url = AlchemyAPI.BASE_URL + endpoint + '?' + urlencode(params).encode('utf-8');
 		
 		try: 
-			#build the request with uri encoding
-			page = urlopen(url, data=urlencode(params).encode('utf-8')).read().decode('utf-8')
-			return json.loads(page)
+			return requests.post(url=post_url, data=post_data).json()
 		except Exception as e:
 			print(e)
 			return { 'status':'ERROR', 'statusInfo':'network-error' }
diff --git a/example.py b/example.py
@@ -23,7 +23,7 @@
 demo_text = 'Yesterday dumb Bob destroyed my fancy iPhone in beautiful Denver, Colorado. I guess I will have to head over to the Apple Store and buy a new one.'
 demo_url = 'http://www.npr.org/2013/11/26/247336038/dont-stuff-the-turkey-and-other-tips-from-americas-test-kitchen'
 demo_html = '<html><head><title>Python Demo | AlchemyAPI</title></head><body><h1>Did you know that AlchemyAPI works on HTML?</h1><p>Well, you do now.</p></body></html>'
-
+image_url = 'http://demo1.alchemyapi.com/images/vision/football.jpg'
 
 print('')
 print('')
@@ -477,6 +477,36 @@
 print('')
 
 
+print('')
+print('')
+print('')
+print('############################################')
+print('#   Image tagging Example                  #')
+print('############################################')
+print('')
+print('')
+
+print('Processing url: ', demo_url)
+print('')
+
+response = alchemyapi.imageTagging('url',image_url)
+
+if response['status'] == 'OK':
+	print('## Response Object ##')
+	print(json.dumps(response, indent=4))
+
+	print('')
+	print('## Keywords ##')
+	for keyword in response['imageKeywords']:
+		print(keyword['text'], ' : ', keyword['score'])
+	print('')
+else:
+	print('Error in image tagging call: ', response['statusInfo'])
+
+print('')
+print('')
+
+
 print('')
 print('')
 print('')
