Merge pull request #62 from TaleLin/feat/access-log

Feat/access log

Author: colorful3

Pipfile

@@ -12,7 +12,7 @@ Flask = "==1.0.2"
 Flask-SQLAlchemy = "==2.3.2"
 Flask-WTF = "==0.14.2"
 Flask-Cors = "==2.1.0"
-Lin-CMS = "==0.1.1b3"
+Lin-CMS = "==0.1.1b4"
 
 [dev-packages]
 pytest = "*"

app/app.py

@@ -2,8 +2,10 @@
     :copyright: © 2019 by the Lin team.
     :license: MIT, see LICENSE for more details.
 """
+import json
+import time
 
-from flask import Flask
+from flask import Flask, request, g
 from flask_cors import CORS
 from lin import Lin
 
@@ -25,14 +27,53 @@ def create_tables(app):
         db.create_all()
 
 
+def register_before_request(app):
+    @app.before_request
+    def request_cost_time():
+        g.request_start_time = time.time()
+        g.request_time = lambda: "%.5f" % (time.time() - g.request_start_time)
+
+
+def register_after_request(app):
+    @app.after_request
+    def log_response(resp):
+        log_config = app.config.get('LOG')
+        if not log_config['REQUEST_LOG']:
+            return resp
+        message = '[%s] -> [%s] from:%s costs:%.3f ms' % (
+            request.method,
+            request.path,
+            request.remote_addr,
+            float(g.request_time()) * 1000
+        )
+        if log_config['LEVEL'] == 'INFO':
+            app.logger.info(message)
+        elif log_config['LEVEL'] == 'DEBUG':
+            req_body = '{}'
+            try:
+                req_body = request.get_json() if request.get_json() else {}
+            except:
+                pass
+            message += " data:{\n\tparam: %s, \n\tbody: %s\n} " % (
+                json.dumps(request.args, ensure_ascii=False),
+                req_body
+            )
+            app.logger.debug(message)
+        return resp
+
+
 def create_app(register_all=True):
     app = Flask(__name__, static_folder='./assets')
     app.config.from_object('app.config.setting')
     app.config.from_object('app.config.secure')
+    app.config.from_object('app.config.log')
     if register_all:
         register_blueprints(app)
         Lin(app)
+        register_before_request(app)
+        register_after_request(app)
         apply_cors(app)
         # 创建所有表格
         create_tables(app)
+
     return app

app/config/log.py

@@ -0,0 +1,11 @@
+"""
+    :copyright: © 2019 by the Lin team.
+    :license: MIT, see LICENSE for more details.
+"""
+LOG = {
+    'LEVEL': 'DEBUG',
+    'DIR': 'logs',
+    'SIZE_LIMIT': 1024 * 1024 * 5,
+    'REQUEST_LOG': True,
+    'FILE': True
+}

app/extensions/file/local_uploader.py

@@ -9,6 +9,7 @@ class LocalUploader(Uploader):
 
     def upload(self):
         ret = []
+        self.mkdir_if_not_exists()
         site_domain = current_app.config.get('SITE_DOMAIN')\
             if current_app.config.get('SITE_DOMAIN') else 'http://127.0.0.1:5000'
         for single in self._file_storage:

requirements.txt

@@ -10,7 +10,7 @@ Flask-WTF==0.14.2
 idna==2.6
 itsdangerous==1.1.0
 Jinja2==2.10
-Lin-CMS==0.1.1b3
+Lin-CMS==0.1.1b4
 MarkupSafe==1.1.1
 pipfile==0.0.2
 PyJWT==1.7.1