log.py

#     Copyright (C) 2023  Coretex LLC
 
#     This file is part of Coretex.ai
 
#     This program is free software: you can redistribute it and/or modify
#     it under the terms of the GNU Affero General Public License as
#     published by the Free Software Foundation, either version 3 of the
#     License, or (at your option) any later version.
 
#     This program is distributed in the hope that it will be useful,
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#     GNU Affero General Public License for more details.
 
#     You should have received a copy of the GNU Affero General Public License
#     along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
from typing import Dict, Any, Tuple
from typing_extensions import Self
 
import time
import json
 
from .severity import LogSeverity
from .utils import colorMessage
from ..utils import mathematicalRound
 
 
class Log:
 
    """
        Represents a single Coretex console log
 
        timestamp : float
            timestamp of the log
        message : str
            message of the log
        severity : LogSeverity
            severity of the log
    """
 
    def __init__(self, severity: LogSeverity, message: str) -> None:
        self.timestamptimestamp = mathematicalRound(time.time(), 6)
        self.severityseverity = severity
        self.messagemessage = colorMessage(severity, message)
 
    def encode(self) -> Dict[str, Any]:
        return {
            "timestamp": self.timestamptimestamp,
            "severity": self.severityseverity.value,
            "content": self.messagemessage,
            "type": 1  # Here for backwards compatibility
        }
 
    @classmethod
    def parse(cls, value: str) -> Tuple[Self, str]:
        try:
            jsonLog = json.loads(value)
 
            if not isinstance(jsonLog, dict):
                raise ValueError
 
            if len(jsonLog) != 2:
                raise ValueError
 
            return cls(LogSeverity(jsonLog["severity"]), jsonLog["message"].rstrip()), jsonLog["message"]
        except:
            return cls(LogSeverity.info, value.rstrip()), value