add timestamps to text logging

This commit is contained in:
Sarah Hoffmann 2023-11-22 17:38:32 +01:00
parent 155f26060d
commit e7dc24c026

View File

@ -235,6 +235,10 @@ class TextLogger(BaseLogger):
self.buffer = io.StringIO() self.buffer = io.StringIO()
def _timestamp(self) -> None:
self._write(f'[{dt.datetime.now()}]\n')
def get_buffer(self) -> str: def get_buffer(self) -> str:
return self.buffer.getvalue() return self.buffer.getvalue()
@ -247,6 +251,7 @@ class TextLogger(BaseLogger):
def section(self, heading: str) -> None: def section(self, heading: str) -> None:
self._timestamp()
self._write(f"\n# {heading}\n\n") self._write(f"\n# {heading}\n\n")
@ -283,6 +288,7 @@ class TextLogger(BaseLogger):
def result_dump(self, heading: str, results: Iterator[Tuple[Any, Any]]) -> None: def result_dump(self, heading: str, results: Iterator[Tuple[Any, Any]]) -> None:
self._timestamp()
self._write(f'{heading}:\n') self._write(f'{heading}:\n')
total = 0 total = 0
for rank, res in results: for rank, res in results:
@ -298,6 +304,7 @@ class TextLogger(BaseLogger):
def sql(self, conn: AsyncConnection, statement: 'sa.Executable', def sql(self, conn: AsyncConnection, statement: 'sa.Executable',
params: Union[Mapping[str, Any], Sequence[Mapping[str, Any]], None]) -> None: params: Union[Mapping[str, Any], Sequence[Mapping[str, Any]], None]) -> None:
self._timestamp()
sqlstr = '\n| '.join(textwrap.wrap(self.format_sql(conn, statement, params), width=78)) sqlstr = '\n| '.join(textwrap.wrap(self.format_sql(conn, statement, params), width=78))
self._write(f"| {sqlstr}\n\n") self._write(f"| {sqlstr}\n\n")