"""Entrada de linha de comando para salvar um plano de edição no SQLite.""" from __future__ import annotations import argparse import json from pathlib import Path import sys CAMINHO_DO_CODIGO = Path(__file__).resolve().parent.parent if str(CAMINHO_DO_CODIGO) not in sys.path: sys.path.insert(0, str(CAMINHO_DO_CODIGO)) from engine.persistencia.conexao import abrir_banco from engine.persistencia.repositorio_de_planos import RepositorioDePlanos, plano_de_dict def registrar( caminho_do_plano: Path, caminho_do_banco: Path, video_id: str | None = None, tipo_de_video: str | None = None, modelo_da_ia: str | None = None, intencao: str | None = None, ) -> int: """Lê o JSON devolvido pela IA e registra o plano associado à edição mais recente.""" dados = json.loads(caminho_do_plano.read_text(encoding="utf-8")) conexao = abrir_banco(caminho_do_banco) try: if video_id is None: origem = dados.get("source") linha = conexao.execute( """SELECT video_id FROM edicoes_de_video WHERE origem = ? OR sequencia = ? ORDER BY concluida_em DESC, id DESC LIMIT 1""", (origem, origem), ).fetchone() video_id = linha["video_id"] if linha is not None else None plano = plano_de_dict( dados, video_id=video_id, tipo_de_video=tipo_de_video, modelo_da_ia=modelo_da_ia, intencao=intencao, ) return RepositorioDePlanos(conexao).registrar_plano(plano) finally: conexao.close() def principal() -> None: """Executa o registro do plano pela linha de comando.""" argumentos = argparse.ArgumentParser() argumentos.add_argument("plano", type=Path) argumentos.add_argument("--banco", type=Path, default=Path(".jhonny/analises.db")) argumentos.add_argument("--video-id") argumentos.add_argument("--tipo-de-video") argumentos.add_argument("--modelo-da-ia") argumentos.add_argument("--intencao") opcoes = argumentos.parse_args() try: identificador = registrar( opcoes.plano, opcoes.banco, video_id=opcoes.video_id, tipo_de_video=opcoes.tipo_de_video, modelo_da_ia=opcoes.modelo_da_ia, intencao=opcoes.intencao, ) print(json.dumps({"ok": True, "plano_id": identificador}, ensure_ascii=False)) except (OSError, ValueError) as erro: print(json.dumps({"ok": False, "erro": str(erro)}, ensure_ascii=False)) raise SystemExit(1) from erro if __name__ == "__main__": principal()