Merge pull request #199 from HarikaBishai/feature/automate-cycle-update

Added current cycle automation changes to cycles sync process

Author: padraic-shafer

src/nsls2api/services/facility_service.py

@@ -89,6 +89,21 @@ async def facility_cycles(facility: str) -> Optional[list[str]]:
     cycle_list = [c.name for c in cycles if c.name is not None]
     return cycle_list
 
+async def facility_cycle_by_date(facility: FacilityName, date: datetime.datetime) -> Optional[Cycle]:
+    """
+    Find the cycle for a facility that contains the given date.
+
+    :param facility: The facility name (FacilityName).
+    :param date: The date to search for (datetime).
+    :return: The matching Cycle or None.
+    """
+    cycle = await Cycle.find_one(
+        Cycle.facility == facility.value,
+        Cycle.start_date <= date,
+        Cycle.end_date >= date
+    )
+    return cycle if cycle else None
+
 
 async def facility_by_pass_id(pass_user_facility_id: str) -> Optional[Facility]:
     """

src/nsls2api/services/sync_service.py

@@ -81,14 +81,11 @@ async def worker_synchronize_dataadmins(skip_beamlines=False) -> None:
             f"Beamline Data Admin permissions synchronized in {time_taken.total_seconds():,.2f} seconds"
         )
 
-
 async def worker_synchronize_cycles_from_pass(
     facility_name: FacilityName = FacilityName.nsls2,
 ) -> None:
     """
-    This method synchronizes the cycles for a facility from PASS.
-
-    :param facility_name: The facility name (FacilityName).
+    Synchronize cycles for a facility from PASS and set the current cycle.
     """
     start_time = datetime.datetime.now()
 
@@ -136,7 +133,7 @@ async def worker_synchronize_cycles_from_pass(
             response_type=UpdateResponse.NEW_DOCUMENT,
         )
 
-        # Now let's update the list of proposals for this cycle
+        # Update proposals for this cycle
         proposals_list = await pass_service.get_proposals_allocated_by_cycle(
             cycle.name, facility=facility_name
         )
@@ -147,6 +144,23 @@ async def worker_synchronize_cycles_from_pass(
             updated_cycle.last_updated = datetime.datetime.now()
             await updated_cycle.save()
 
+    # --- Set current operating cycle from today's date ---
+    today = datetime.datetime.now()
+    found_cycle = await facility_service.facility_cycle_by_date(facility_name, today)
+    if not found_cycle:
+        logger.warning(
+            f"No cycle found for today's date in facility {facility_name}"
+        )
+    else:
+        logger.info(
+            f"Found cycle {found_cycle.name} for today's date in facility {facility_name}"
+        )
+        # Set the current operating cycle for the facility
+        await facility_service.set_current_operating_cycle(facility_name, found_cycle.name)
+        logger.info(
+            f"Set current operating cycle for {facility_name} to {found_cycle.name}"
+        )
+
     time_taken = datetime.datetime.now() - start_time
     logger.info(
         f"Cycle information (for {facility_name}) synchronized in {time_taken.total_seconds():,.2f} seconds"