70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import configparser
|
|
import os
|
|
from os.path import exists
|
|
from flask import Flask
|
|
from flask import request
|
|
|
|
debug = False
|
|
|
|
if debug:
|
|
config_path_dir = '/tmp'
|
|
else:
|
|
config_path_dir = './instance'
|
|
config_filename = f"{config_path_dir}/config.ini"
|
|
|
|
if exists(config_filename):
|
|
config = configparser.ConfigParser()
|
|
config.read(config_filename)
|
|
if debug:
|
|
print("Sections found in the config file: ", config.sections())
|
|
if 'create-homedir' in config.sections():
|
|
for key in config['create-homedir']:
|
|
if debug:
|
|
print('Key: ', key)
|
|
if config['create-homedir']['SECRET_KEY']:
|
|
SECRET_KEY = config['create-homedir']['SECRET_KEY']
|
|
else:
|
|
raise SystemExit('Error: config file, no SECRET_KEY found.')
|
|
if config['create-homedir']['DIR_PREFIX']:
|
|
DIR_PREFIX = config['create-homedir']['DIR_PREFIX']
|
|
else:
|
|
raise SystemExit('Error: config file, no DIR_PREFIX found.')
|
|
else:
|
|
raise SystemExit('Error: no create-homedir section found.')
|
|
else:
|
|
raise SystemExit('Error: no config file found.')
|
|
|
|
def make_homedir(dir_path):
|
|
if not os.path.isdir(dir_path):
|
|
permissions = 0o700
|
|
try:
|
|
os.mkdir(dir_path, permissions)
|
|
except OSError:
|
|
return("Creation of the directory %s failed" % dir_path), 500
|
|
else:
|
|
return("Directory %s created successfully" % dir_path), 200
|
|
else:
|
|
return("Directory %s already exists" % dir_path), 200
|
|
|
|
def create_app(test_config=None):
|
|
# create and configure the app
|
|
create_home = Flask(__name__, instance_relative_config=True)
|
|
# ensure the instance folder exists
|
|
try:
|
|
os.path.isdir(DIR_PREFIX)
|
|
except OSError:
|
|
pass
|
|
|
|
# a simple page that says hello
|
|
@create_home.route('/', methods=['GET', 'POST'])
|
|
def makedir():
|
|
if request.method == 'POST':
|
|
username = request.form['user']
|
|
return make_homedir(f"{DIR_PREFIX}/{username}")
|
|
else:
|
|
return "Nosey, aren't you?"
|
|
|
|
return create_home
|