Commit dccf1a82 authored by Pierre-Antoine Rouby's avatar Pierre-Antoine Rouby
Browse files

Checker: Add study geometry checker.

Showing with 58 additions and 1 deletion
+58 -1
# -*- coding: utf-8 -*-
import time
from PyQt5.QtCore import QCoreApplication
from Checker.Checker import AbstractModelChecker, STATUS
......@@ -26,7 +28,8 @@ class StudyNetworkReachChecker(AbstractModelChecker):
self._summary = "no_river_found"
return False
if len(river.edges()) == 0:
edges = list(filter(lambda e: e.is_enable(), river.edges()))
if len(edges) == 0:
self._status = STATUS.ERROR
self._summary = "no_reach_defined"
return False
......@@ -35,6 +38,48 @@ class StudyNetworkReachChecker(AbstractModelChecker):
self._status = STATUS.OK
return True
class StudyGeometryChecker(AbstractModelChecker):
def __init__(self):
super(StudyGeometryChecker, self).__init__()
self._name = _translate("Checker", "Study geometry checker")
self._description = _translate("Checker", "Check if exists geometry for each reach of study")
self._reachs = []
def run(self, study):
ok = True
nerror = 0
summary = "ok"
status = STATUS.OK
if study is None:
self._status = STATUS.ERROR
self._summary = "invalid_study"
return False
river = study.river
if river is None:
self._status = STATUS.ERROR
self._summary = "no_river_found"
return False
edges = list(filter(lambda e: e.is_enable(), river.edges()))
if len(edges) == 0:
self._status = STATUS.ERROR
self._summary = "no_reach_defined"
return False
for edge in edges:
if edge.is_enable() and len(edge.reach.profiles) < 3:
summary = f"no_geometry_defined"
status = STATUS.ERROR
ok = False
self._reachs.append(edge)
self._summary = summary
self._status = status
return ok
class DummyOK(AbstractModelChecker):
def __init__(self):
......@@ -44,8 +89,11 @@ class DummyOK(AbstractModelChecker):
self._description = _translate("Checker", "Dummy ok")
def run(self, study):
time.sleep(1)
self._summary = "ok"
self._status = STATUS.OK
return True
......@@ -57,8 +105,11 @@ class DummyWARNING(AbstractModelChecker):
self._description = _translate("Checker", "Dummy warning")
def run(self, study):
time.sleep(1)
self._summary = "Warning detected"
self._status = STATUS.WARNING
return True
class DummyERROR(AbstractModelChecker):
......@@ -69,6 +120,8 @@ class DummyERROR(AbstractModelChecker):
self._description = _translate("Checker", "Dummy error")
def run(self, study):
time.sleep(1)
self._summary = "Error detected"
self._status = STATUS.ERROR
return True
......@@ -53,6 +53,9 @@ class Edge(object):
def name(self):
return self._name if self._name != "" else f"{self.node1.name} -> {self.node2.name}"
def is_enable(self):
return self._enable
def enable(self, enable=True):
self._enable = enable
self._status.modified()
......
......@@ -36,6 +36,7 @@ class Study(Serializable):
def checkers(cls):
lst = [
StudyNetworkReachChecker(),
StudyGeometryChecker(),
DummyOK(),
DummyWARNING(),
DummyERROR(),
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment