Merge pull request #137 from kkaris/src-count-patch

Add source count dumping option, run by default
gyorilab · Oct 8, 2020 · 80ea8aa · 80ea8aa
2 parents 00e24fc + ceb5043
commit 80ea8aa
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/indra_db/managers/dump_manager.py b/indra_db/managers/dump_manager.py
@@ -151,13 +151,17 @@ def __init__(self, db_label='primary', use_principal=False, **kwargs):
         self.use_principal = use_principal
         super(Sif, self).__init__(db_label, **kwargs)
 
-    def dump(self, continuing=False):
+    def dump(self, continuing=False, include_src_counts=True):
         if self.use_principal:
             ro = get_db(self.db_label)
         else:
             ro = get_ro(self.db_label)
         s3_path = self.get_s3_path()
-        dump_sif(s3_path, ro=ro)
+        if include_src_counts:
+            srcc = SourceCount()
+            dump_sif(s3_path, src_count_file=srcc.get_s3_path(), ro=ro)
+        else:
+            dump_sif(s3_path, ro=ro)
 
 
 class Belief(Dumper):