Skip to content
Snippets Groups Projects
Commit 332eb399 authored by Peter Lünenschloß's avatar Peter Lünenschloß
Browse files

adde getflags functionality to dmp and baseflagger / changed isFlagged...

adde getflags functionality to dmp and baseflagger / changed isFlagged structure (now internally calls getFlags)
parent 126a8cd1
No related branches found
No related tags found
No related merge requests found
......@@ -99,6 +99,9 @@ class BaseFlagger:
flag = self.GOOD
return pd.notnull(flags) & cp(flags, self._checkFlag(flag))
def getFlags(self, flags):
return flags
def _checkFlag(self, flag):
if flag not in self.flags:
raise ValueError(
......
......@@ -71,9 +71,13 @@ class DmpFlagger(BaseFlagger):
return flags.values
def isFlagged(self, flags, flag=None, comparator=">"):
flagcol = self.getFlags(flags)
return super().isFlagged(flagcol, flag, comparator)
def getFlags(self, flags):
flags = self._reduceColumns(flags)
flagcol = flags.loc[:, FlagFields.FLAG].squeeze()
return super().isFlagged(flagcol, flag, comparator)
return super().getFlags(flagcol)
def _reduceColumns(self, flags):
if set(flags.columns) == set(self.flag_fields):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment