weeeee

etc/database/screen-capture-annotate/screen-capture-annotate.py

@@ -83,7 +83,6 @@ def aggregate_string_replacement(key, value, base_string):
 class ScreenCaptureDatasets(contextlib.ExitStack):
     def __init__(self):
         super().__init__()
-        self.captured_datasets = {}
 
     def __enter__(self):
         super().__enter__()
@@ -177,6 +176,7 @@ def screencapture_dataset(self, unit_type, unit, unit_annotations, dataset):
 class RosalutionAnalysis():
     def __init__(self, analysis_name):
         self.analysis_name = analysis_name
+        self.captured_datasets = {}
 
     def get_genomic_units(self):
         response = requests.get(f"{config['ROSALUTION_API_URL']}analysis/{self.analysis_name}/genomic_units", verify=False)
@@ -197,8 +197,9 @@ def capture_analysis(self, capture):
             for unit in genomic_units[genomic_unit_type]:
                 unit_annotations = self.get_annotations(genomic_unit_type, unit)
                 for genomic_unit_dataset in DATASETS[genomic_unit_type]:
-                    capture.screencapture_dataset(genomic_unit_type, unit, unit_annotations, genomic_unit_dataset)
-
+                    captured_dataset = capture.screencapture_dataset(genomic_unit_type, unit, unit_annotations, genomic_unit_dataset)
+                    if captured_dataset:
+                        captured_dataset[()]
 rosalution_analyses = sys.argv[1:]
 
 print("Capturing Rosalution Analyses")