]> git.lizzy.rs Git - amogus.git/blob - backend.py
Ignore furrybot commands
[amogus.git] / backend.py
1 import cleverbotfree
2 from urllib.parse import urlparse, parse_qs
3 from http.server import HTTPServer, BaseHTTPRequestHandler
4
5 cleverbot = None
6
7 class AmogusServer(BaseHTTPRequestHandler):
8         def _set_headers(self):
9                 self.send_response(200)
10                 self.send_header("Content-type", "text/plain")
11                 self.end_headers()
12
13         def do_GET(self):
14                 self._set_headers()
15                 query_components = parse_qs(urlparse(self.path).query)
16                 self.wfile.write(cleverbot.single_exchange(query_components["message"][0]).encode("utf8"))
17
18 def main():
19         global cleverbot
20
21         with cleverbotfree.sync_playwright() as cleverbot_pw:
22                 cleverbot = cleverbotfree.Cleverbot(cleverbot_pw)
23
24                 httpd = HTTPServer(("localhost", 6969), AmogusServer)
25
26                 try:
27                         httpd.serve_forever()
28                 except KeyboardInterrupt:
29                         pass
30                 finally:
31                         cleverbot.close()
32
33 if __name__ == "__main__":
34         main()