libretificacaotjcore 0.1.37__py3-none-any.whl → 0.1.46__py3-none-any.whl

libretificacaotjcore/database/config_db.py

@@ -35,6 +35,21 @@ class ConfigDb:
             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
+
+            if "rubricas" not in (await self.__db.list_collection_names()):
+                await self.__db.create_collection("rubricas")
+                
+            await self.__db.rubricas.create_index([("cnpj", 1)])
+            await self.__db.rubricas.create_index([("solicitacaoId", 1)], unique=True)
+            await self.__db.rubricas.create_index([("id", 1)], unique=True)
+            self.db_initialized = True
+
+            if "tempo_processos" not in (await self.__db.list_collection_names()):
+                await self.__db.create_collection("tempo_processos")
+                
+            await self.__db.rubricas.create_index([("solicitacaoId", 1)])
+            await self.__db.rubricas.create_index([("id", 1)], unique=True)
+            self.db_initialized = True
             
     async def get_db(self):
         await self.criar_schema()

libretificacaotjcore/database/rubrica_repository.py

@@ -0,0 +1,32 @@
+from pymongo.errors import BulkWriteError
+
+class RubricaRepository:
+    def __init__(self, db):
+        self.__db = db
+
+    async def inserir_rubrica(self, rubrica: dict) -> bool:
+        
+        try:
+            rubricas_no_db = await self.__db.rubricas.find_one(
+                {"solicitacaoId": rubrica["solicitacaoId"]}
+            )
+
+            if rubricas_no_db is None:
+                await self.__db.rubricas.insert_one(rubrica)
+                return True
+
+            await self.__db.rubricas.delete_one(
+                   {"solicitacaoId": rubrica["solicitacaoId"]}
+            )
+            await self.__db.rubricas.insert_one(rubrica)
+            return True
+        except Exception as e:
+            print(f"❌ Erro ao inserir o rubrica: {e}")
+            return False
+        
+    async def buscar_por_solicitacao_id(self, solicitacaoId: int) -> dict:
+        try:
+            return await self.__db.rubricas.find_one({"solicitacaoId": solicitacaoId})
+        except Exception as e:
+            print(f"❌ Erro ao buscar rubricas por solicitacaoId: {e}")
+            return {}

libretificacaotjcore/database/tempo_processo_repository.py

@@ -0,0 +1,63 @@
+from datetime import datetime
+from pymongo.errors import BulkWriteError
+
+class TempoProcessoRepository:
+    def __init__(self, db):
+        self.__db = db
+
+    async def inserir_processo(self, processo: dict) -> bool:
+        try:
+            processo_no_db = await self.__db.tempo_processos.find_one(
+                {"solicitacaoId": processo["solicitacaoId"], "fase": processo["fase"], "data_inicio": processo["data_inicio"], "data_fim": processo["data_fim"]}
+            )
+
+            if processo_no_db is None:
+                await self.__db.tempo_processos.insert_one(processo)
+                return True
+
+            await self.__db.tempo_processos.delete_one(
+                {"solicitacaoId": processo["solicitacaoId"], "fase": processo["fase"], "data_inicio": processo["data_inicio"], "data_fim": processo["data_fim"]}
+            )
+
+            processo['inicio_processo'] = datetime.now()
+            await self.__db.tempo_processos.insert_one(processo)
+            return True
+        except Exception as e:
+            print(f"❌ Erro ao inserir o processo: {e}")
+            return False
+        
+    async def atualizar_processo(self, *, solicitacaoId: int, fase: int, data_inicio: str, data_fim: str) -> bool:
+        try:
+            processo_no_db = await self.__db.tempo_processos.find_one(
+                {"solicitacaoId": solicitacaoId, "fase": fase, "data_inicio": data_inicio, "data_fim": data_fim}
+            )
+
+            if processo_no_db is None:
+                return False
+            
+            processo_no_db['fim_processo'] = datetime.now()
+            tempo_de_processo = self._tempo_de_processo(processo_no_db['inicio_processo'], processo_no_db['fim_processo'])
+            processo_no_db['tempo_de_processo'] = tempo_de_processo
+
+            await self.__db.tempo_processos.replace_one(
+                {"solicitacaoId": solicitacaoId, "fase": fase, "data_inicio": data_inicio, "data_fim": data_fim},
+                processo_no_db
+            )
+            return True
+        except Exception as e:
+            print(f"❌ Erro ao atualizar o processo: {e}")
+            return False
+        
+    def _tempo_de_processo(self, tempo_inicio: datetime, tempo_fim: datetime) -> str | None:
+        if tempo_inicio:
+                delta = tempo_inicio - tempo_fim
+                total_segundos = int(delta.total_seconds())
+
+                horas = total_segundos // 3600
+                minutos = (total_segundos % 3600) // 60
+                segundos = total_segundos % 60
+
+                tempo_formatado = f"{horas:02d}:{minutos:02d}:{segundos:02d}"
+                return tempo_formatado
+         
+        return None

libretificacaotjcore/enums/e_eventos.py

@@ -2,6 +2,7 @@ from enum import Enum
 
 
 class EEventos(str, Enum):
+    S1010 = "S-1010"
     S1200 = "S-1200"
     S1210 = "S-1210"
     S1298 = "S-1298"

libretificacaotjcore/enums/e_fase_retificacao.py

@@ -9,23 +9,24 @@ class EFaseRetificacao(Enum):
     ExtraindoDadosDoXml = 4
     AguardandoRubrica = 5
     EstruturandoXmlAberturaCompetencia = 6
-    EstruturandoXmlExclusaoPagamentos = 7
-    EstruturandoXmlRetificacaoRemuneracao = 8
-    EstruturandoXmlInclusaoPagamentos = 9
-    EstruturandoXmlDesligamento = 10
-    EstruturandoXmlFechamentoCompetencia = 11
-    AberturaDeCompetencia = 12
-    ConsultandoESocialAberturaCompetencia = 13
-    InclusaoDasRubricas = 14
-    ConsultandoESocialInclusaoRubricas = 15
-    ExclusaoDePagamentos = 16
-    ConsultandoESocialExclusaoPagamentos = 17
-    RetificacaoDaRemuneracao = 18
-    ConsultandoESocialRetificacaoRemuneracao = 19
-    InclusaoDosPagamentos = 20
-    ConsultandoESocialInclusaoPagamentos = 21
-    Desligamento = 22
-    ConsultandoESocialDesligamento = 23
-    FechamentoDeCompetencia = 24
-    ConsultandoESocialFechamentoCompetencia = 25
-    Finalizado = 26
+    EstruturandoXmlInclusaoRubricas = 7
+    EstruturandoXmlExclusaoPagamentos = 8
+    EstruturandoXmlRetificacaoRemuneracao = 9
+    EstruturandoXmlInclusaoPagamentos = 10
+    EstruturandoXmlDesligamento = 11
+    EstruturandoXmlFechamentoCompetencia = 12
+    AberturaDeCompetencia = 13
+    ConsultandoESocialAberturaCompetencia = 14
+    InclusaoDasRubricas = 15
+    ConsultandoESocialInclusaoRubricas = 16
+    ExclusaoDePagamentos = 17
+    ConsultandoESocialExclusaoPagamentos = 18
+    RetificacaoDaRemuneracao = 19
+    ConsultandoESocialRetificacaoRemuneracao = 20
+    InclusaoDosPagamentos = 21
+    ConsultandoESocialInclusaoPagamentos = 22
+    Desligamento = 23
+    ConsultandoESocialDesligamento = 24
+    FechamentoDeCompetencia = 25
+    ConsultandoESocialFechamentoCompetencia = 26
+    Finalizado = 27

libretificacaotjcore/services/rabbitmq_consumer.py

@@ -1,15 +1,16 @@
-import pika
+import asyncio
+import aio_pika
 import json
 
-
 class RabbitMQConsumer:
     def __init__(
         self,
+        *,
         host: str,
         queue: str,
         username: str,
         password: str,
-        vhost: str ="/",
+        vhost: str = "/",
     ):
         self.host = host
         self.queue = queue
@@ -18,39 +19,41 @@ class RabbitMQConsumer:
         self.vhost = vhost
         self.connection = None
         self.channel = None
-        self.connect()
-
-    def connect(self):
-        credentials = pika.PlainCredentials(self.username, self.password)
-        parameters = pika.ConnectionParameters(
-            host=self.host, 
-            credentials=credentials, 
-            virtual_host=self.vhost,
+
+    async def connect(self):
+        self.connection = await aio_pika.connect_robust(
+            host=self.host,
+            login=self.username,
+            password=self.password,
+            virtualhost=self.vhost,
             heartbeat=600,
-            blocked_connection_timeout=1200,
         )
-        
-        self.connection = pika.BlockingConnection(parameters)
-        self.channel = self.connection.channel()
-        self.channel.queue_declare(queue=self.queue, durable=True)
-
-    def start_consuming(self, callback):
-        def on_message(ch, method, properties, body):
-            mensagem = json.loads(body)
-            callback(mensagem)
-            ch.basic_ack(delivery_tag=method.delivery_tag)
-            self.close()
-            
+        self.channel = await self.connection.channel()
+        await self.channel.set_qos(prefetch_count=1)
+        await self.channel.declare_queue(self.queue, durable=True)
+
+    async def start_consuming(self, callback):
         if not self.channel:
             raise RuntimeError("❌ Canal RabbitMQ não conectado. Chame connect() antes.")
 
-        self.channel.basic_qos(prefetch_count=1)
-        self.channel.basic_consume(queue=self.queue, on_message_callback=on_message)
-        print(
-            f'[*] Aguardando mensagens na fila "{self.queue}". Para sair pressione CTRL+C'
-        )
-        self.channel.start_consuming()
+        queue = await self.channel.get_queue(self.queue)
+
+        async def on_message(message):
+            async with message.process():
+                try:
+                    mensagem = json.loads(message.body.decode())
+                    await callback(mensagem)  # aqui sim passa o DTO
+                except Exception as e:
+                    print(f"❌ Erro ao processar mensagem: {e}")
+
+        await queue.consume(on_message)  # registra callback
+
+        print(f'[*] Aguardando mensagens na fila "{self.queue}". Para sair pressione CTRL+C')
+
+        # Mantém o consumer rodando
+        await asyncio.Future()
+
 
-    def close(self):
+    async def close(self):
         if self.connection:
-            self.connection.close()
+            await self.connection.close()

{libretificacaotjcore-0.1.37.dist-info → libretificacaotjcore-0.1.46.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: libretificacaotjcore
-Version: 0.1.37
+Version: 0.1.46
 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.37.dist-info → libretificacaotjcore-0.1.46.dist-info}/RECORD

@@ -2,22 +2,24 @@ libretificacaotjcore/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 libretificacaotjcore/database/__init__.py,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
 libretificacaotjcore/database/arquivo_repository.py,sha256=Wv42CWjddS8RPVR3aAOGwuFuLyGqmgq-TXOc-V0BTvE,2154
 libretificacaotjcore/database/certificado_repository.py,sha256=LF3rV1rQmRGZVB4wPh_vmDj81Gf_env_5hqtTbxXNFM,1396
-libretificacaotjcore/database/config_db.py,sha256=1SNpQV6GcrS0aZGX12wXmxhmLb8oc3uV4_H6HQ1z74w,1775
+libretificacaotjcore/database/config_db.py,sha256=K7N0p3Q3ldGXsOJpZHC3XwDWNxjuKZbtd22uMv_-6BE,2572
 libretificacaotjcore/database/protocolo_repository.py,sha256=kb-mXxiOYDPbrPbYMhOb9xDZHknUdVSfVlmQ_Cn0l1s,3788
+libretificacaotjcore/database/rubrica_repository.py,sha256=BVZLukKVH1YKRgPkk18aqck90DdNGrqv4cYr1IRSbL4,1130
+libretificacaotjcore/database/tempo_processo_repository.py,sha256=XY8MQge8VE2BBe3IxNKdk-EE3seMi7P1xDL4lm5Bupo,2730
 libretificacaotjcore/dtos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libretificacaotjcore/dtos/arquivo_dto.py,sha256=D4LLJZf-Z0XhJmOvu7nAcBq0j70m9TFtbgxI9ikHXQ0,1045
 libretificacaotjcore/dtos/solicitacao_dto.py,sha256=dmI-JGgE0wZ9zXqej6pqSIIaopkROnBQo2Hj3pEXcZ8,2576
 libretificacaotjcore/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-libretificacaotjcore/enums/e_eventos.py,sha256=EClyRpHtL21t_s1b_26bbBKpvd6QgOtIe87FmSzh34E,185
-libretificacaotjcore/enums/e_fase_retificacao.py,sha256=g7xVlIMKpo9QUDnOdwYzsx8ojYt5s29JKs8FFIB5Dxk,1032
+libretificacaotjcore/enums/e_eventos.py,sha256=4NICIKFpxEvxMbHu_kcKqNXqylDETcr6qY1tK2bKmi4,207
+libretificacaotjcore/enums/e_fase_retificacao.py,sha256=v7ERN6MuhdiuT2Yhzu5yzvR6rpKaxqoNMKrBlk4zhkY,1074
 libretificacaotjcore/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libretificacaotjcore/services/crypto_pass_service.py,sha256=9D0vyjan6f_8AfNxGkLpGdvyMpojsJq_AAySpv_zKMc,740
 libretificacaotjcore/services/file_service.py,sha256=14CJokBbrsryQGmL0_unH2QKZpnteEAfxf5CPFdv6cE,2075
-libretificacaotjcore/services/rabbitmq_consumer.py,sha256=zk3DDjWZMDHcj32pTeu4zQTabRbDtvreypmr_loNRFg,1762
+libretificacaotjcore/services/rabbitmq_consumer.py,sha256=yRS3sAkaMIIn8Y06icwkO1bpddlLC7BmNSHJVhUalzY,1818
 libretificacaotjcore/services/rabbitmq_publisher.py,sha256=xsb9LmIuZizzq5lxnEEZKMK3U952KSQ2oqlTBZi2Lt8,1787
 libretificacaotjcore/services/request_servico_api.py,sha256=G7vnmvOfwrCGL-Jy_5tCKG-l5E00OCkJfkuoa5Y6sHo,613
 libretificacaotjcore/services/s3_service.py,sha256=HKR_jt2H3XdV1PCzo5R5bnhmoQ3I46Yn5IqAvVPhsjs,2946
-libretificacaotjcore-0.1.37.dist-info/METADATA,sha256=Z9mls_R_1NA7XPbeJPvPRzWWeeNCjJ8euXjZqsIKuDc,2564
-libretificacaotjcore-0.1.37.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-libretificacaotjcore-0.1.37.dist-info/top_level.txt,sha256=J9vnz_X9OUnxC-eXHiAzlc9xIrWBwZ5bgnIDQIIFY4c,21
-libretificacaotjcore-0.1.37.dist-info/RECORD,,
+libretificacaotjcore-0.1.46.dist-info/METADATA,sha256=V_FREYB7uKCZMVNsy3gW5CQCEdDltUN25RU7HCpgASQ,2564
+libretificacaotjcore-0.1.46.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+libretificacaotjcore-0.1.46.dist-info/top_level.txt,sha256=J9vnz_X9OUnxC-eXHiAzlc9xIrWBwZ5bgnIDQIIFY4c,21
+libretificacaotjcore-0.1.46.dist-info/RECORD,,