Commit 061bf13a authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

refactor message and timestamp

parent 90ae3a25
Pipeline #2432 failed with stage
...@@ -880,7 +880,7 @@ class CsstMsDataManager: ...@@ -880,7 +880,7 @@ class CsstMsDataManager:
self.logger_ppl.info( self.logger_ppl.info(
f"Results from DFS CATAPI:\n" f"Results from DFS CATAPI:\n"
f" - code = {cat['code']}\n" f" - code = {cat['code']}\n"
f" - msg = {cat['msg']}\n" f" - message = {cat['message']}\n"
f" - totalCount = {cat['totalCount']}\n" f" - totalCount = {cat['totalCount']}\n"
f" - columns = {cat['columns']}" f" - columns = {cat['columns']}"
) )
...@@ -889,7 +889,7 @@ class CsstMsDataManager: ...@@ -889,7 +889,7 @@ class CsstMsDataManager:
self.logger_ppl.info( self.logger_ppl.info(
f"Results from DFS CATAPI:\n" f"Results from DFS CATAPI:\n"
f" - code = {cat['code']}\n" f" - code = {cat['code']}\n"
f" - msg = {cat['msg']}" f" - message = {cat['message']}"
) )
raise ValueError("Bad catalog query result!") raise ValueError("Bad catalog query result!")
...@@ -921,7 +921,7 @@ class CsstMsDataManager: ...@@ -921,7 +921,7 @@ class CsstMsDataManager:
# query for L0 data # query for L0 data
print(f"Query obs_id={obs_id} ...", end="") print(f"Query obs_id={obs_id} ...", end="")
recs = Level0DataApi().find(obs_id=obs_id) recs = Level0DataApi().find(obs_id=obs_id)
print("MessageWriter: \n", recs) print("Message: \n", recs)
print(f"{recs['totalCount']} records obtained!") print(f"{recs['totalCount']} records obtained!")
assert recs["code"] == 0 assert recs["code"] == 0
assert recs["totalCount"] > 0 assert recs["totalCount"] > 0
......
...@@ -51,7 +51,7 @@ def get_logger(name: str = "CSST pipeline", filename: Optional[str] = None): ...@@ -51,7 +51,7 @@ def get_logger(name: str = "CSST pipeline", filename: Optional[str] = None):
# logging formatter # logging formatter
formatter = logging.Formatter( formatter = logging.Formatter(
"%(asctime)s - %(levelname)s - %(module)s.py:%(lineno)d - %(msg)s" "%(asctime)s - %(levelname)s - %(module)s.py:%(lineno)d - %(message)s"
) )
# stream handler # stream handler
......
...@@ -98,8 +98,8 @@ class Pipeline: ...@@ -98,8 +98,8 @@ class Pipeline:
os.chdir(self.dir_output) os.chdir(self.dir_output)
# Frequently used files # Frequently used files
self.msg = MessageWriter(os.path.join(self.dir_output, "message.txt")) self.message = Message(os.path.join(self.dir_output, "message.txt"))
self.tsr = TimestampRecorder(os.path.join(self.dir_output, "timestamp.txt")) self.timestamp = Timestamp(os.path.join(self.dir_output, "timestamp.txt"))
# self.exit_code = ExitCode(os.path.join(self.dir_output, "exit_code")) # self.exit_code = ExitCode(os.path.join(self.dir_output, "exit_code"))
# self.error_trace = ErrorTrace(os.path.join(self.dir_output, "error_trace")) # self.error_trace = ErrorTrace(os.path.join(self.dir_output, "error_trace"))
...@@ -168,10 +168,10 @@ class Pipeline: ...@@ -168,10 +168,10 @@ class Pipeline:
# f.write(s) # f.write(s)
class MessageWriter: class Message:
"""Write JSON format messages to file.""" """Write JSON format messages to file."""
def __init__(self, file_path=""): def __init__(self, file_path: str = ""):
self.file_path = file_path self.file_path = file_path
def __repr__(self): def __repr__(self):
...@@ -211,12 +211,12 @@ class MessageWriter: ...@@ -211,12 +211,12 @@ class MessageWriter:
# print(f"Exit with code {code} (written to '{self.file_path}')") # print(f"Exit with code {code} (written to '{self.file_path}')")
class TimestampRecorder: class Timestamp:
def __init__(self, file_path: str = "tsr.txt"): def __init__(self, file_path: str = "timestamp.txt"):
""" """
TimestampRecorder Class. Timestamp Class.
Initialize a TimestampRecorder object anc connect it to `file_path`. Initialize a Timestamp object anc connect it to `file_path`.
Parameters Parameters
---------- ----------
...@@ -226,7 +226,7 @@ class TimestampRecorder: ...@@ -226,7 +226,7 @@ class TimestampRecorder:
self.file_path = file_path self.file_path = file_path
def __repr__(self): def __repr__(self):
return f"< TimestampRecorder [{self.file_path}] >" return f"< Timestamp [{self.file_path}] >"
def empty(self): def empty(self):
"""Clean time stamp file.""" """Clean time stamp file."""
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment