libretificacaotjcore 0.1.33__py3-none-any.whl → 0.1.49__py3-none-any.whl

libretificacaotjcore/database/arquivo_repository.py

@@ -1,3 +1,4 @@
+
 from pymongo.errors import BulkWriteError
 
 class ArquivoRepository:

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,99 @@
+from datetime import datetime
+from pymongo.errors import BulkWriteError
+
+from libretificacaotjcore.dtos.processo_dto import ProcessoDto
+from libretificacaotjcore.enums.e_fase_retificacao import EFaseRetificacao
+
+class TempoProcessoRepository:
+    def __init__(self, db):
+        self.__db = db
+
+    async def inserir_processo(self, *, processo: ProcessoDto) -> bool:
+        try:
+            processo_dict = processo.model_dump()
+            processo_dict["DataInicio"] = processo.DataInicio.strftime("%Y-%m-%d")
+            processo_dict["DataFim"] = processo.DataFim.strftime("%Y-%m-%d")
+            processo_dict["FaseDescricao"] = EFaseRetificacao(processo.Fase).name
+
+            processo_no_db = await self.__db.tempo_processos.find_one(
+                {
+                    "SolicitacaoId": processo.SolicitacaoId,
+                    "Fase": processo.Fase,
+                    "DataInicio": processo_dict["DataInicio"],
+                    "DataFim": processo_dict["DataFim"],
+                }
+            )
+            
+
+
+            if processo_no_db is None:
+                processo_dict["InicioProcesso"] = datetime.now()
+                await self.__db.tempo_processos.insert_one(processo_dict)
+                return True
+
+            await self.__db.tempo_processos.delete_one(
+                {
+                    "SolicitacaoId": processo.SolicitacaoId,
+                    "Fase": processo.Fase,
+                    "DataInicio": processo_dict["DataInicio"],
+                    "DataFim": processo_dict["DataFim"],
+                }
+            )
+
+            processo_dict["InicioProcesso"] = datetime.now()
+            await self.__db.tempo_processos.insert_one(processo_dict)
+            return True
+        
+        except Exception as e:
+            print(f"❌ Erro ao inserir o processo: {e}")
+            return False
+        
+    async def atualizar_processo(self, *, processo: ProcessoDto) -> bool:
+        try:
+            processo_dict = processo.model_dump()
+            processo_dict["DataInicio"] = processo.DataInicio.strftime("%Y-%m-%d")
+            processo_dict["DataFim"] = processo.DataFim.strftime("%Y-%m-%d")
+             
+            processo_no_db = await self.__db.tempo_processos.find_one(
+                {
+                    "SolicitacaoId": processo.SolicitacaoId,
+                    "Fase": processo.Fase,
+                    "DataInicio": processo_dict["DataInicio"],
+                    "DataFim": processo_dict["DataFim"],
+                }
+            )
+
+            if processo_no_db is None:
+                return False
+            
+            processo_no_db['FimProcesso'] = datetime.now()
+            tempo_de_processo = self._tempo_de_processo(processo_no_db['InicioProcesso'], processo_no_db['FimProcesso'])
+            processo_no_db['TempoDeProcesso'] = tempo_de_processo
+
+            await self.__db.tempo_processos.replace_one(
+                {
+                    "SolicitacaoId": processo.SolicitacaoId,
+                    "Fase": processo.Fase,
+                    "DataInicio": processo_dict["DataInicio"],
+                    "DataFim": processo_dict["DataFim"],
+                },
+                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 and tempo_fim:
+            delta = tempo_fim - tempo_inicio  # <-- inverter a ordem
+            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/dtos/processo_dto.py

@@ -0,0 +1,78 @@
+from datetime import date, datetime, timedelta
+from pydantic import BaseModel, Field, field_validator
+from libretificacaotjcore.enums.e_fase_retificacao import EFaseRetificacao
+
+
+class ProcessoDto(BaseModel):
+    SolicitacaoId: int = Field(..., description="ID da solicitação")
+    DataInicio: date = Field(..., description="Data de início no formato YYYY-MM-DD")
+    DataFim: date = Field(..., description="Data de fim no formato YYYY-MM-DD")
+    Fase: int | None = Field(None, description="Fase de retificação")
+    InicioProcesso: datetime | None = Field(None, description="Data e hora de início do processo")
+    FimProcesso: datetime | None = Field(None, description="Data e hora de fim do processo")
+    TempoDeProcesso: str | None = Field(None, description="Tempo de processamento")
+    Observacoes: str | None = Field(None, description="Observações adicionais")
+
+    # --- Validadores ---
+
+    @field_validator("SolicitacaoId")
+    @classmethod
+    def validar_solicitacao_id(cls, v: int) -> int:
+        if v <= 0:
+            raise ValueError("O solicitacaoId deve ser um inteiro positivo.")
+        return v
+
+    @field_validator("DataInicio", mode="before")
+    @classmethod
+    def formatar_data_inicio(cls, value) -> date:
+        """
+        Aceita date, datetime ou string.
+        """
+        if isinstance(value, date):
+            return value
+        if isinstance(value, datetime):
+            return value.date()
+        if isinstance(value, str):
+            try:
+                return datetime.strptime(value, "%Y-%m-%d").date()
+            except ValueError as ve:
+                raise ValueError("A DataInicio deve estar no formato YYYY-MM-DD.") from ve
+        raise TypeError("Tipo inválido para DataInicio.")
+
+    @field_validator("DataFim", mode="before")
+    @classmethod
+    def ajustar_data_fim(cls, value) -> date:
+        """
+        Aceita date, datetime ou string.
+        Retorna o último dia do mês da data fornecida.
+        """
+        if isinstance(value, datetime):
+            value = value.date()
+        if isinstance(value, date):
+            ano, mes = value.year, value.month
+        elif isinstance(value, str):
+            try:
+                ano, mes = map(int, value.split("-")[:2])
+            except Exception:
+                raise ValueError("A DataFim deve estar no formato YYYY-MM-DD.")
+        else:
+            raise TypeError("Tipo inválido para DataFim.")
+
+        if mes < 1 or mes > 12:
+            raise ValueError("A DataFim deve conter um mês válido (1–12).")
+
+        # calcula o último dia do mês
+        if mes == 12:
+            proximo_mes = datetime(ano + 1, 1, 1)
+        else:
+            proximo_mes = datetime(ano, mes + 1, 1)
+        return (proximo_mes - timedelta(days=1)).date()
+
+    @field_validator("Fase")
+    @classmethod
+    def validar_fase(cls, v):
+        if v is None:
+            return v
+        if v not in EFaseRetificacao:
+            raise ValueError(f"Fase '{v}' não é válida.")
+        return v

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

@@ -8,18 +8,25 @@ class EFaseRetificacao(Enum):
     DownloadXml = 3
     ExtraindoDadosDoXml = 4
     AguardandoRubrica = 5
-    AberturaDeCompetencia = 6
-    ConsultandoESocialAberturaCompetencia = 7
-    InclusaoDasRubricas = 8
-    ConsultandoESocialInclusaoRubricas = 9
-    ExclusaoDePagamentos = 10
-    ConsultandoESocialExclusaoPagamentos = 11
-    RetificacaoDaRemuneracao = 12
-    ConsultandoESocialRetificacaoRemuneracao = 13
-    InclusaoDosPagamentos = 14
-    ConsultandoESocialInclusaoPagamentos = 15
-    Desligamento = 16
-    ConsultandoESocialDesligamento = 17
-    FechamentoDeCompetencia = 18
-    ConsultandoESocialFechamentoCompetencia = 19
-    Finalizado = 20
+    EstruturandoXmlAberturaCompetencia = 6
+    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,
-            heartbeat=60,
-            blocked_connection_timeout=300,
+
+    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,
         )
-        
-        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/services/request_servico_api.py

@@ -1,16 +1,20 @@
 import httpx
 
+
 class RequestServicoApi:
     def __init__(self, url, token):
         self.url = url
         self.token = token
+        self.client = httpx.AsyncClient(timeout=120, verify=False)
 
     async def handler(self, *, mensagem_atualizacao: dict):
-        async with httpx.AsyncClient(timeout=120, verify=False) as client:
-            print(self.token)
-            response = await client.post(self.url, json=mensagem_atualizacao)
+        print(self.token)
+        response = await self.client.post(self.url, json=mensagem_atualizacao)
+
+        if response.status_code != 200:
+            raise Exception(f"Erro ao fazer request ao servico de API: {response.status_code}")
+        
+        await self.close()
 
-            if response.status_code != 200:
-                raise Exception(f"Erro ao fazer request ao servico de API: {response.status_code}")
-            
-            return response.json()
+    async def close(self):
+        await self.client.aclose()

{libretificacaotjcore-0.1.33.dist-info → libretificacaotjcore-0.1.49.dist-info}/METADATA

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

@@ -1,23 +1,26 @@
 libretificacaotjcore/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libretificacaotjcore/database/__init__.py,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
-libretificacaotjcore/database/arquivo_repository.py,sha256=Wv42CWjddS8RPVR3aAOGwuFuLyGqmgq-TXOc-V0BTvE,2154
+libretificacaotjcore/database/arquivo_repository.py,sha256=sWXdaHzS0xogZfa4VtdMcd7P7QOmStn12GrNVcGAuYo,2156
 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=ZrOynusf7YIrkITEZQ-8KLMPJLBiHWIAg1EyFvHjVSE,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=3pbeh69qI3I-Ucw-mGYrVLkyeh_wKTKAorq71ZHuc_k,3981
 libretificacaotjcore/dtos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libretificacaotjcore/dtos/arquivo_dto.py,sha256=D4LLJZf-Z0XhJmOvu7nAcBq0j70m9TFtbgxI9ikHXQ0,1045
+libretificacaotjcore/dtos/processo_dto.py,sha256=KbG3TjcxHS4UTdflApChBrq6BpM8xS1Cd6PNba--hA0,3096
 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=-MCqN3s3MHu2W-MlfLugGsI91CwoBUObqxJJGW9wGL8,766
+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=XPbOvFR3HySfA1IROLPrsb4AmeHeq1Unthxt44vaqik,1760
+libretificacaotjcore/services/rabbitmq_consumer.py,sha256=yRS3sAkaMIIn8Y06icwkO1bpddlLC7BmNSHJVhUalzY,1818
 libretificacaotjcore/services/rabbitmq_publisher.py,sha256=xsb9LmIuZizzq5lxnEEZKMK3U952KSQ2oqlTBZi2Lt8,1787
-libretificacaotjcore/services/request_servico_api.py,sha256=nv5DdFh12fLDUkQQSW4Q_tezASYFVAqMvrm7ZUaAZXw,576
+libretificacaotjcore/services/request_servico_api.py,sha256=G7vnmvOfwrCGL-Jy_5tCKG-l5E00OCkJfkuoa5Y6sHo,613
 libretificacaotjcore/services/s3_service.py,sha256=HKR_jt2H3XdV1PCzo5R5bnhmoQ3I46Yn5IqAvVPhsjs,2946
-libretificacaotjcore-0.1.33.dist-info/METADATA,sha256=9kzSaYE2C99t-ap0SG7l3sc0hEd4PtqYbwV_qXR0xDU,2564
-libretificacaotjcore-0.1.33.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-libretificacaotjcore-0.1.33.dist-info/top_level.txt,sha256=J9vnz_X9OUnxC-eXHiAzlc9xIrWBwZ5bgnIDQIIFY4c,21
-libretificacaotjcore-0.1.33.dist-info/RECORD,,
+libretificacaotjcore-0.1.49.dist-info/METADATA,sha256=1x9RtwNRlGp8NrJLfPW8HE4s_16vE-th92cweckLYek,2564
+libretificacaotjcore-0.1.49.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+libretificacaotjcore-0.1.49.dist-info/top_level.txt,sha256=J9vnz_X9OUnxC-eXHiAzlc9xIrWBwZ5bgnIDQIIFY4c,21
+libretificacaotjcore-0.1.49.dist-info/RECORD,,