libretificacaotjcore 0.1.16__tar.gz → 0.1.18__tar.gz

{libretificacaotjcore-0.1.16 → libretificacaotjcore-0.1.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: libretificacaotjcore
-Version: 0.1.16
+Version: 0.1.18
 Summary: Biblioteca para centralizar conexao com filas no rabbit e banco de dados no mongodb para os servicos de retificacao da TJ
 Author-email: Jhonatan Azevedo <dev.azevedo@outlook.com>
 Project-URL: Homepage, https://github.com/seu-usuario/libretificacaotjcore

{libretificacaotjcore-0.1.16 → libretificacaotjcore-0.1.18}/libretificacaotjcore/database/config_db.py

@@ -29,11 +29,13 @@ class ConfigDb:
             await self.__db.arquivos.create_index([("SolicitacaoId", 1), ("cpf", 1)], unique=True)
             self.db_initialized = True
             
-            if "certificados" not in (await self.__db.list_collection_names()):
-                await self.__db.create_collection("certificados")
+            if "protocolos" not in (await self.__db.list_collection_names()):
+                await self.__db.create_collection("protocolos")
             
             await self.__db.certificados.create_index([("cnpj", 1)], unique=True)
-            await self.__db.arquivos.create_index([("id", 1)], unique=True)
+            await self.__db.protocolos.create_index([("SolicitacaoId", 1)])
+            await self.__db.protocolos.create_index([("id", 1)], unique=True)
+            await self.__db.protocolos.create_index([("SolicitacaoId", 1), ("evento", 1)], unique=True)
             self.db_initialized = True
             
     async def get_db(self):

libretificacaotjcore-0.1.18/libretificacaotjcore/database/protocolo_repository.py

@@ -0,0 +1,57 @@
+from pymongo.errors import BulkWriteError
+
+class ProtocoloRepository:
+    def __init__(self, db):
+        self.__db = db
+
+    async def inserir_protocolo(self, protocolo: dict) -> bool:
+        
+        try:
+            protocolo_no_db = await self.__db.protocolo.find_one(
+                {"SolicitacaoId": protocolo["solicitacaoId"], "evento": protocolo["evento"]}
+            )
+
+            if protocolo_no_db is None:
+                await self.__db.protocolo.insert_one(protocolo)
+                return True
+
+            await self.__db.protocolo.delete_one(
+                   {"SolicitacaoId": protocolo["solicitacaoId"], "evento": protocolo["evento"]}
+            )
+            await self.__db.protocolo.insert_one(protocolo)
+            return True
+        except Exception as e:
+            print(f"❌ Erro ao inserir o protocolo: {e}")
+            return False
+        
+    async def inserir_protocolos_em_lote(self, protocolos: list[dict]) -> bool:
+        try:
+            if not protocolos:
+                return False
+
+            filtros = [{"SolicitacaoId": a["solicitacaoId"], "evento": a["evento"]} for a in protocolos]
+            await self.__db.protocolo.delete_many({"$or": filtros})
+
+            await self.__db.protocolo.insert_many(protocolos)
+            return True
+        except BulkWriteError as bwe:
+            print(f"❌ Erro de escrita em lote: {bwe.details}")
+            return False
+        except Exception as e:
+            print(f"❌ Erro ao inserir protocolos em lote: {e}")
+            return False
+
+    async def remover_protocolo(self, solicitacaoId: int) -> bool:
+        try:
+            await self.__db.protocolo.delete_many({"SolicitacaoId": solicitacaoId})
+            return True
+        except Exception as e:
+            print(f"❌ Erro ao remover o protocolo: {e}")
+            return False
+
+    async def buscar_por_solicitacao_id(self, solicitacaoId: int) -> list[dict]:
+        try:
+            return await self.__db.protocolo.find({"solicitacaoId": solicitacaoId}).to_list(length=None)
+        except Exception as e:
+            print(f"❌ Erro ao buscar protocolo por solicitacaoId: {e}")
+            return []

{libretificacaotjcore-0.1.16 → libretificacaotjcore-0.1.18}/libretificacaotjcore.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: libretificacaotjcore
-Version: 0.1.16
+Version: 0.1.18
 Summary: Biblioteca para centralizar conexao com filas no rabbit e banco de dados no mongodb para os servicos de retificacao da TJ
 Author-email: Jhonatan Azevedo <dev.azevedo@outlook.com>
 Project-URL: Homepage, https://github.com/seu-usuario/libretificacaotjcore

{libretificacaotjcore-0.1.16 → libretificacaotjcore-0.1.18}/libretificacaotjcore.egg-info/SOURCES.txt

@@ -10,6 +10,7 @@ libretificacaotjcore/database/__init__.py
 libretificacaotjcore/database/arquivo_repository.py
 libretificacaotjcore/database/certificado_repository.py
 libretificacaotjcore/database/config_db.py
+libretificacaotjcore/database/protocolo_repository.py
 libretificacaotjcore/dtos/solicitacao_dto.py
 libretificacaotjcore/enums/e_eventos.py
 libretificacaotjcore/services/__init__.py

{libretificacaotjcore-0.1.16 → libretificacaotjcore-0.1.18}/pyproject.toml

@@ -11,7 +11,7 @@ keywords = ["tj", "tributo justo", "retificação", "automação", "pydantic", "
 name = "libretificacaotjcore"
 readme = "README.md"
 requires-python = ">=3.12"
-version = "0.1.16"
+version = "0.1.18"
 
 classifiers = [
   "Development Status :: 3 - Alpha",