53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
from subprocess import run, check_output
|
|
from urllib.parse import parse_qs
|
|
|
|
from http.server import BaseHTTPRequestHandler
|
|
from adduser import adduser
|
|
from ldappass import code
|
|
|
|
|
|
class srvuser(BaseHTTPRequestHandler):
|
|
"""User management HTTP server"""
|
|
|
|
def do_GET(self):
|
|
"""Handle GET requests"""
|
|
|
|
body = 'It worked!'
|
|
self.send_response(200)
|
|
self.send_header('Content-Type', 'text/plain')
|
|
self.send_header('Content-Length', str(len(body)))
|
|
self.end_headers()
|
|
self.wfile.write(body.encode('utf-8'))
|
|
|
|
def do_POST(self):
|
|
"""Handle API POST requests"""
|
|
|
|
content_length = int(self.headers['Content-Length'])
|
|
data = parse_qs(self.rfile.read(content_length).decode('utf-8'))
|
|
|
|
# Print debug data
|
|
debug = data.copy()
|
|
debug.pop('password')
|
|
debug.pop('confirmpassword')
|
|
print(debug)
|
|
|
|
if self.path == '/api/join':
|
|
# New user
|
|
if data['code'] != code():
|
|
print('Incorrect code')
|
|
self.send_response(403)
|
|
return
|
|
if not all(c.isdigit() or c.islower() for c in data['username']) or data['username'][0].isdigit():
|
|
print('Bad username')
|
|
self.send_response(403)
|
|
return
|
|
|
|
# Add the user
|
|
adduser(data['username'], data['firstname'].capitalize(), data['lastname'].capitalize(),
|
|
data['email'], data['password'])
|
|
else:
|
|
print('Bad request')
|
|
self.send_response(400)
|
|
return
|
|
|
|
self.send_response(200)
|