diff --git a/database/forest_models.py b/database/forest_models.py index 9a515ae36..28059cbbf 100644 --- a/database/forest_models.py +++ b/database/forest_models.py @@ -345,3 +345,23 @@ class Meta: constraints = [ models.UniqueConstraint(fields=['date', 'participant'], name="unique_summary_statistic") ] + + @classmethod + def beiwe_fields(cls): + return [field.name for field in cls._meta.get_fields() if field.name.startswith("beiwe_")] + + @classmethod + def jasmine_fields(cls): + return [field.name for field in cls._meta.get_fields() if field.name.startswith("jasmine_")] + + @classmethod + def willow_fields(cls): + return [field.name for field in cls._meta.get_fields() if field.name.startswith("willow_")] + + @classmethod + def sycamore_fields(cls): + return [field.name for field in cls._meta.get_fields() if field.name.startswith("sycamore_")] + + @classmethod + def oak_fields(cls): + return [field.name for field in cls._meta.get_fields() if field.name.startswith("oak_")]