Merge pull request #32 from PolBaladas/suggestCLI

Add suggest method in loklak.py and link to loklak CLI command

Author: sudheesh001

bin/loklak

@@ -1,14 +1,14 @@
 #!/usr/bin/env python
 # encoding: utf-8
 import argparse
-import loklak
+from loklak import Loklak
 from pprint import pprint
 import os
 
 parser = argparse.ArgumentParser(description="Python wrapper around the loklak API.")
-parser.add_argument('-s', '--search', type=str, help='Search API Wrapper which helps to query loklak for JSON results.')
+parser.add_argument('-s', '--search', nargs='+', help='Search API Wrapper which helps to query loklak for JSON results.')
 parser.add_argument('-t', '--status', action='store_true', help='Status API Wrapper for the loklak status check.')
-parser.add_argument('-st', '--suggest', action='store_true', help='Suggestions API Wrapper , Works better with local loklak instance.')
+parser.add_argument('-st', '--suggest', nargs='+', help='Suggestions API Wrapper , Works better with local loklak instance.')
 parser.add_argument('-c', '--crawler', action='store_true', help='Crawler API Wrapper on Loklak to crawl for tweets for a particular crawl depth.')
 parser.add_argument('-hl', '--hello', action='store_true', help='Loklak status check API.')
 parser.add_argument('-g', '--geocode', help='Geocode API for geolocation based information.')
@@ -19,13 +19,14 @@ parser.add_argument('-m', '--map', help='Map Visualization render using Loklak s
 parser.add_argument('-md', '--markdown', help='Markdown conversion API to render markdown as image using Loklak.')
 args = parser.parse_args()
 
-loklak = loklak.Loklak()
+loklak = Loklak()
 if args.search:
     pprint(loklak.search(args.search))
 elif args.status:
     pprint(loklak.status())
 elif args.suggest:
-    pass
+    print(args.suggest)
+    pprint(loklak.suggest(args.suggest))
 elif args.crawler:
     pass
 elif args.hello:

loklak.py

@@ -157,6 +157,30 @@ def search(self, query=None, since=None, until=None, from_user=None):
                                        ' given to run a query for')
             return json.dumps(return_to_user)
 
+    def suggest(self, query=None, count=None, order=None, orderby=None,since=None, until=None):
+        suggest_application = 'api/suggest.json'
+        url_to_give = self.baseUrl+suggest_application
+        params  = {}
+        if query:
+            params['query'] = query
+        if count:
+            params['count'] = count
+        if order:
+            params['order'] = order
+        if since:
+            params['since'] = since
+        if until:
+            params['until'] = until
+        print(params)
+        return_to_user = requests.get(url_to_give, params=params)
+        print(return_to_user.url)
+        if return_to_user.status_code == 200:
+            return return_to_user.json()
+        else :
+            return_to_user = {}
+            return_to_user['error'] = ('Something went wrong, looks like the server is down.')
+            return json.dumps(return_to_user)
+
     def aggregations(self, query=None, since=None, until=None,
                      fields=None, limit=None):
         """Gives the aggregations of the application"""

sample.py

@@ -103,6 +103,17 @@
 search6 = l.search('sudheesh001', '2015-01-10', '2015-01-21','sudheesh001')
 pprint(search6)
 
+###loklak Suggestion API
+suggestion1 = l.suggest()
+pprint(suggest1)
+
+suggestion2 = l.suggest(count=20, order='desc')
+pprint(suggestion2)
+
+
+suggestion3 = l.suggest(count=10, order='asc')
+pprint(suggestion3)
+
 ###  account
 account1 = l.account('name')
 pprint(account1)