From bf91cd1f5e2af910951f06dc003b685cdc8c4976 Mon Sep 17 00:00:00 2001
From: Tor-Einar Skog <tor-einar.skog@nibio.no>
Date: Tue, 11 Jun 2019 10:38:30 +0200
Subject: [PATCH] Adding multiple orgs support for forecasts list

---
 forecasts/models.py | 46 +++++++++++++++++++++++++++------------------
 1 file changed, 28 insertions(+), 18 deletions(-)

diff --git a/forecasts/models.py b/forecasts/models.py
index 42b2a96c..e62bb8fd 100755
--- a/forecasts/models.py
+++ b/forecasts/models.py
@@ -230,24 +230,34 @@ class ForecastConfiguration:
             for crop_organism_id in crop_organism_ids:
                 crop_organism_id_paramstring += "&cropOrganismId=%s" % crop_organism_id
         
-        print "%s://%s/rest/organizationforecastconfigurations/%s?from=%s-01-01&to=%s-12-31%s" % (
-                                                                              settings.VIPSLOGIC_PROTOCOL,
-                                                                              settings.VIPSLOGIC_SERVER_NAME, 
-                                                                              settings.VIPS_ORGANIZATION_ID,
-                                                                              season,
-                                                                              season,
-                                                                              crop_organism_id_paramstring
-                                                                              )
-        
-        request_result = requests.get("%s://%s/rest/organizationforecastconfigurations/%s?from=%s-01-01&to=%s-12-31%s" % (
-                                                                              settings.VIPSLOGIC_PROTOCOL,
-                                                                              settings.VIPSLOGIC_SERVER_NAME, 
-                                                                              settings.VIPS_ORGANIZATION_ID,
-                                                                              season,
-                                                                              season,
-                                                                              crop_organism_id_paramstring
-                                                                              )
-                                      )
+        request_result = None
+        if settings.INCLUDE_ORGANIZATION_IDS == None:
+            print "%s://%s/rest/organizationforecastconfigurations/%s?from=%s-01-01&to=%s-12-31%s" % (
+                                                                                  settings.VIPSLOGIC_PROTOCOL,
+                                                                                  settings.VIPSLOGIC_SERVER_NAME, 
+                                                                                  settings.VIPS_ORGANIZATION_ID,
+                                                                                  season,
+                                                                                  season,
+                                                                                  crop_organism_id_paramstring
+                                                                                  )
+            
+            request_result = requests.get("%s://%s/rest/organizationforecastconfigurations/%s?from=%s-01-01&to=%s-12-31%s" % (
+                                                                                  settings.VIPSLOGIC_PROTOCOL,
+                                                                                  settings.VIPSLOGIC_SERVER_NAME, 
+                                                                                  settings.VIPS_ORGANIZATION_ID,
+                                                                                  season,
+                                                                                  season,
+                                                                                  crop_organism_id_paramstring
+                                                                                  )
+                                          )
+        else:
+            request_result = requests.get("%s://%s/rest/forecastconfigurationsincludeorgs/%s?includeOrganizationIds=%s" % (
+                                                                                  settings.VIPSLOGIC_PROTOCOL,
+                                                                                  settings.VIPSLOGIC_SERVER_NAME, 
+                                                                                  settings.VIPS_ORGANIZATION_ID,
+                                                                                  settings.INCLUDE_ORGANIZATION_IDS
+                                                                                  )
+                                          )
         return request_result
     
     @staticmethod
-- 
GitLab