tomasino
/
agena
forked from solderpunk/agena
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add query support.

This commit is contained in:
Solderpunk 2019-08-14 21:36:21 +03:00
parent bf78dd5d35
commit 808edc594a
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
agena.py
View File

@ -76,6 +76,7 @@ class AgenaHandler(socketserver.BaseRequestHandler):
# Use item type 1 for top-level selector
self.gopher_itemtype = "1"
self.gopher_selector = parsed.path
self.gopher_query = parsed.query
def download_gopher_resource(self):
"""
@ -85,7 +86,12 @@ class AgenaHandler(socketserver.BaseRequestHandler):
# Send request
s = socket.create_connection((self.gopher_host, self.gopher_port))
s.sendall((self.gopher_selector + '\r\n').encode("UTF-8"))
if self.gopher_query:
request = self.gopher_selector + '\t' + self.gopher_query
else:
request = self.gopher_selector
request += '\r\n'
s.sendall(request.encode("UTF-8"))
# Write gopher response to temp file
tmpf = tempfile.NamedTemporaryFile("wb", delete=False)