libretificacaotjcore 0.1.0__py3-none-any.whl

libretificacaotjcore/database/__init__.py

@@ -0,0 +1 @@
+

libretificacaotjcore/database/arquivo_repository.py

@@ -0,0 +1,31 @@
+class ArquivoRepository:
+    def __init__(self, db):
+        self.__db = db
+
+    async def inserir_arquivo(self, arquivo: dict) -> bool:
+        
+        try:
+            arquivo_no_db = await self.__db.arquivos.find_one(
+                {"SolicitacaoId": arquivo["solicitacaoId"], "cpf": arquivo["cpf"]}
+            )
+
+            if arquivo_no_db is None:
+                await self.__db.arquivos.insert_one(arquivo)
+                return True
+
+            await self.__db.arquivos.delete_one(
+                {"SolicitacaoId": arquivo["solicitacaoId"], "cpf": arquivo["cpf"]}
+            )
+            await self.__db.arquivos.insert_one(arquivo)
+            return True
+        except Exception as e:
+            print(f"❌ Erro ao inserir o arquivo: {e}")
+            return False
+
+    async def remover_arquivo(self, solicitacaoId: int) -> bool:
+        try:
+            await self.__db.arquivos.delete_many({"SolicitacaoId": solicitacaoId})
+            return True
+        except Exception as e:
+            print(f"❌ Erro ao remover o arquivo: {e}")
+            return False

libretificacaotjcore/database/config_db.py

@@ -0,0 +1,33 @@
+import motor.motor_asyncio
+from bson import ObjectId
+from datetime import datetime
+
+class ConfigDb:
+    def __init__(self, host: str, user: str, password: str, port: int, db_name: str):
+        self.host = host
+        self.user = user
+        self.password = password
+        self.port = port
+        self.db_name = db_name
+        
+        self.client = motor.motor_asyncio.AsyncIOMotorClient(
+            f"mongodb://{self.host}:{self.port}"
+        )
+        self.db = self.client[self.db_name]
+        self.db_initialized = False
+        
+        
+    async def criar_schema(self):
+        global _db_initialized
+        if not self.db_initialized:
+            if "arquivos" not in (await self.db.list_collection_names()):
+                await self.db.create_collection("arquivos")
+            
+            await self.db.arquivos.create_index([("cnpj", 1)])
+            await self.db.arquivos.create_index([("SolicitacaoId", 1)])
+            await self.db.arquivos.create_index([("id", 1)], unique=True)
+            self.db_initialized = True
+            
+    async def get_db(self):
+        await self.criar_schema()
+        return self.db

libretificacaotjcore/dtos/solicitacao_dto.py

@@ -0,0 +1,63 @@
+from datetime import datetime, date, timedelta
+import re
+from pydantic import BaseModel, Field, field_validator
+
+
+class SolicitacaoDTO(BaseModel):
+    solicitacaoId: int = Field(..., description="ID da solicitação")
+    cnpj: str = Field(..., description="CNPJ da empresa")
+    dataInicio: str = Field(..., description="Data de início no formato YYYY-MM-DD")
+    dataFim: str = Field(..., description="Data de fim no formato YYYY-MM-DD")
+    certificadoId: int = Field(..., description="ID do certificado")
+    
+    
+    # Validar e transformar solicitacaoId
+    @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
+
+    # Validar e transformar certificadoId
+    @field_validator('certificadoId')
+    @classmethod
+    def validar_certificado_id(cls, v: int) -> int:
+        if v <= 0:
+            raise ValueError("O certificadoId deve ser um inteiro positivo.")
+        return v
+
+    # Validar CNPJ (apenas estrutura com 14 dígitos)
+    @field_validator('cnpj')
+    @classmethod
+    def validar_cnpj(cls, v: str) -> str:
+        cnpj_limpo = re.sub(r'\D', '', v)
+        if len(cnpj_limpo) != 14 or not cnpj_limpo.isdigit():
+            raise ValueError("O CNPJ deve conter 14 dígitos numéricos.")
+        return cnpj_limpo
+
+    # Validar e transformar dataInicio
+    @field_validator('dataInicio', mode='before')
+    @classmethod
+    def formatar_data_inicio(cls, value: str) -> date:
+        try:
+            return datetime.strptime(value, "%Y-%m-%d").date()
+        except ValueError:
+            raise ValueError("A dataInicio deve estar no formato YYYY-MM-DD.")
+
+    # Validar e transformar dataFim (último dia do mês)
+    @field_validator('dataFim', mode='before')
+    @classmethod
+    def ajustar_data_fim(cls, value: str) -> date:
+        try:
+            ano, mes = map(int, value.split("-")[:2])
+            if mes < 1 or mes > 12:
+                raise ValueError
+            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()
+        except Exception:
+            raise ValueError("A dataFim deve estar no formato YYYY-MM-DD e conter um mês válido.")
+    

libretificacaotjcore/services/rabbitmq_consumer.py

@@ -0,0 +1,54 @@
+import pika
+import json
+
+
+class RabbitMQConsumer:
+    def __init__(
+        self,
+        host,
+        queue,
+        username,
+        password,
+        vhost="/",
+    ):
+        self.host = host
+        self.queue = queue
+        self.username = username
+        self.password = password
+        self.vhost = vhost
+        self.connection = None
+        self.channel = None
+
+    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,
+        )
+        
+        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)
+            
+        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()
+
+    def close(self):
+        if self.connection:
+            self.connection.close()

libretificacaotjcore/services/s3_service.py

@@ -0,0 +1,83 @@
+import asyncio
+from functools import partial
+import os
+import boto3
+from botocore.exceptions import ClientError
+
+class S3Service:
+    def __init__(self, aws_access_key_id, aws_secret_access_key, region_name, bucket_name, bucket_path):
+        self.s3 = boto3.client(
+            "s3",
+            aws_access_key_id=aws_access_key_id,
+            aws_secret_access_key=aws_secret_access_key,
+            region_name=region_name,
+        )
+        
+        self.bucket_name = bucket_name
+        self.bucket_path = bucket_path
+
+        self.loop = asyncio.get_running_loop()
+
+    async def save_on_s3(self, file_path, file_name):
+        try:
+            await self.loop.run_in_executor(
+                None,
+                self.s3.upload_file,
+                file_path,
+                self.bucket_name,
+                self.bucket_path + file_name,
+            )
+        except Exception as e:
+            print(f"❌ Erro ao salvar o arquivo no S3: {e}")
+
+    async def save_many_paths_on_s3(self, file_paths):
+        """
+        Salva vários arquivos no S3 informando apenas os caminhos locais.
+        O nome do arquivo no S3 será o mesmo do arquivo local.
+        """
+        tasks = [
+            self.save_on_s3(
+                file_path["caminho_arquivo_local"], file_path["nome_arquivo"]
+            )
+            for file_path in file_paths
+        ]
+        await asyncio.gather(*tasks)
+
+    async def file_on_s3(self, file_name):
+        try:
+            head_object_func = partial(
+                self.s3.head_object,
+                Bucket=self.bucket_name,
+                Key=self.bucket_path + file_name,
+            )
+
+            await self.loop.run_in_executor(None, head_object_func)
+            return True
+        except ClientError as e:
+            if e.response["Error"]["Code"] == "404":
+                return False
+
+            raise
+        except Exception as e:
+            print(f"❌ Erro ao obter o arquivo do S3: {e}")
+            return False
+
+    async def get_file_from_s3(self, file_name, destination_path):
+        try:
+            os.makedirs(os.path.dirname(destination_path), exist_ok=True)
+
+            key = self.bucket_path + file_name
+            print(f"🔍 Verificando chave: {key}")
+            download_func = partial(
+                self.s3.download_file, self.bucket_name, key, destination_path
+            )
+
+            await self.loop.run_in_executor(None, download_func)
+            print(f"📖✅ Arquivo '{file_name}' baixado com sucesso.")
+        except ClientError as e:
+            if e.response["Error"]["Code"] == "404":
+                print(f"⚠️ Arquivo '{file_name}' não encontrado no S3.")
+            else:
+                print(f"❌ Erro ao baixar o arquivo do S3: {e}")
+        except Exception as e:
+            print(f"❌ Erro inesperado ao baixar o arquivo do S3: {e}")

libretificacaotjcore-0.1.0.dist-info/METADATA

@@ -0,0 +1,27 @@
+Metadata-Version: 2.4
+Name: libretificacaotjcore
+Version: 0.1.0
+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
+Project-URL: Issues, https://github.com/seu-usuario/libretificacaotjcore/issues
+Project-URL: Repository, https://github.com/seu-usuario/libretificacaotjcore
+Keywords: tj,tributo justo,retificação,automação,pydantic,rabbitmq,boto3,motor
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Libraries
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+Requires-Dist: boto3>=1.39.16
+Requires-Dist: motor>=3.7.1
+Requires-Dist: pika>=1.3.2
+Requires-Dist: pydantic>=2.11.7

libretificacaotjcore-0.1.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+libretificacaotjcore/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+libretificacaotjcore/database/__init__.py,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
+libretificacaotjcore/database/arquivo_repository.py,sha256=Mt-5IWp7GVbMTAsCdFhi3TkIf2d__ccCGgJI_z-PE2Y,1118
+libretificacaotjcore/database/config_db.py,sha256=bixGQjK-QHk8KAqSadHMHZe1mU57FbVXI0rdW0PvNSs,1178
+libretificacaotjcore/dtos/solicitacao_dto.py,sha256=NEbC-A-OgVicw3WO3YfO5M5RrXoResya_20GGO_eJOM,2458
+libretificacaotjcore/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+libretificacaotjcore/services/rabbitmq_consumer.py,sha256=wj6GmNlWJiUpYRWNhfR_vhpN0tRJi5oKfZDyfpnOSQs,1684
+libretificacaotjcore/services/s3_service.py,sha256=HKR_jt2H3XdV1PCzo5R5bnhmoQ3I46Yn5IqAvVPhsjs,2946
+libretificacaotjcore-0.1.0.dist-info/METADATA,sha256=XFNQ8AFWB7ET883XfDwhV1DD5QGf162q1lEyUz4fSjw,1368
+libretificacaotjcore-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+libretificacaotjcore-0.1.0.dist-info/top_level.txt,sha256=J9vnz_X9OUnxC-eXHiAzlc9xIrWBwZ5bgnIDQIIFY4c,21
+libretificacaotjcore-0.1.0.dist-info/RECORD,,

libretificacaotjcore-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

libretificacaotjcore-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+libretificacaotjcore