Added route to watch used storage and max storage

Author: t33nsy

app/mongo_odm.py

@@ -11,7 +11,7 @@
 from gridfs import GridFSBucket, NoFile
 from pymodm import connect
 from pymodm.connection import _get_db
-from pymodm.errors import ValidationError
+from pymodm.errors import ValidationError, DoesNotExist
 from pymodm.files import GridFSStorage
 from pymongo import ReturnDocument
 from pymongo.errors import CollectionInvalid
@@ -81,7 +81,7 @@ def get_file(self, file_id):
     def _get_or_create_storage_meta(self):
         try:
             return StorageMeta.objects.get({})
-        except StorageMeta.DoesNotExist:
+        except DoesNotExist:
             meta = StorageMeta(used_size=0).save()
             return meta
 
@@ -99,7 +99,7 @@ def update_storage_size(self, deltasize):
         meta.save()
 
     def check_storage_limit(self, new_file_size):
-        max_size = int(Config.c.constants.storage_max_size_bytes)*1024*1024
+        max_size = int(Config.c.constants.storage_max_size_mbytes)*1024*1024
         current_size = self.get_used_storage_size()
         if current_size + new_file_size > max_size:
             logger.warning('current_size = {}, new file size = {}'.format(current_size, new_file_size))

app/routes/capacity.py

@@ -0,0 +1,25 @@
+from flask import Blueprint, render_template
+from app.mongo_odm import DBManager
+from app.config import Config
+from app.root_logger import get_root_logger
+from app.lti_session_passback.auth_checkers import is_admin
+
+logger = get_root_logger()
+
+routes_capacity = Blueprint(
+    'routes_capacity', __name__, url_prefix='/capacity')
+
+@routes_capacity.route('/', methods=['GET'])
+def storage_capacity():
+    if not is_admin():
+        return {}, 404
+    current_size = DBManager().get_used_storage_size()
+    max_size = int(Config.c.constants.storage_max_size_mbytes)*1024*1024
+    ratio = current_size / max_size
+    return render_template(
+        'capacity.html',
+        size=round(current_size / 1, 2),
+        max_size=round(max_size / 1, 2),
+        ratio=round(ratio * 100, 1)
+    )
+    

app/templates/capacity.html

@@ -0,0 +1,19 @@
+{% extends 'base.html' %}
+
+{% block content %}
+<div class="container mt-4">
+    <div class="card">
+        <div class="card-body">
+            <h4 class="card-title">Загруженность Базы Данных</h4>
+            <p><strong>Использовано:</strong> {{ size }} байт</p>
+            <p><strong>Максимум:</strong> {{ max_size }} байт</p>
+            <p><strong>Заполнено:</strong> {{ ratio }}%</p>
+            <div class="progress">
+                <div class="progress-bar" role="progressbar" style="width: {{ ratio }}%;" aria-valuenow="{{ ratio }}" aria-valuemin="0" aria-valuemax="100">
+                    {{ ratio }}
+                </div>
+            </div>
+        </div>
+    </div>
+</div>
+{% endblock %}

app/web_speech_trainer.py

@@ -30,6 +30,7 @@
 from app.routes.presentations import routes_presentations
 from app.routes.trainings import routes_trainings
 from app.routes.version import routes_version
+from app.routes.capacity import routes_capacity
 from app.status import PassBackStatus, TrainingStatus
 from app.training_manager import TrainingManager
 from app.utils import ALLOWED_EXTENSIONS, DEFAULT_EXTENSION
@@ -53,6 +54,7 @@
 app.register_blueprint(routes_presentations)
 app.register_blueprint(routes_trainings)
 app.register_blueprint(routes_version)
+app.register_blueprint(routes_capacity)
 
 logger = get_root_logger(service_name='web')
 

app_conf/config.ini

@@ -5,7 +5,7 @@ lti_consumer_key=secretconsumerkey
 lti_consumer_secret=supersecretconsumersecret
 version_file=VERSION.json
 backup_path=../dump/database-dump/
-storage_max_size_bytes=5000
+storage_max_size_mbytes=20000
 
 [mongodb]
 url=mongodb://db:27017/