|
|
@@ -1,107 +0,0 @@ |
|
|
|
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() |