From 71e9f2b82ba9e461ca52b37c34fe554f687ab85e Mon Sep 17 00:00:00 2001
From: Jan Grewe <jan.grewe@g-node.org>
Date: Mon, 16 Sep 2019 19:55:28 +0200
Subject: [PATCH] [Database] add some helper functions

---
 database.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/database.py b/database.py
index a4468e6..5a328ea 100644
--- a/database.py
+++ b/database.py
@@ -63,6 +63,16 @@ class Dataset(dj.Manual):
             sane = False
         return sane
 
+    @property
+    def cells(self, restrictions:dict=None):
+        cs = (Cell & (CellDatasetMap & self) & restrictions).fetch()
+        return cs
+
+    @property
+    def subjects(self, restrictions:dict=None):
+        subjs = (Subject & (SubjectDatasetMap & self) & restrictions).fetch()
+        return subjs
+
 
 @schema
 class Subject(dj.Manual):
@@ -98,6 +108,14 @@ class Subject(dj.Manual):
         self.insert1(inserts, skip_duplicates=True)
         nix_file.close()
 
+    @staticmethod
+    def subjects(species=None):
+        subjs = []
+        if species:
+            subjs = (Subject & "species like '\%%s\%'"%species).fetch()
+        else:
+            subjs = (Subject & True).fetch()
+        return subjs
 
 @schema
 class SubjectDatasetMap(dj.Manual):