def handleSquirrelsIndex(self):
db = SquirrelDB()
squirrelsList = db.getSquirrels()
self.send_response(200)
self.send_header("Content-Type", "application/json")
self.end_headers()
self.wfile.write(bytes(json.dumps(squirrelsList), "utf-8"))
def handleSquirrelsDelete(self, squirrelId):
db = SquirrelDB()
squirrel = db.getSquirrel(squirrelId)
if squirrel:
db.deleteSquirrel(squirrelId)
self.send_response(204)
self.end_headers()
else:
self.handle404()
def handleSquirrelsUpdate(self, squirrelId):
db = SquirrelDB()
squirrel = db.getSquirrel(squirrelId)
if squirrel:
db.updateSquirrel(squirrelId, self.getRequestData())
self.send_response(204)
self.end_headers()
else:
self.handle404()
def handleSquirrelsRetrieve(self, squirrelId):
db = SquirrelDB()
squirrel = db.getSquirrel(squirrelId)
if squirrel:
self.send_response(200)
self.send_header("Content-Type", "application/json")
self.end_headers()
self.wfile.write(bytes(json.dumps(squirrel), "utf-8"))
else:
self.handle404()
def handleSquirrelsUpdate(self, squirrelId):
db = SquirrelDB()
squirrel = db.getSquirrel(squirrelId)
if squirrel:
body = self.getRequestData()
db.updateSquirrel(squirrelId, body["name"], body["size"])
self.send_response(204)
self.end_headers()
else:
self.handle404()
def run():
db = SquirrelDB()
db.createSquirrelsTable()
db = None # disconnect
port = 8080
if len(sys.argv) > 1:
port = int(sys.argv[1])
listen = ("0.0.0.0", port)
server = HTTPServer(listen, SquirrelServerHandler)
print("Server listening on", "{}:{}".format(*listen))
server.serve_forever()
def handleSquirrelsCreate(self):
db = SquirrelDB()
body = self.getRequestData()
db.createSquirrel(body["name"], body["size"])
self.send_response(201)
self.end_headers()
def handleSquirrelsCreate(self):
db = SquirrelDB()
db.createSquirrel(self.getRequestData())
self.send_response(201)
self.end_headers()