Skip to content

Commit e353a97

Browse files
authored
Merge pull request #51 from loklak/Patch50
Fixes #50 Integrates susi capabilities into python SDK
2 parents 1d11e4e + 85939a2 commit e353a97

File tree

3 files changed

+29
-0
lines changed

3 files changed

+29
-0
lines changed

loklak.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -174,6 +174,26 @@ def settings(self):
174174
' only localhost clients are granted.')
175175
return json.dumps(return_to_user)
176176

177+
def susi(self, query=None):
178+
"""Hits Susi with the required query and returns back the susi response"""
179+
susi_application = 'api/susi.json'
180+
url_to_give = self.baseUrl + susi_application
181+
self.query = query
182+
if query:
183+
params = {}
184+
params['q'] = self.query
185+
return_to_user = requests.get(url_to_give, params=params)
186+
if return_to_user.status_code == 200:
187+
return return_to_user.json()
188+
else:
189+
return_to_user = {}
190+
return_to_user['error'] = ('Looks like there is a problem in susi replying.')
191+
return json.dumps(return_to_user)
192+
else:
193+
return_to_user = {}
194+
return_to_user['error'] = ('Please ask susi something.')
195+
return json.dumps(return_to_user)
196+
177197
def search(self, query=None, since=None, until=None, from_user=None, count=None):
178198
"""Handles the searching"""
179199
search_application = 'api/search.json'

sample.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,10 @@
2424
hello = l.hello()
2525
pprint(hello)
2626

27+
# Loklak Susi API
28+
susi = l.susi('Hi I am Sudheesh')
29+
pprint(susi)
30+
2731
# Loklak Peers API
2832

2933
peers = l.peers()

tests/test.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,11 @@ def test_user(self):
8585
self.assertTrue('name' in result['user'])
8686
self.assertTrue('screen_name' in result['user'])
8787

88+
def test_susi(self):
89+
"""Susi Test"""
90+
result = self.loklak.susi('Hi I am Sudheesh')
91+
self.assertTrue(len(result['answers']) >= 1)
92+
8893
def test_search(self):
8994
"""test search result"""
9095
result = self.loklak.search('doctor who')

0 commit comments

Comments
 (0)