ipiranga-inovai-project-lib 4.3__py3-none-any.whl → 4.5__py3-none-any.whl

inovai/log.py

@@ -0,0 +1,119 @@
+import logging
+import logging.config
+from contextvars import ContextVar
+from typing import Optional, List
+import asgi_correlation_id
+
+
+integration_batch_id: ContextVar[Optional[str]] = ContextVar('integration_batch_id', default=None)
+document_correlation_id: ContextVar[Optional[str]] = ContextVar('document_correlation_id', default=None)
+origin: ContextVar[Optional[str]] = ContextVar('origin', default=None)
+document_type: ContextVar[Optional[str]] = ContextVar('document_type', default=None)
+documents_error: ContextVar[Optional[List[any]]] = ContextVar('documents_error', default=[])
+branch_code: ContextVar[Optional[str]] = ContextVar('branch_code', default=None)
+movement_type: ContextVar[Optional[str]] = ContextVar('movement_type', default=None)
+origin_id: ContextVar[Optional[str]] = ContextVar("origin_id", default=None)
+internal_batch_id: ContextVar[Optional[str]] = ContextVar('internal_batch_id', default=None)
+integration_employer_number: ContextVar[Optional[str]] = ContextVar(
+    "integration_employer_number", default=None
+)
+content_type: ContextVar[Optional[str]] = ContextVar("content_type", default=None)
+processing_codes: ContextVar[Optional[list[str]]] = ContextVar('processing_codes', default=None)
+kit_id: ContextVar[Optional[str]] = ContextVar("kit_id", default='4')
+
+
+class ChunkedStreamHandler(logging.StreamHandler):
+    def emit(self, record):
+        # Format the record into a log message
+        msg = self.format(record)
+        msg_size = len(msg.encode('utf-8'))  # Get size of the message in bytes
+
+        max_bytes = 50 * 1024  # 50kb in bytes
+
+        # If the message is larger than max_bytes, split it into chunks
+        if msg_size > max_bytes:
+            message = record.msg
+            for i in range(0, len(message), max_bytes):
+                chunk = message[i:i + max_bytes]
+                original_msg = record.msg  # Backup original message
+                try:
+                    record.msg = chunk  # Alter only the message part
+                    record.args = None  # Reset args to avoid formatting issues
+                    super().emit(record)  # Emit the chunked message
+                finally:
+                    record.msg = original_msg  # Restore original message after chunk is emitted
+        else:
+            # If the message fits, emit it as usual
+            super().emit(record)
+
+
+class CustomLogger(logging.Logger):
+
+    def _add_extra_args(self, kwargs):
+        # Define os `extra_args` a serem adicionados
+        extra_args = {
+            "integration_batch_id": str(internal_batch_id.get()),
+            "correlation_id": str(asgi_correlation_id.correlation_id.get()),
+            "document_correlation_id": str(document_correlation_id.get())
+            if document_correlation_id.get() is not None else str(asgi_correlation_id.correlation_id.get()).lower(),
+            "origin": str(origin.get()),
+            "document_type": str(document_type.get()),
+            "documents_error": str(documents_error.get()),
+            "branch_code": str(branch_code.get()),
+            "movement_type": str(movement_type.get()),
+            "origin_id": str(origin_id.get()),
+            "internal_batch_id": str(internal_batch_id.get()),
+            "integration_employer_number": str(integration_employer_number.get()),
+            "content_type": str(content_type.get()),
+            "processing_codes": str(processing_codes.get()),
+            "kit_id": str(kit_id.get())
+        }
+
+        # Atualiza o kwargs do log com `extra_args`
+        if "extra" in kwargs:
+            kwargs["extra"].update(extra_args)
+        else:
+            kwargs["extra"] = extra_args
+
+        return kwargs
+
+    def info(self, msg, *args, **kwargs):
+        # Adicionar os extras para logs do nível INFO
+        kwargs = self._add_extra_args(kwargs)
+        super().info(msg, *args, **kwargs)
+
+    def error(self, msg, *args, **kwargs):
+        # Adicionar os extras para logs do nível ERROR
+        kwargs = self._add_extra_args(kwargs)
+        super().error(msg, *args, **kwargs)
+
+    def warning(self, msg, *args, **kwargs):
+        # Adicionar os extras para logs do nível WARNING
+        kwargs = self._add_extra_args(kwargs)
+        super().warning(msg, *args, **kwargs)
+
+    def debug(self, msg, *args, **kwargs):
+        # Adicionar os extras para logs do nível DEBUG
+        kwargs = self._add_extra_args(kwargs)
+        super().debug(msg, *args, **kwargs)
+
+    def critical(self, msg, *args, **kwargs):
+        # Adicionar os extras para logs do nível CRITICAL
+        kwargs = self._add_extra_args(kwargs)
+        super().critical(msg, *args, **kwargs)
+
+
+def get_logger(name):
+    logging.setLoggerClass(CustomLogger)
+
+    custom_handler = ChunkedStreamHandler()
+    custom_handler.setLevel(logging.INFO)
+
+    log_formatter = logging.Formatter('%(asctime)s.%(msecs)03d %(levelname)s %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
+    custom_handler.setFormatter(log_formatter)
+
+    logger = logging.getLogger(name)
+    logger.setLevel(logging.INFO)
+    logger.addHandler(custom_handler)
+
+    return logger

{ipiranga_inovai_project_lib-4.3.dist-info → ipiranga_inovai_project_lib-4.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ipiranga-inovai-project-lib
-Version: 4.3
+Version: 4.5
 Summary: Projeto criado para importação genérica de entidades
 Home-page: https://gitlab.ipirangacloud.com/clayton.monteiro.ext/ipiranga-inovai-project-lib
 Author: Clayton Sandes Monteiro

{ipiranga_inovai_project_lib-4.3.dist-info → ipiranga_inovai_project_lib-4.5.dist-info}/RECORD

@@ -2,6 +2,7 @@ inovai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 inovai/dtos.py,sha256=qaWZB7n5SHcC58RP351IQ1HLjv7fx3VAdILj-GugVWs,1587
 inovai/entities.py,sha256=-ANFEKFUI-j020CXo7NQnwI9jgFlbuqR1Takkwiq91E,20766
 inovai/enums.py,sha256=hj6KZRzV_LZIEnpycCl1yrP2aQdm75pm8WN8CM3TBfI,2367
+inovai/log.py,sha256=ICySFkiOaJnXuXk3ScYNDU2K9g723h5MFDoiafcU8Q4,5201
 inovai/models/Endereco.py,sha256=Vft20vMwYwTXtEuh5NH7O5FxXVahnLmgaXMDOVdBxY4,3519
 inovai/models/NFe.py,sha256=MNs9hxUcfUXzrYqUIOEkg-4wNGQQ4gKuduw_gJLHpKk,13241
 inovai/models/NFeBoleto.py,sha256=tLWqsOlszA6UiJdxesJKQkgwNk_4DuI3NRBCYZh7RKc,5006
@@ -34,8 +35,8 @@ inovai/models/NFeVolume.py,sha256=_cayb-rrz1mSxQ3wFFnB4KCi524jtAsdTTcS4WQTBOc,19
 inovai/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 inovai/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 inovai/utils/util.py,sha256=CsxaZmeKWG8A4w_1XzVm_F9hdWkDVgmHDYL2XF8v5dQ,906
-ipiranga_inovai_project_lib-4.3.dist-info/LICENSE,sha256=UJ3ql8eZec9Y1kER7QSHAuvdgFPuL47asePHYmdzG_o,198
-ipiranga_inovai_project_lib-4.3.dist-info/METADATA,sha256=xdVDnBWH-8dY_YBdnmpWjtJ_oKI4oOZSjK2DbOjUb6Q,518
-ipiranga_inovai_project_lib-4.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-ipiranga_inovai_project_lib-4.3.dist-info/top_level.txt,sha256=BD-APH5oIEb_KDXwrfnZiu9QDsD9ttTEkYc32r9thqg,7
-ipiranga_inovai_project_lib-4.3.dist-info/RECORD,,
+ipiranga_inovai_project_lib-4.5.dist-info/LICENSE,sha256=UJ3ql8eZec9Y1kER7QSHAuvdgFPuL47asePHYmdzG_o,198
+ipiranga_inovai_project_lib-4.5.dist-info/METADATA,sha256=To7vdhOmZbYOitWtWCZTLK8qIUU1CXgy-AK7iJ9xESE,518
+ipiranga_inovai_project_lib-4.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+ipiranga_inovai_project_lib-4.5.dist-info/top_level.txt,sha256=BD-APH5oIEb_KDXwrfnZiu9QDsD9ttTEkYc32r9thqg,7
+ipiranga_inovai_project_lib-4.5.dist-info/RECORD,,