the-toolbox/main.py at master · andressegeo/the-toolbox · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# coding: utf-8

import os
import fnmatch
import logging
from config import CONFIG
from user_api.user_api import UserApi
from user_api.flask_user_api import FlaskUserApi
from flask import Flask, send_from_directory, send_file
from pyrestdbapi.api import Api
from server.api.task_api import TaskApi
from pyrestdbapi.db_api_blueprint import FlaskRestDBApi
from pysqlcollection.client import Client
from server.api.comment_api import CommentApi
from server.api.user_has_task_api import UserHasTaskApi
from server.service.standard_mail_io import StandardMailIO
from notification_config import NOTIFICATION_CONFIG

formatter = logging.Formatter(u'%(message)s')
logger = logging.getLogger()

if not logger.handlers:
    handler = logging.StreamHandler()
else:
    handler = logger.handlers[0]

handler.setFormatter(formatter)
logger.addHandler(handler)
logger.setLevel(logging.INFO)

# create flask server
APP = Flask(__name__)

# Init & register User API
USER_API = UserApi(**CONFIG[u"user-api"])
FLASK_USER_API = FlaskUserApi(USER_API)
USER_API_BLUEPRINT = FLASK_USER_API.construct_blueprint()

# Init & register DB API
DB_API_CONF = CONFIG[u"db-api"]

CLIENT = Client(
    unix_socket=DB_API_CONF.get(u"db_unix_socket"),
    host=DB_API_CONF.get(u"db_host"),
    user=DB_API_CONF[u"db_user"],
    password=DB_API_CONF[u"db_password"]
)
DB = getattr(CLIENT, DB_API_CONF[u"db_name"])

MAIL_IO = StandardMailIO(NOTIFICATION_CONFIG)

DB_REST_API_CONFIG = {
    u"projects": Api(DB, default_table_name=u"project"),
    u"clients": Api(DB, default_table_name=u"client"),
    u"hours": Api(DB, default_table_name=u"hour"),
    u"project_assignements": Api(DB, default_table_name=u"project_assignement"),
    u"users": Api(DB, default_table_name=u"user"),
    u"task-lists": Api(DB, default_table_name=u"task_list"),
    u"tasks": TaskApi(DB, MAIL_IO, NOTIFICATION_CONFIG),
    u"comments": CommentApi(DB, MAIL_IO, NOTIFICATION_CONFIG),
    u"tags": Api(DB, default_table_name="tag"),
    u"task-tags": Api(DB, default_table_name="task_has_tag"),
    u"clients_affected_to_users": Api(DB, default_table_name=u"clients_affected_to_users"),
    u"project_consumption": Api(DB, default_table_name=u"project_consumption"),
    u"project_consumption_per_user": Api(DB, default_table_name=u"project_consumption_per_user"),
    u"project-loads": Api(DB, default_table_name=u"project_load"),
    u"cras": Api(DB, default_table_name=u"cra"),
    u"roles": Api(DB, default_table_name=u"role"),
    u"project_files": Api(DB, default_table_name=u"project_file"),
    u"task-assignements": UserHasTaskApi(DB, MAIL_IO, NOTIFICATION_CONFIG),
    u"tasks-sum-up": Api(DB, default_table_name=u"task_sum_up"),
    u"tasks-left": Api(DB, default_table_name=u"tasks_left")
}

DB_FLASK_API = FlaskRestDBApi(DB_REST_API_CONFIG)
DB_API_BLUEPRINT = DB_FLASK_API.construct_blueprint()

# App routes.

@APP.route('/<path:path>')
def send_client(path):
    """
    Handles client resources.
    """
    return send_from_directory('dist/', path)

@APP.errorhandler(404)
def send_index(e):
    """
    Redirect to client index file if not found.
    """
    return send_file("dist/index.html")

@DB_API_BLUEPRINT.before_request
@FLASK_USER_API.is_connected(login_url="/login")
def is_connected():
    pass

# Finally register the Db API blueprint
APP.register_blueprint(DB_API_BLUEPRINT, url_prefix=u"/api/db")
APP.register_blueprint(USER_API_BLUEPRINT, url_prefix=u"/api/users")
if __name__ == "__main__":
    APP.run(threaded=True, host=u"0.0.0.0", debug=True)