Archived
135 lines
4.5 KiB
Python
135 lines
4.5 KiB
Python
from http.server import BaseHTTPRequestHandler, HTTPServer
|
|
from views import get_all_animals, get_single_animal
|
|
from http.server import BaseHTTPRequestHandler, HTTPServer
|
|
from views.employees_requests import get_all_employees, get_single_employee
|
|
from views.location_requests import get_all_locations, get_single_location
|
|
|
|
|
|
|
|
# Here's a class. It inherits from another class.
|
|
# For now, think of a class as a container for functions that
|
|
# work together for a common purpose. In this case, that
|
|
# common purpose is to respond to HTTP requests from a client.
|
|
class HandleRequests(BaseHTTPRequestHandler):
|
|
def parse_url(self, path):
|
|
# Just like splitting a string in JavaScript. If the
|
|
# path is "/animals/1", the resulting list will
|
|
# have "" at index 0, "animals" at index 1, and "1"
|
|
# at index 2.
|
|
path_params = path.split("/")
|
|
resource = path_params[1]
|
|
id = None
|
|
|
|
# Try to get the item at index 2
|
|
try:
|
|
# Convert the string "1" to the integer 1
|
|
# This is the new parseInt()
|
|
id = int(path_params[2])
|
|
except IndexError:
|
|
pass # No route parameter exists: /animals
|
|
except ValueError:
|
|
pass # Request had trailing slash: /animals/
|
|
|
|
return (resource, id) # This is a tuple
|
|
# This is a Docstring it should be at the beginning of all classes and functions
|
|
# It gives a description of the class or function
|
|
"""Controls the functionality of any GET, PUT, POST, DELETE requests to the server
|
|
"""
|
|
|
|
# Here's a class function
|
|
def _set_headers(self, status):
|
|
# Notice this Docstring also includes information about the arguments passed to the function
|
|
"""Sets the status code, Content-Type and Access-Control-Allow-Origin
|
|
headers on the response
|
|
|
|
Args:
|
|
status (number): the status code to return to the front end
|
|
"""
|
|
self.send_response(status)
|
|
self.send_header('Content-type', 'application/json')
|
|
self.send_header('Access-Control-Allow-Origin', '*')
|
|
self.end_headers()
|
|
# Another method! This supports requests with the OPTIONS verb.
|
|
|
|
def do_OPTIONS(self):
|
|
"""Sets the options headers
|
|
"""
|
|
self.send_response(200)
|
|
self.send_header('Access-Control-Allow-Origin', '*')
|
|
self.send_header('Access-Control-Allow-Methods',
|
|
'GET, POST, PUT, DELETE')
|
|
self.send_header('Access-Control-Allow-Headers',
|
|
'X-Requested-With, Content-Type, Accept')
|
|
self.end_headers()
|
|
|
|
# Here's a method on the class that overrides the parent's method.
|
|
# It handles any GET request.
|
|
def do_GET(self):
|
|
self._set_headers(200)
|
|
response = {} # Default response
|
|
|
|
# Parse the URL and capture the tuple that is returned
|
|
(resource, id) = self.parse_url(self.path)
|
|
|
|
if resource == "animals":
|
|
if id is not None:
|
|
response = f"{get_single_animal(id)}"
|
|
|
|
else:
|
|
response = f"{get_all_animals()}"
|
|
|
|
self.wfile.write(response.encode())
|
|
|
|
if resource == "locations":
|
|
if id is not None:
|
|
response = f"{get_single_location(id)}"
|
|
|
|
else:
|
|
response = f"{get_all_locations()}"
|
|
|
|
self.wfile.write(response.encode())
|
|
|
|
if resource == "employees":
|
|
if id is not None:
|
|
response = f"{get_single_employee(id)}"
|
|
|
|
else:
|
|
response = f"{get_all_employees()}"
|
|
|
|
self.wfile.write(response.encode())
|
|
|
|
# Here's a method on the class that overrides the parent's method.
|
|
# It handles any POST request.
|
|
def do_POST(self):
|
|
"""Handles POST requests to the server
|
|
"""
|
|
# Set response code to 'Created'
|
|
self._set_headers(201)
|
|
|
|
content_len = int(self.headers.get('content-length', 0))
|
|
post_body = self.rfile.read(content_len)
|
|
response = f"received post request:<br>{post_body}"
|
|
self.wfile.write(response.encode())
|
|
|
|
# Here's a method on the class that overrides the parent's method.
|
|
# It handles any PUT request.
|
|
|
|
def do_PUT(self):
|
|
"""Handles PUT requests to the server
|
|
"""
|
|
self.do_POST()
|
|
|
|
|
|
# This function is not inside the class. It is the starting
|
|
# point of this application.
|
|
def main():
|
|
"""Starts the server on port 8088 using the HandleRequests class
|
|
"""
|
|
host = ''
|
|
port = 8088
|
|
HTTPServer((host, port), HandleRequests).serve_forever()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|