Make configurable

Author: haneslinger

seed/audit_template/audit_template.py

@@ -276,6 +276,7 @@ def build_xml(self, state, report_type, display_field):
             return None, messages
 
         view = state.propertyview_set.first()
+        org = view.property.organization
 
         gfa = state.gross_floor_area
         if isinstance(gfa, int):
@@ -335,15 +336,15 @@ def build_xml(self, state, report_type, display_field):
                             ),
                         )
                     ),
-                    *_build_measures_element(em, view.property),
+                    *([] if not org.audit_template_export_measures else _build_measures_element(em, view.property)),
                     em.Reports(
                         em.Report(
                             em.Scenarios(
                                 {},
                                 em.Scenario(
                                     {"ID": "ScenarioType-69817879941680"},
-                                    *_build_resource_uses(em, view.property),
-                                    *build_time_series_data(em, view.property),
+                                    *([] if not org.audit_template_export_meters else _build_resource_uses(em, view.property)),
+                                    *([] if not org.audit_template_export_meters else build_time_series_data(em, view.property)),
                                 ),
                             ),
                             {"ID": "ReportType-69909846999993"},

seed/lib/superperms/orgs/migrations/0040_auto_20250203_1332.py

@@ -0,0 +1,22 @@
+# Generated by Django 3.2.25 on 2025-02-03 21:32
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("orgs", "0039_alter_organization_ubid_threshold"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="organization",
+            name="audit_template_export_measures",
+            field=models.BooleanField(default=True),
+        ),
+        migrations.AddField(
+            model_name="organization",
+            name="audit_template_export_meters",
+            field=models.BooleanField(default=True),
+        ),
+    ]

seed/lib/superperms/orgs/models.py

@@ -300,6 +300,8 @@ class Meta:
     audit_template_city_id = models.IntegerField(blank=True, null=True)
     audit_template_conditional_import = models.BooleanField(default=True)
     audit_template_sync_enabled = models.BooleanField(default=False)
+    audit_template_export_meters = models.BooleanField(default=True)
+    audit_template_export_measures = models.BooleanField(default=True)
 
     # Salesforce Functionality
     salesforce_enabled = models.BooleanField(default=False)

seed/static/seed/partials/organization_settings.html

@@ -251,6 +251,28 @@ <h3 translate>Audit Template Submission Status</h3>
         </div>
       </div>
 
+      <div class="section_content_container">
+        <div class="section_content with_padding">
+          <h3 translate>Audit Template Fields</h3>
+          <form class="form-horizontal" role="form">
+            <div class="form-group">
+              <div class="col-sm-5">
+                <div class="multi-select-container">
+                  <div class="col-sm-5">
+                    <input type="checkbox" class="form-control" style="width: 34px" ng-model="org.audit_template_export_meters" ng-disabled="::!auth.requires_owner" />
+                    <h3 translate style="padding-left: 5px; align-content: center;">meters</h3>
+                  </div>
+                  <div class="col-sm-5">
+                    <input type="checkbox" class="form-control" style="width: 34px" ng-model="org.audit_template_export_measures" ng-disabled="::!auth.requires_owner" />
+                    <h3 translate style="padding-left: 5px; align-content: center;">measures</h3>
+                  </div>
+                </div>
+              </div>
+            </div>
+          </form>
+        </div>
+      </div>
+
       <div class="section_content_container">
         <div class="section_content with_padding">
           <h3 translate>Conditional Import</h3>

seed/tests/test_audit_template.py

@@ -17,6 +17,7 @@
 # from seed.audit_template.audit_template import build_xml
 from seed.audit_template.audit_template import AuditTemplate
 from seed.landing.models import SEEDUser as User
+from seed.lib.tkbl.tkbl import SCOPE_ONE_EMISSION_CODES
 from seed.models import Meter, MeterReading, Uniformat
 from seed.test_helpers.fake import (
     FakeCycleFactory,
@@ -240,8 +241,6 @@ def test_build_xml_from_property_with_meter_readings(self):
         self.assertEqual(float(reading.text), 123)
 
     def test_build_xml_from_property_with_measures(self):
-        from seed.lib.tkbl.tkbl import SCOPE_ONE_EMISSION_CODES
-
         # Set Up
         self.element1 = self.element_factory.get_element(
             property=self.view1.property, code=Uniformat.objects.filter(code__in=SCOPE_ONE_EMISSION_CODES)[1]
@@ -285,6 +284,54 @@ def test_build_xml_from_property_with_measures(self):
             {m.find("auc:MeasureName", namespaces=tree.nsmap).text for m in chilled_water_hot_water_and_steam_distribution_systems},
         )
 
+    def test_build_xml_from_property_with_measures_and_meters_org_settings(self):
+        # Set Up
+        self.org.audit_template_export_meters = False
+        self.org.audit_template_export_measures = False
+        self.org.save()
+
+        self.element1 = self.element_factory.get_element(
+            property=self.view1.property, code=Uniformat.objects.filter(code__in=SCOPE_ONE_EMISSION_CODES)[1]
+        )
+        self.element2 = self.element_factory.get_element(
+            property=self.view1.property, code=Uniformat.objects.filter(code__in=SCOPE_ONE_EMISSION_CODES)[2]
+        )
+
+        self.meter = Meter.objects.create(property_id=self.view1.property_id, type=Meter.ELECTRICITY_GRID)
+        MeterReading.objects.create(
+            start_time=datetime(2019, 1, 1, 0, 0, 0, tzinfo=tz.utc),
+            end_time=datetime(2019, 2, 1, 0, 0, 0, tzinfo=tz.utc),
+            reading=123,
+            meter_id=self.meter.id,
+            conversion_factor=1,
+        )
+
+        # Action
+        at = AuditTemplate(self.org.id)
+        response = at.build_xml(self.state1, "Demo City Report", self.state1.pm_property_id)
+
+        # Assert
+        # # is tree
+        self.assertEqual(tuple, type(response))
+        tree = etree.XML(response[0])
+
+        # # has 1 scenario
+        scenarios = tree.findall("auc:Facilities/auc:Facility/auc:Reports/auc:Report/auc:Scenarios/auc:Scenario", namespaces=tree.nsmap)
+        self.assertEqual(1, len(scenarios))
+        scenario = scenarios[0]
+
+        # # scenario has 0 meters
+        meters = scenario.findall("auc:ResourceUses/auc:ResourceUse", namespaces=tree.nsmap)
+        self.assertEqual(0, len(meters))
+
+        # # scenario has 0 meter readings
+        meter_readings = scenario.findall("auc:TimeSeriesData/auc:TimeSeries", namespaces=tree.nsmap)
+        self.assertEqual(0, len(meter_readings))
+
+        # # tree has 0 measures
+        measures = tree.findall("auc:Facilities/auc:Facility/auc:Measures", namespaces=tree.nsmap)
+        self.assertEqual(0, len(measures))
+
     @mock.patch("requests.request")
     def test_export_to_audit_template(self, mock_request):
         """

seed/views/v3/organizations.py

@@ -153,6 +153,8 @@ def _dict_org(request, organizations):
             "audit_template_user": o.audit_template_user,
             "audit_template_password": decrypt(o.audit_template_password)[0] if o.audit_template_password else "",
             "audit_template_city_id": o.audit_template_city_id,
+            "audit_template_export_meters": o.audit_template_export_meters,
+            "audit_template_export_measures": o.audit_template_export_measures,
             "audit_template_conditional_import": o.audit_template_conditional_import,
             "audit_template_report_type": o.audit_template_report_type,
             "audit_template_status_types": o.audit_template_status_types,
@@ -657,6 +659,14 @@ def warn_bad_units(kind, unit_string):
         if audit_template_city_id != org.audit_template_city_id:
             org.audit_template_city_id = audit_template_city_id
 
+        audit_template_export_meters = posted_org.get("audit_template_export_meters", False)
+        if audit_template_export_meters != org.audit_template_export_meters:
+            org.audit_template_export_meters = audit_template_export_meters
+
+        audit_template_export_measures = posted_org.get("audit_template_export_measures", False)
+        if audit_template_export_measures != org.audit_template_export_measures:
+            org.audit_template_export_measures = audit_template_export_measures
+
         audit_template_conditional_import = posted_org.get("audit_template_conditional_import", False)
         if audit_template_conditional_import != org.audit_template_conditional_import:
             org.audit_template_conditional_import = audit_template_conditional_import