From c66bf51878efc951db510530ef86d64950849919 Mon Sep 17 00:00:00 2001 From: kisiel Date: Fri, 20 Nov 2015 22:25:34 +0100 Subject: [PATCH] added Video API support --- diffbot.py | 22 +++++++++++++--------- 1 file changed, 13 insertions(+), 9 deletions(-) diff --git a/diffbot.py b/diffbot.py index e433eab..20d053b 100644 --- a/diffbot.py +++ b/diffbot.py @@ -21,8 +21,8 @@ class Client(object): """Diffbot client.""" - _apis = frozenset(('article', 'frontpage', 'product', 'image', 'analyze', - 'discussion')) + _apis = frozenset(('article', 'frontpage', 'product', 'image', 'video', + 'analyze', 'discussion')) def __init__(self, token, version=API_VERSION): """Initialise the client.""" @@ -105,6 +105,10 @@ def image(self, url, **kwargs): """Image API.""" return self.api('image', url, **kwargs) + def video(self, url, **kwargs): + """Video API.""" + return self.api('video', url, **kwargs) + def analyze(self, url, **kwargs): """Classifier (analyze) API.""" return self.api('analyze', url, **kwargs) @@ -123,15 +127,10 @@ def crawl(self, urls, name='crawl', api='analyze', **kwargs): urls = ' '.join(urls) url = self.endpoint('crawl') process_url = self.endpoint(api) - params = { - 'token': self._token, - 'seeds': urls, - 'name': name, - 'apiUrl': process_url, - } + params = {'token': self._token, 'seeds': urls, 'name': name, + 'apiUrl': process_url, 'maxToCrawl': 10} # Add any additional named parameters as accepted by Crawlbot - params['maxToCrawl'] = 10 params.update(kwargs) self._get(url, params=params) @@ -217,6 +216,11 @@ def image(url, token, **kwargs): return api('image', url, token, **kwargs) +def video(url, token, **kwargs): + """Shortcut for `Client(token, version).video(url)`.""" + return api('video', url, token, **kwargs) + + def analyze(url, token, **kwargs): """Shortcut for `Client(token, version).analyze(url)`.""" return api('analyze', url, token, **kwargs)