Kralot
aff0446458
--- - Cria singleton de `client` com func `configure` para inicializar e `get_instance` para buscar instância do client; - Ajusta clients para buscar headers vindo do pai `client` e fixa versão legacy no client de databases; - Adiciona inicialização de `client` no init do projeto com api_token e api_version informados pelo usuário; - Altera `NotionConfig` para inserir `database_id` no lugar de `database_name`; - Altera sistema para receber `database_id` no lugar de `database_name`; - Altera tipo de `properties` em `schemas.responses.pages.Page` de `Union[Dict[str, Any]], TDB` para `Union[Any, TDB]` para resolver reclamações de type hint; - Adiciona param `generic_response` no init de `client` e nos clients e databases e pages para pular uso de mapping ao usar `.generic`; - Adiciona param `raw_response` para pular parser e mappings e retornar resposta original da api; - Finaliza `types` com subpastas para importações mas com init mãe vazio para evitar dependência circular e permitir uso de `notion.types.` pelo usuário; - Remove importações do projeto original não relacionadas com o SDK; - Adiciona param `timezone` na func `start_date` em `orm.common.SetProperty` que antes vinha do env, para posteriormente puxar da init da integração; - Monta `LICENSE`, `README.md` e `pyproject.toml` base simples para commit inicial do projeto permitindo build de pacote; ---
93 lines
2.4 KiB
Python
Executable File
93 lines
2.4 KiB
Python
Executable File
from typing import Dict, Any, Optional, Callable, Type
|
|
from .database import NotionDatabase as _NotionDatabase
|
|
|
|
class DatabaseRegistry:
|
|
|
|
"""
|
|
Registry global de databases do Notion.
|
|
Permite acessar schemas e criar parsers dinamicamente.
|
|
"""
|
|
|
|
_databases: Dict[str, Type[_NotionDatabase]] = {}
|
|
|
|
@classmethod
|
|
def register(cls,
|
|
database_class : Type[_NotionDatabase]
|
|
) -> None:
|
|
|
|
"Registra uma database no registry"
|
|
|
|
db_name = database_class.get_database_id()
|
|
cls._databases[db_name] = database_class
|
|
|
|
@classmethod
|
|
def get(cls,
|
|
database_id : str
|
|
) -> Optional[Type[_NotionDatabase]]:
|
|
|
|
"Retorna a classe de schema de uma database"
|
|
|
|
return cls._databases.get(database_id)
|
|
|
|
@classmethod
|
|
def get_parser(cls,
|
|
database_id : str,
|
|
page_parser : Callable
|
|
) -> Callable:
|
|
|
|
"""
|
|
Retorna uma função parser para a database.
|
|
|
|
Args:
|
|
database_id: ID da Database
|
|
page_parser: Função que parseia páginas (ex: PageProperties.parse)
|
|
|
|
Returns:
|
|
Função parser ou None se database não encontrada
|
|
"""
|
|
|
|
db_class = cls.get(database_id)
|
|
if not db_class:
|
|
raise ValueError(f"Database de ID '{database_id}' não está registrada")
|
|
|
|
def parser(page : Dict[str, Any]) -> Optional[_NotionDatabase]:
|
|
return db_class.from_notion_page(page, page_parser)
|
|
|
|
return parser
|
|
|
|
@classmethod
|
|
def list_databases(cls) -> list[str]:
|
|
|
|
"Lista todas as databases registradas"
|
|
|
|
return list(cls._databases.keys())
|
|
|
|
@classmethod
|
|
def auto_register(cls,
|
|
*database_classes : Type[_NotionDatabase]
|
|
) -> None:
|
|
|
|
"Registra múltiplas databases de uma vez"
|
|
|
|
for db_class in database_classes:
|
|
cls.register(db_class)
|
|
|
|
@classmethod
|
|
def generate_literal_type(cls) -> str:
|
|
|
|
"Gera o código do Literal com todas as databases registradas"
|
|
|
|
db_names = list(cls._databases.keys())
|
|
literal_str = ", ".join([f'"{name}"' for name in db_names])
|
|
|
|
return f"Literal[{literal_str}]"
|
|
|
|
@classmethod
|
|
def get_database_class_by_id(cls,
|
|
db_id : str
|
|
) -> Optional[Type[_NotionDatabase]]:
|
|
|
|
"Retorna a classe schema baseado no nome da database"
|
|
|
|
return cls._databases.get(db_id)
|