diff --git a/modules/src/__init__.py b/modules/src/__init__.py index d00f2009..0988d343 100644 --- a/modules/src/__init__.py +++ b/modules/src/__init__.py @@ -3,6 +3,7 @@ 'book', 'currency', 'dictionary', + 'fact', 'hello', 'help', 'joke', diff --git a/modules/src/fact.py b/modules/src/fact.py new file mode 100644 index 00000000..8d3910d2 --- /dev/null +++ b/modules/src/fact.py @@ -0,0 +1,14 @@ +import requests +from templates.text import TextTemplate + +def process(input, entities=None): + output = {} + try: + r = requests.get('http://numbersapi.com/random/trivia',headers={'Content-Type':'application/json'}) + data = r.json() + output['input'] = input + output['output'] = TextTemplate(data['text']).get_message() + output['success'] = True + except: + output['success'] = False + return output diff --git a/modules/tests/test_fact.py b/modules/tests/test_fact.py new file mode 100644 index 00000000..77d474a2 --- /dev/null +++ b/modules/tests/test_fact.py @@ -0,0 +1,7 @@ +import modules + +def test_fact(): + assert('fact' == modules.process_query('tell me a fact')[0]) + assert('fact' == modules.process_query('Do you know a fact?')[0]) + assert('fact' == modules.process_query('random facts')[0]) + assert('fact' != modules.process_query('something random')[0])