mirror of https://gitlab.com/pamhyr/pamhyr2
Checker: Add study geometry checker.
parent
acaa369e77
commit
dccf1a82ac
|
|
@ -1,5 +1,7 @@
|
|||
# -*- 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(),
|
||||
|
|
|
|||
Loading…
Reference in New Issue