File size: 4,750 Bytes
47d267e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
import datetime
import gspread
import os
from dotenv import load_dotenv
from typing import Optional
# Importa as strings para usar mensagens padronizadas
from .strings import STRINGS
# --- Configuração Inicial do Google Sheets ---
load_dotenv()
# Obtém o caminho do arquivo da chave de serviço da variável de ambiente
service_account_key_path: Optional[str] = os.getenv('SERVICE_ACCOUNT_KEY_PATH')
spreadsheet_name: Optional[str] = os.getenv('GOOGLE_SHEETS_NAME')
# Variáveis globais para armazenar a conexão com o Google Sheets
# São inicializadas como None e configuradas pela função _initialize_google_sheets()
gc: Optional[gspread.Client] = None
worksheet: Optional[gspread.Worksheet] = None
initialization_error_message: Optional[str] = None
def _initialize_google_sheets() -> bool:
"""
Inicializa a conexão com o Google Sheets (gspread) e a planilha (worksheet) uma única vez.
Esta função tenta estabelecer a conexão com o Google Sheets usando as credenciais
fornecidas pelas variáveis de ambiente. Se a conexão for bem-sucedida, ela armazena
as referências globais 'gc' e 'worksheet'. Em caso de falha, armazena uma mensagem
de erro em 'initialization_error_message' e imprime no console.
Returns:
bool: True se a conexão e o acesso à planilha foram estabelecidos com sucesso,
False caso contrário.
"""
global gc, worksheet, initialization_error_message
if initialization_error_message:
return False
if gc and worksheet:
return True
if not service_account_key_path:
initialization_error_message = STRINGS["ERROR_ENV_KEY_PATH_NOT_DEFINED"]
print(initialization_error_message)
return False
if not spreadsheet_name:
initialization_error_message = STRINGS["ERROR_ENV_SHEET_NAME_NOT_DEFINED"]
print(initialization_error_message)
return False
try:
gc = gspread.service_account(filename=service_account_key_path)
except Exception as e:
initialization_error_message = f"{STRINGS['ERROR_LOAD_SERVICE_KEY']} {e}." # Verifique o caminho e o conteúdo do arquivo JSON.
print(initialization_error_message)
return False
try:
sh = gc.open(spreadsheet_name)
worksheet = sh.sheet1
print(STRINGS["INFO_CONNECTION_SUCCESS"])
return True
except gspread.exceptions.SpreadsheetNotFound:
initialization_error_message = f"{STRINGS['ERROR_SHEET_NOT_FOUND']} {spreadsheet_name}." # Verifique o nome e o compartilhamento com a conta de serviço.
print(initialization_error_message)
return False
except Exception as e:
initialization_error_message = f"{STRINGS['ERROR_ACCESS_SHEET']} {e}." # Verifique permissões ou o ID da planilha.
print(initialization_error_message)
return False
# Chama a função de inicialização uma única vez quando o módulo é carregado.
_initialize_google_sheets()
def submit_feedback_and_handle_errors(feedback_type: str, comment_text: str) -> str:
"""
Processa o feedback do usuário, valida a entrada e tenta enviá-lo para
o Google Sheets.
Gerencia erros de validação da entrada do usuário e erros na conexão
ou escrita no Google Sheets, retornando mensagens amigáveis para a UI.
Args:
feedback_type (str): O tipo de feedback selecionado pelo usuário (ex: "Bug", "Sugestão").
comment_text (str): O texto do comentário ou descrição do feedback.
Returns:
str: Uma mensagem de status para ser exibida na interface do usuário (Gradio),
indicando sucesso ou erro.
"""
# 1. Validação de campos do formulário Gradio
if not comment_text.strip():
return STRINGS["ERROR_COMMENT_EMPTY"]
if not feedback_type:
return STRINGS["ERROR_TYPE_EMPTY"]
# 2. Verificação de status da inicialização do Google Sheets
global gc, worksheet, initialization_error_message
if initialization_error_message:
return f"{STRINGS['ERROR_SERVICE_UNAVAILABLE_CONFIG']} {initialization_error_message}"
if not gc or not worksheet:
if not _initialize_google_sheets():
return f"{STRINGS['ERROR_SERVICE_UNAVAILABLE_CONNECTION']} {initialization_error_message}"
# Gera o timestamp atual no formato desejado.
timestamp: str = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
# Tenta adicionar a linha à planilha do Google Sheets.
try:
worksheet.append_row([timestamp, feedback_type, comment_text]) # type: ignore [union-attr]
return STRINGS["OUTPUT_SUCCESS"]
except Exception as e:
return f"{STRINGS['OUTPUT_ERROR']} {e}. Tente novamente mais tarde." |