import hashlib import requests import shutil import time import textwrap import sys class Authorization: def __init__(self, filename): with open(filename, "r") as file: lines = file.readlines() if len(lines) != 2: sys.exit("Incorrect api-keys file") self.public_key = lines[0].strip() self.private_key = lines[1].strip() def generate_params(self): params = dict() params["apikey"] = self.public_key ts = str(time.time()) to_hash = ts + self.private_key + self.public_key hasher = hashlib.md5() hasher.update(to_hash.encode()) digest = hasher.hexdigest() params["ts"] = ts params["hash"] = digest return params class Client(): base_url = "http://gateway.marvel.com/v1/public" def __init__(self, auth): self.auth = auth def make_request(self, path, query_params): params = self.auth.generate_params() params.update(query_params) url = Client.base_url + path response = requests.get(url, params=params) status_code = response.status_code if status_code != 200: sys.exit("got status: " + str(status_code)) body = response.json() return (body, body["attributionText"]) class CharacterDescriptionGetter(Client): def __init__(self, auth): super().__init__(auth) def get_character_description(self, name): params = { "name": name } response, attribution = super().make_request("/characters", params) description = response["data"]["results"][0]["description"] return description, attribution class CreatorNumberOfSeriesGetter(Client): def __init__(self, auth): super().__init__(auth) def get_number_of_series(self, first_name, last_name): params = { "firstName": first_name, "lastName": last_name } response, attribution = super().make_request("/creators", params) result = response["data"]["results"][0]["series"]["available"] return result, attribution class Display: def __init__(self, width=80): self.width = width def display(self, text): print("\n".join(textwrap.wrap(text, width=self.width))) def main(): auth = Authorization("api-keys.txt") query = sys.argv[1] if query == "character-description": name = sys.argv[2] character_description_getter = CharacterDescriptionGetter(auth) description, attribution = character_description_getter.get_character_description(name) text = description elif query == "creator-number-of-series": first_name = sys.argv[2] last_name = sys.argv[3] creator_number_of_series_getter = CreatorNumberOfSeriesGetter(auth) result, attribution = creator_number_of_series_getter.get_number_of_series(first_name, last_name) text = first_name + " " + last_name + " worked on " + str(result) + " series" terminal_size = shutil.get_terminal_size() columns = terminal_size.columns terminal = Display(width=columns) terminal.display(text) terminal.display("---") terminal.display(attribution) if __name__ == "__main__": main()