Change indentation from spaces to tabs as discussed here https://github.com/marekjm...

[diaspy.git] / diaspy / search.py

diff --git a/diaspy/search.py b/diaspy/search.py
index 3c66ec5a89d8e3799fa4737584da60e148d2e919..513917531e11a934ac9636e311697de3e0964c58 100644 (file)
--- a/diaspy/search.py
+++ b/diaspy/search.py
@@ -4,30 +4,45 @@
 """
 
 
+from diaspy import errors
+
+
 class Search():
-    """This object is used for searching for content on Diaspora*.
-    """
-    def __init__(self, connection):
-        self._connection = connection
-
-    def lookup_user(self, handle):
-        """This function will launch a webfinger lookup from the pod for the
-        handle requested. Response code is returned and if the lookup was successful,
-        user should soon be searchable via pod used for connection.
-
-        :param string: Handle to search for.
-        """
-        request = self._connection.get('people', headers={'accept': 'text/html'}, params={'q': handle})
-        return request.status_code
-
-    def users(self, query):
-        """Searches for a user.
-        Will return list of dictionaries containing
-        data of found users.
-        """
-        request = self._connection.get('people.json', params={'q': query, 'utf-8': ''})
-        if request.status_code == 200:
-            result = request.json()
-        else:
-            raise errors.SearchError('wrong status code: {0}'.format(request.status_code))
-        return result
+       """This object is used for searching for content on Diaspora*.
+       """
+       def __init__(self, connection):
+               self._connection = connection
+
+       def lookupUser(self, handle):
+               """This function will launch a webfinger lookup from the pod for the
+               handle requested. Response code is returned and if the lookup was successful,
+               user should soon be searchable via pod used for connection.
+
+               :param string: Handle to search for.
+               """
+               request = self._connection.get('people', headers={'accept': 'text/html'}, params={'q': handle})
+               return request.status_code
+
+       def user(self, query):
+               """Searches for a user.
+               Will return list of dictionaries containing
+               data of found users.
+               """
+               request = self._connection.get('people.json', params={'q': query, 'utf-8': '%u2713'})
+               if request.status_code != 200:
+                       raise errors.SearchError('wrong status code: {0}'.format(request.status_code))
+               return request.json()
+
+       def tags(self, query, limit=10):
+               """Retrieve tag suggestions.
+
+               :param query: query used to search
+               :type query: str
+               :param limit: maxmal number of suggestions returned
+               :type limit: int
+               """
+               params = {'q': query, 'limit': limit}
+               request = self._connection.get('tags', params=params, headers={'x-csrf-token': repr(self._connection)})
+               if request.status_code != 200:
+                       raise errors.SearchError('wrong status code: {0}'.format(request.status_code))
+               return [i['name'] for i in request.json()]