Coverage for lst_auto_rta/High_Level_analysis_Maps.py: 0%
95 statements
« prev ^ index » next coverage.py v7.6.7, created at 2024-11-17 14:47 +0000
« prev ^ index » next coverage.py v7.6.7, created at 2024-11-17 14:47 +0000
1#!/usr/bin/env python
3import datetime
4import logging
5import os
6import time
8import pymongo
11def now():
12 """
13 Returns the current timestamp in seconds, relative to the Unix epoch.
15 :return: float
16 """
17 return datetime.datetime.timestamp(datetime.datetime.now())
20def get_current_run():
21 """
22 Attempts to retrieve the current run number from the lst1_obs_summary database.
24 :return: int
25 """
26 try:
27 client = pymongo.MongoClient("lst101")
28 database = client["lst1_obs_summary"]
29 camera_collection = database["camera"]
30 summaries = camera_collection.find({})
31 run_table = []
32 summaries = camera_collection.find({})
33 for summary in summaries:
34 if summary["tstart"] > datetime.datetime.timestamp(datetime.datetime.now()) - 3600 * 24 * 100:
35 run_table.append(summary["run_number"])
36 if run_table[-2] != run_table[-1]:
37 return run_table[-1]
38 else:
39 return run_table[-1] + 1
40 except Exception as e:
41 logging.error("Error DB !!!!!")
42 return -1
45def get_night_timestamp(today):
46 """
47 Get current timestamp (YYYYMMDD) with respect to the night.
48 It will consider a same night everything run between 08h00 until 07h59 of the next day.
50 Ex: An observation launched at 04h00 (local time) will refer to the previous day.
52 :param today: datetime object
53 datetime object with current local time
55 :return:
56 timestamp: str
57 """
58 # TODO check that today is a datetime object ?
59 if today.hour < 8: # script launched between
60 yesterday = today - datetime.timedelta(days=1)
61 timestamp = f"{yesterday.year:04d}{yesterday.month:02d}{yesterday.day:02d}"
62 else:
63 timestamp = f"{today.year:04d}{today.month:02d}{today.day:02d}"
65 return timestamp
68def Obtain_RA_DEC(runid):
69 try:
70 client = pymongo.MongoClient("lst101")
71 database = client["bridgesmonitoring"]
72 print(client)
74 client = pymongo.MongoClient("lst101")
75 database = client["lst1_obs_summary"]
76 camera_collection = database["telescope"]
77 summaries = camera_collection.find({"data.camera.run_number": runid})
79 for summary in summaries:
80 camera = summary["data"]["camera"]
81 count = 0
82 structure = summary["data"]["structure"]
83 for c in camera:
84 if c["run_number"] == runid:
85 run_count = count
86 count = count + 1
87 run_parameters = structure[run_count]
88 return (
89 run_parameters["target"]["ra"],
90 run_parameters["target"]["dec"],
91 run_parameters["target"]["source_ra"],
92 run_parameters["target"]["source_dec"],
93 )
94 except Exception as e:
95 # client = pymongo.MongoClient('lst101')
96 # database = client['lst1_obs_summary']
97 # camera_collection = database['camera']
98 # summaries = camera_collection.find({})
99 # for summary in summaries:
100 # if (summary['tstart'] > datetime.datetime.timestamp(datetime.datetime.now())-3600*24*100):
101 # if (summary['run_number']==runid and summary["kind"] != "data_taking"):
102 # return [0,0,0,0]
103 logging.error("Error DB !!!!!")
104 return [0, 0, 0, 0]
106def today_to_directory(today):
107 return today[0:4]+"/"+today[4:6]+"/"+today[6:8]
109def main():
110 """
111 Initialize current_run_is and quit variables and get the current timestamp with Start_time = now()
112 Initialize the logging system by setting the log file name and logging level
113 Change the current working directory to /fefs/onsite/pipeline/rta/data
114 Find the current reservations for nodes using the scontrol command and parse the output to find the individual node names
115 Modify a configuration file with sed commands, replacing placeholders with the node names found in step 4
116 Check if the ib0 network interface is in connected mode for each of the nodes and log the results
117 """
118 current_run_is = -1
119 quit = False
121 Start_time = now()
123 # today = str(datetime.datetime.now().year) + str(datetime.datetime.now().month) + str(datetime.datetime.now().day)
124 today = get_night_timestamp(datetime.datetime.now())
125 logging.basicConfig(filename="log_High_Level_Maps_" + today + ".txt", level=logging.INFO)
126 logging.info("Start RTA High Level for the day " + today)
128 RTA_ready = True
130 loop_id = 0
131 ra_dec = [0, 0, 0, 0]
132 data_directory = "'/fefs/onsite/pipeline/rta/data/"
133 while quit == False and RTA_ready:
134 time.sleep(10)
135 # print(current_run_is+1)
136 if current_run_is == -1:
137 current_run_is = get_current_run()
138 if current_run_is == -1:
139 logging.info("None")
140 current_run_is = -1
141 continue
142 if current_run_is != get_current_run():
143 time.sleep(1)
144 current_run_is = get_current_run()
145 if current_run_is != -1:
146 logging.info("Start RTA check for run " + str(current_run_is))
147 time.sleep(2)
148 ra_dec = Obtain_RA_DEC(current_run_is + 1)
149 logging.info(
150 "srun --reservation=rta_one_node ./Ring_Background_Maps.py -da "
151 + today_to_directory(today)
152 + " -r "
153 + str(current_run_is + 1)
154 + " -RA "
155 + str(ra_dec[2])
156 + " -DEC "
157 + str(ra_dec[3])
158 )
159 os.system(
160 "srun --reservation=rta_one_node ./Ring_Background_Maps.py -da "
161 + today_to_directory(today)
162 + " -r "
163 + str(current_run_is + 1)
164 + " -RA "
165 + str(ra_dec[2])
166 + " -DEC "
167 + str(ra_dec[3])
168 )
170 if loop_id % 1 == 0:
171 logging.info("Start RTA check for run " + str(current_run_is))
172 time.sleep(2)
173 logging.info(datetime.datetime.now())
174 ra_dec = Obtain_RA_DEC(current_run_is + 1)
175 logging.info(
176 "srun --reservation=rta_one_node ./Ring_Background_Maps.py -da "
177 + today_to_directory(today)
178 + " -r "
179 + str(current_run_is + 1)
180 + " -RA "
181 + str(ra_dec[2])
182 + " -DEC "
183 + str(ra_dec[3])
184 )
185 os.system(
186 "srun --reservation=rta_one_node ./Ring_Background_Maps.py -da "
187 + today_to_directory(today)
188 + " -r "
189 + str(current_run_is + 1)
190 + " -RA "
191 + str(ra_dec[2])
192 + " -DEC "
193 + str(ra_dec[3])
194 )
196 loop_id = loop_id + 1
197 if (datetime.datetime.timestamp(datetime.datetime.now()) - Start_time) > 14 * 3600:
198 quit = True
199 logging.info("End of the night, Stop the RTA check")
202if __name__ == "__main__":
203 main()