manage.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys
import os
import getpass
import hashlib
import binascii

from flask import Flask, send_from_directory
from flask_mail import Mail

from measor.views import (
    IndexView, LogoutView, CreateTaskView, TaskDetailView, TaskEditView,
    TaskDeleteView, ApiTask, ApiTasks, PauseTaskView, ApiTaskLogStatus, ApiStats
)
from measor.filters import format_datetime, timestamp2date, decodeUnicode
from measor.docker import init_docker


def create_app():
    app = Flask(__name__)

    app.config.from_object('settings')
    sys.path.insert(0, os.path.join(app.config['BASE_DIR']))

    @app.route('/static/<path:path>')
    def send_static(path):
        return send_from_directory('static', path)

    if app.config['RUN_DOCKER_ON_START']:
        container = init_docker(app.config)
        app.config['CONTAINER_ID'] = container.id

    app.add_url_rule('/', view_func=IndexView.as_view('index'))
    app.add_url_rule('/new_task', view_func=CreateTaskView.as_view('create_task'))
    app.add_url_rule('/logout', view_func=LogoutView.as_view('logout'))
    app.add_url_rule('/task/<slug>', view_func=TaskDetailView.as_view('task_detail'))
    app.add_url_rule('/task/<slug>/edit', view_func=TaskEditView.as_view('edit_task'))
    app.add_url_rule('/task/<slug>/pause', view_func=PauseTaskView.as_view('pause_task'))
    app.add_url_rule('/task/<slug>/delete', view_func=TaskDeleteView.as_view('delete_task'))
    app.add_url_rule('/task/<slug>/<log_name>', view_func=TaskDetailView.as_view('log_detail'))

    app.add_url_rule('/api/stats/', view_func=ApiStats.as_view('stats'))
    app.add_url_rule('/api/task/<slug>', view_func=ApiTask.as_view('task_api'))
    app.add_url_rule('/api/task/', view_func=ApiTasks.as_view('tasks_api'))
    app.add_url_rule('/api/<slug>/log/<log_name>/status', view_func=ApiTaskLogStatus.as_view('log_status_api'))

    app.jinja_env.filters['decodeUnicode'] = decodeUnicode
    app.jinja_env.filters['datetime'] = format_datetime
    app.jinja_env.filters['timestamp2date'] = timestamp2date
    return app


def create_user(settings):
    print('Username: ')
    name = input()
    if not name:
        print("Username can not be empty")
        return False
    password = getpass.getpass('Password: ')
    passw2 = getpass.getpass('Password again: ')
    if password == passw2:
        path = settings.get('AUTH_FILE_PATH')
        pass_str = 'username=%s; password=%s' % (name, password)
        dk = hashlib.pbkdf2_hmac('sha256', pass_str.encode(), settings.get('AUTH_SALT'), 100000)
        token = binascii.hexlify(dk).decode()
        f = open(path, 'a')
        f.write(token + '\n')
        return True
    else:
        print('Passwords does not match')
    return False


if 'runserver' in sys.argv:
    port = int(os.environ.get('PORT', 8000))
    app = create_app()
    app.run(port=port)
elif 'create_user' in sys.argv:
    app = create_app()
    create_user(app.config)