cours-python/saison-1/sources/numbers/numbers_object.py

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]
    query = sys.argv[2]
    numbers_api = NumbersApi()
    trivia = numbers_api.get(query, number)
    print(trivia)



if __name__ == "__main__":
    main()