scripts/srvuser.py

39 lines
1.2 KiB
Python

from subprocess import run, check_output
from json import loads
from http.server import BaseHTTPRequestHandler
from adduser import adduser
from ldappass import code
class srvuser(BaseHTTPRequestHandler):
"""User management HTTP server"""
def do_POST(self):
"""Handle registration POST requests"""
content_length = int(self.headers['Content-Length'])
data = loads(self.rfile.read(content_length).decode('utf-8'))
print([data[key] for key in data.keys() if key != 'password']) # Print data
if data['type'] == 'new':
# 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)