Archiveteam/grab-site
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Read bytes from file and send the correct Content-Length

Browse Source
This commit is contained in:
Ivan Kozik 2016-02-24 23:52:37 +00:00
parent 7486daa33f
commit 25b3c9c076
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libgrabsite/server.py
View File

@ -67,21 +67,22 @@ class GrabberServerProtocol(WebSocketServerProtocol):
def sendServerStatus(self, redirectUrl=None, redirectAfter=0):
requestPath = self.http_request_uri.split("?")[0]
if requestPath == "/":
with open(os.path.join(os.path.dirname(__file__), "dashboard.html"), "r") as f:
with open(os.path.join(os.path.dirname(__file__), "dashboard.html"), "r", encoding="utf-8") as f:
dashboardHtml = f.read()
self.sendHtml(dashboardHtml)
else:
self.send404()
def send404(self):
with open(os.path.join(os.path.dirname(__file__), "404.html"), "r") as f:
responseHtml = f.read()
with open(os.path.join(os.path.dirname(__file__), "404.html"), "rb") as f:
responseBody = f.read()
response = "HTTP/1.1 404 Not Found\r\n"
response += "Content-Type: text/html; charset=UTF-8\r\n"
response += "Content-Length: {}\r\n".format(len(responseHtml))
response += "Content-Length: {}\r\n".format(len(responseBody))
response += "\r\n"
response += responseHtml
self.sendData(response.encode("utf_8"))
self.sendData(response.encode("utf-8"))
self.sendData(responseBody)
class GrabberServerFactory(WebSocketServerFactory):
protocol = GrabberServerProtocol