ig-pixelfed-mirror/server.py

from http.server import BaseHTTPRequestHandler, HTTPServer
import igmirror
import sys
import re

def update_igaccount(name):
    print(name)

class MyServer(BaseHTTPRequestHandler):
    def do_GET(self):
        self.send_response(200)
        self.send_header("Content-type", "application/json")
        self.end_headers()
        path = self.path.strip('/')
        parts = path.split('/')
        if len(parts) == 2:
            # make sure account name contains only safe characters
            accname = re.sub(r'[^a-zA-Z0-9_]+', '', parts[0])
            if parts[1].lower() == 'add':
                igmirror.add_igaccount(accname)
            elif parts[1].lower() == 'update':
                update_igaccount(accname)
        self.wfile.write(bytes('OK', "utf-8"))

if __name__ == "__main__": 
    port = 8080
    if len(sys.argv) > 1:
        port = int(sys.argv[1])
    webServer = HTTPServer(('0.0.0.0', port), MyServer)
    print("Server started http://%s:%s" % ('0.0.0.0', port))

    try:
        webServer.serve_forever()
    except KeyboardInterrupt:
        pass

    webServer.server_close()
    print("Server stopped.")