You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
This repo is archived. You can view files and clone it, but cannot push or open issues/pull-requests.
 
 
 
 
 
 

153 rivejä
3.7 KiB

  1. import abc
  2. import hashlib
  3. import json
  4. import shutil
  5. import time
  6. import urllib.parse
  7. import urllib.request
  8. import textwrap
  9. import sys
  10. class Authorization:
  11. def __init__(self, filename):
  12. with open(filename, "r") as file:
  13. lines = file.readlines()
  14. if len(lines) != 2:
  15. sys.exit("Incorrect api-keys file")
  16. self.public_key = lines[0].strip()
  17. self.private_key = lines[1].strip()
  18. def generate_params(self):
  19. params = dict()
  20. params["apikey"] = self.public_key
  21. ts = str(time.time())
  22. to_hash = ts + self.private_key + self.public_key
  23. hasher = hashlib.md5()
  24. hasher.update(to_hash.encode())
  25. digest = hasher.hexdigest()
  26. params["ts"] = ts
  27. params["hash"] = digest
  28. return params
  29. class Client:
  30. base_url = "http://gateway.marvel.com/v1/public"
  31. def __init__(self, auth):
  32. self.auth = auth
  33. def make_request(self, query):
  34. params = self.auth.generate_params()
  35. params.update(query.params())
  36. query_string = urllib.parse.urlencode(params)
  37. full_url = Client.base_url + query.path() + "?" + query_string
  38. with urllib.request.urlopen(full_url) as response:
  39. status_code = response.getcode()
  40. if status_code != 200:
  41. sys.exit("got status: " + str(status_code))
  42. body = json.loads(response.read())
  43. attribution = body["attributionText"]
  44. response = query.extract(body)
  45. return (response, attribution)
  46. class Display:
  47. def __init__(self, width=80):
  48. self.width = width
  49. def display(self, text):
  50. print("\n".join(textwrap.wrap(text, width=self.width)))
  51. class Query(metaclass=abc.ABCMeta):
  52. @abc.abstractmethod
  53. def params(self):
  54. pass
  55. @abc.abstractmethod
  56. def path(self):
  57. pass
  58. @abc.abstractmethod
  59. def extract(self, body):
  60. pass
  61. @abc.abstractmethod
  62. def text(self, response):
  63. pass
  64. class CharacterDescription(Query):
  65. def __init__(self, name):
  66. self.name = name
  67. def params(self):
  68. return {"name": self.name}
  69. def path(self):
  70. return "/characters"
  71. def extract(self, body):
  72. first_result = body["data"]["results"][0]
  73. description = first_result["description"]
  74. return description
  75. def text(self, response):
  76. return response
  77. class CreatorNumberOfSeries(Query):
  78. def __init__(self, first_name, last_name):
  79. self.first_name = first_name
  80. self.last_name = last_name
  81. def params(self):
  82. return {"firstName": self.first_name, "lastName": self.last_name}
  83. def path(self):
  84. return "/creators"
  85. def extract(self, body):
  86. first_result = body["data"]["results"][0]
  87. return first_result["series"]["available"]
  88. def text(self, response):
  89. return "{} {} worked on {} series".format(
  90. self.first_name, self.last_name, response
  91. )
  92. def main():
  93. auth = Authorization("api-keys.txt")
  94. query_type = sys.argv[1]
  95. if query_type == "character-description":
  96. name = sys.argv[2]
  97. query = CharacterDescription(name)
  98. elif query_type == "creator-number-of-series":
  99. first_name = sys.argv[2]
  100. last_name = sys.argv[3]
  101. query = CreatorNumberOfSeries(first_name, last_name)
  102. else:
  103. sys.exit("Unkwnon query type: {}".format(query_type))
  104. client = Client(auth)
  105. response, attribution = client.make_request(query)
  106. text = query.text(response)
  107. terminal_size = shutil.get_terminal_size()
  108. columns = terminal_size.columns
  109. terminal = Display(width=columns)
  110. terminal.display(text)
  111. terminal.display("---")
  112. terminal.display(attribution)
  113. if __name__ == "__main__":
  114. main()