diff --git a/mo/imports.py b/mo/imports.py index f326b767d5b97dfae180ee0b6a88439148dbf9d7..6bccc47d8a2d6e8f56f424acced6a3a0f48047a1 100644 --- a/mo/imports.py +++ b/mo/imports.py @@ -76,16 +76,25 @@ class Import: self.to_log = [] self.user = user - def error(self, msg: str) -> Any: + def _format_message(self, msg: str) -> str: if self.line_number > 0: if self.row_name: - msg = f"Řádek {self.line_number} ({self.row_name}): {msg}" + return f"Řádek {self.line_number} ({self.row_name}): {msg}" else: - msg = f"Řádek {self.line_number}: {msg}" + return f"Řádek {self.line_number}: {msg}" + return msg + + def error(self, msg: str) -> Any: + msg = self._format_message(msg) self.errors.append(msg) - logger.info('Import: >> %s', msg) + logger.info('Import: E> %s', msg) return None # Kdyby bylo otypováno správně jako -> None, při volání by si mypy stěžoval + def warning(self, msg: str) -> None: + msg = self._format_message(msg) + self.warnings.append(msg) + logger.info('Import: W> %s', msg) + def parse_user_id(self, user_id_str: str) -> Optional[int]: if user_id_str == "": return self.error('Chybí ID uživatele')