Refactor using a class, support /math url

sources/numbers/numbers.py

@@ -2,13 +2,38 @@ import sys
 import urllib.request
 
 
+class NumbersApi:
+    def __init__(self):
+        self.base_url = "http://numbersapi.com/"
+
+    def build_url(self, query, number):
+        if query == "trivia":
+            return self.base_url + number
+        elif query == "math":
+            return self.base_url + number + "/math"
+        else:
+            sys.exit("Unknown query: " + query)
+
+    def do_request(self, url):
+        with urllib.request.urlopen(url) as request:
+            response = request.read().decode("utf-8")
+            return response
+
+    def get(self, query, number):
+        url = self.build_url(query, number)
+        return self.do_request(url)
+
+
 def main():
+    if len(sys.argv) < 3:
+        sys.exit("not enough arguments")
     number = sys.argv[1]
-    with urllib.request.urlopen("http://numbersapi.com/" + number) as request:
-        response = request.read().decode("utf-8")
-        print(response)
+    query = sys.argv[2]
+    numbers_api = NumbersApi()
+    trivia = numbers_api.get(query, number)
+    print(trivia)
+
 
 
 if __name__ == "__main__":
     main()
-