feat(risk-matrix): use plotly

seantis · Jan 23, 2024 · 733d61c · 733d61c
1 parent efc2f99
commit 733d61c
Show file tree

Hide file tree

Showing 6 changed files with 253 additions and 263 deletions.
diff --git a/src/riskmatrix/models/__init__.py b/src/riskmatrix/models/__init__.py
@@ -9,7 +9,7 @@
 from .asset import Asset
 from .organization import Organization
 from .risk import Risk
-from .risk_assessment import RiskAssessment
+from .risk_assessment import RiskAssessment, RiskMatrixAssessment
 from .risk_catalog import RiskCatalog
 from .risk_category import RiskCategory
 from .user import User
@@ -60,5 +60,6 @@ def includeme(config: 'Configurator') -> None:
     'RiskAssessment',
     'RiskCatalog',
     'RiskCategory',
+    'RiskMatrixAssessment',
     'User'
 )
diff --git a/src/riskmatrix/models/risk_assessment.py b/src/riskmatrix/models/risk_assessment.py
@@ -146,3 +146,7 @@ def __acl__(self) -> list['ACL']:
         return [
             (Allow, f'org_{self.risk.organization_id}', ['view']),
         ]
+
+
+class RiskMatrixAssessment(RiskAssessment):
+    nr: Mapped[int | None] = mapped_column(default=None)
diff --git a/src/riskmatrix/static/__init__.py b/src/riskmatrix/static/__init__.py
@@ -58,3 +58,4 @@ def css(
 )
 datatable_js = js('datatables_custom.js', depends=[datatable_bootstrap])
 xhr_edit_js = js('xhr_edit.js', depends=[datatable_js])
+plotly_js = js('plotly.min.js', depends=[datatable_js])
diff --git a/src/riskmatrix/static/js/plotly.min.js b/src/riskmatrix/static/js/plotly.min.js