Handle reading encrypted Energy Balances (#551)

* Updated the energy-balance reader to check for .encrypted.csv files as well * Minor fix in check for existing file --------- Co-authored-by: Kyra de Haan <[email protected]>
quintel · Oct 18, 2024 · 5a11018 · 5a11018
1 parent e59834c
commit 5a11018
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/app/lib/dataset_source/entso/file.rb b/app/lib/dataset_source/entso/file.rb
@@ -28,7 +28,16 @@ def self.energy_balance(csv_table)
       end
 
       def self.from_dataset(dataset)
-        new(dataset.geo_id, Rails.root.join('data', 'datasets', "#{dataset.geo_id}_energy_balance_enriched.csv"))
+        base_path = Rails.root.join('data', 'datasets', "#{dataset.geo_id}_energy_balance_enriched")
+        paths = ["#{base_path}.csv", "#{base_path}.encrypted.csv"]
+
+        path = paths.find { |p| ::File.exist?(p) }
+
+        unless path
+          raise "Neither the CSV nor the encrypted file exists for dataset with geo_id: #{dataset.geo_id}"
+        end
+
+        new(dataset.geo_id, path)
       end
 
       def initialize(geo_id, path)