Add ephem_target2night in Agent.py

Alexis Koralewski
1 parent 3be8b2fc
Showing 1 changed file with 41 additions and 1 deletions Show diff stats
src/core/pyros_django/majordome/agent/Agent.py
@@ -153,6 +153,12 @@ import config.old_config as config_old
 #from config import *
  
 from majordome.models import AgentSurvey, AgentCmd, AgentLogs
+from user_mgmt.models import Period
+
+from vendor.guitastro.src.guitastro import Ephemeris
+import pickle 
+
+
 CmdException = AgentCmd.CmdException
 CmdExceptionUnknown = AgentCmd.CmdExceptionUnknown
 CmdExceptionUnimplemented = AgentCmd.CmdExceptionUnimplemented
@@ -3476,7 +3482,41 @@ class Agent:
         if loginfo:
             log.info(self._oc['config'].fn.fcontexts_human())
  
-
+    def ephem_target2night(self, target:str) -> dict:
+        eph = Ephemeris()
+        eph.set_home(self._oc['config'].getHome())
+        fcontext0 = self._oc['config'].fn.fcontext
+        # --- Build the path and file name of the sun ephemeris
+        self._oc['config'].fn.fcontext = "pyros_eph"
+        night = self._oc['config'].fn.date2night("now")
+        operiod = Period.objects.exploitation_period()
+        if operiod == None:
+            log.info("No period valid in the database")
+            self._routine_running = self.RUNNING_NOTHING
+            return
+        period_id = str(operiod.id)
+        if len(str(operiod.id)) < 3:
+            while len(period_id) < 3:
+                period_id = "0" + period_id
+        period_id = "P" + period_id
+        fn_param = {
+            "period" : f"{period_id}",
+            "date": night,
+            "version": "1",
+            "unit": self._oc['config'].unit_name,
+            "target": target
+        }
+        fname = self._fn.naming_set(fn_param)
+        eph_file = self._fn.joinabs(fname)
+        # --- Compute the sun ephemeris if needed. Save it as pickle.
+        if not os.path.exists(eph_file):
+            ephem_target = eph.target2night(target, night, None, None)
+            os.makedirs(os.path.dirname(eph_file), exist_ok=True)
+            pickle.dump(ephem_target, open(eph_file,"wb"))
+        else:
+            ephem_target = pickle.load(open(eph_file,"rb"))
+        self._oc['config'].fn.fcontext = fcontext0
+        return ephem_target
  
 """
 =================================================================