From 67f534a6f6291b14418165491ea15eb277d45172 Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <pierre-antoine.rouby@inrae.fr>
Date: Tue, 19 Mar 2024 12:00:44 +0100
Subject: [PATCH] Checker: Add related module(s).

---
 src/Checker/Checker.py | 3 +++
 src/Checker/Mage.py    | 3 +++
 src/Checker/Study.py   | 9 +++++++--
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/Checker/Checker.py b/src/Checker/Checker.py
index d0bff2c4..4591beda 100644
--- a/src/Checker/Checker.py
+++ b/src/Checker/Checker.py
@@ -19,6 +19,8 @@
 import logging
 
 from enum import Enum
+
+from Modules import Modules
 from Model.Except import NotImplementedMethodeError
 
 logger = logging.getLogger()
@@ -55,6 +57,7 @@ class AbstractModelChecker(object):
     def __init__(self):
         self._name = ""
         self._description = ""
+        self._modules = Modules.NONE
 
         self._status = STATUS.UNKNOWN
         self._summary = "Unknown"
diff --git a/src/Checker/Mage.py b/src/Checker/Mage.py
index 5676dce8..260431a5 100644
--- a/src/Checker/Mage.py
+++ b/src/Checker/Mage.py
@@ -44,6 +44,7 @@ class MageNetworkGraphChecker(AbstractModelChecker):
             "Checker", f"Mage network graph {mode} checker")
         self._description = _translate(
             "Checker", "Check if the network graph is valid")
+        self._modules = Modules.NETWORK
 
     @timer
     def _connectivity(self, summary, status, graph):
@@ -208,6 +209,8 @@ class MageGeometryGuideLineChecker(StudyGeometryChecker):
             "Check if exists geometry guide line is correctly defined \
 for each reaches of the study"
         )
+        self._modules = Modules.GEOMETRY
+
         self._reachs = []
 
     def run(self, study):
diff --git a/src/Checker/Study.py b/src/Checker/Study.py
index ae94bd7b..01dbc309 100644
--- a/src/Checker/Study.py
+++ b/src/Checker/Study.py
@@ -31,7 +31,9 @@ class StudyNetworkReachChecker(AbstractModelChecker):
 
         self._name = _translate("Checker", "Study network reach checker")
         self._description = _translate(
-            "Checker", "Check if exists at least one reach for study")
+            "Checker", "Check if exists at least one reach for study"
+        )
+        self._modules = Modules.NETWORK
 
     def run(self, study):
         if study is None:
@@ -62,7 +64,10 @@ class StudyGeometryChecker(AbstractModelChecker):
 
         self._name = _translate("Checker", "Study geometry checker")
         self._description = _translate(
-            "Checker", "Check if exists geometry for each reach of study")
+            "Checker", "Check if exists geometry for each reach of study"
+        )
+        self._modules = Modules.GEOMETRY
+
         self._reachs = []
 
     def run(self, study):
-- 
GitLab