fix: ems export command

did that ever work?

contxt/cli/commands/ems.py

@@ -129,11 +129,11 @@ def usage(
 
 
 @ems.command()
-@click.argument("facility_ids", nargs=-1)
+@click.argument("facility_ids", nargs=-1, type=click.INT)
 @click.option(
     "--include",
     default="all",
-    callback=csv_callback(options=["spend", "usage"]),
+    callback=csv_callback(options=["mains", "usage"]),
     help="Data to export",
 )
 @click.option("--start", type=click.DateTime(), default=LAST_WEEK.isoformat(), help="Start time")
@@ -151,9 +151,9 @@ def export(
     """Export data for facilities"""
     with click.progressbar(
         facility_ids, label="Downloading data", item_show_func=lambda f: f"Facility {f}" if f else ""
-    ) as facility_ids_:
+    ):
         for facility in clients.nionic.get_facilities():
-            if facility.id not in facility_ids_:
+            if facility.id not in facility_ids:
                 continue
             fpath = output / facility.slug
 

contxt/services/nionic.py

@@ -1,9 +1,10 @@
-from typing import Any, Iterable, List
+from typing import Any, Iterable, List, Optional
 
 from sgqlc.endpoint.requests import RequestsEndpoint
 
 from contxt.generated.nionic_queries import Operations as nionic_operations
 from contxt.generated.nionic_schema import Facility, MainService, MetricLabel
+from contxt.models.ems import ResourceType
 from contxt.models.iot import Window
 from contxt.services.api import ApiEnvironment, BaseGraphService
 
@@ -60,7 +61,9 @@ def get_metric_data(self, label: str, facility_id: int, start: str) -> List[Any]
         resp = self.query(query, {"facilityId": int(facility_id), "label": label, "from": start})
         return resp["facility"]["metricData"]["nodes"]
 
-    def get_main_services(self, facility_id: int, resource_type) -> List[MainService]:
+    def get_main_services(
+        self, facility_id: int, resource_type: Optional[ResourceType] = None
+    ) -> List[MainService]:
         op = nionic_operations.query.main_services
         filterValues = {"facilityId": int(facility_id)}
         return [