diff --git a/.gitignore b/.gitignore
index 5ac0cda..38daa5e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -8,3 +8,6 @@ app/core/__pycache__/renamer.cpython-311.pyc
app/core/__pycache__/state.cpython-311.pyc
app/core/__pycache__/watcher.cpython-311.pyc
app/data/clei.db
+app/ui/__pycache__/settings.cpython-311.pyc
+app/ui/__pycache__/manual_tools.cpython-311.pyc
+app/__pycache__/main.cpython-311.pyc
diff --git a/app/core/bot.py b/app/core/bot.py
index ee15807..2bb7b70 100755
--- a/app/core/bot.py
+++ b/app/core/bot.py
@@ -1,23 +1,21 @@
import asyncio
-import httpx # Usamos para testar a rede antes
+import httpx
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.ext import ApplicationBuilder, ContextTypes, CallbackQueryHandler, CommandHandler
from database import AppConfig
import logging
-# Configuração de Log
logging.getLogger("httpx").setLevel(logging.WARNING)
logging.getLogger("telegram").setLevel(logging.INFO)
class TelegramManager:
def __init__(self):
- # NÃO carregamos o token aqui. Carregamos na hora de iniciar.
self.app = None
self.active_requests = {}
self.is_connected = False
async def check_internet(self):
- """Testa se o container tem internet antes de tentar o Telegram"""
+ """Testa se o container tem internet"""
try:
async with httpx.AsyncClient(timeout=5.0) as client:
await client.get("https://www.google.com")
@@ -27,85 +25,87 @@ class TelegramManager:
async def start(self):
"""Inicia o Bot"""
- # 1. Pega o Token Fresquinho do Banco
token = AppConfig.get_val('telegram_token')
chat_id = AppConfig.get_val('telegram_chat_id')
if not token:
- print("🟡 Bot: Token não configurado. Aguardando você salvar no Painel...")
+ print("🟡 Bot: Token não configurado.")
return
- # 2. Teste de Rede Prévio
- print("🤖 Bot: Verificando conectividade...")
+ if self.is_connected: return
+
+ print("🤖 Bot: Iniciando conexão...")
if not await self.check_internet():
- print("❌ Bot: ERRO DE REDE! O container não consegue acessar a internet.")
- print(" -> Verifique DNS ou Firewall.")
+ print("❌ Bot: Sem internet.")
return
-
- print(f"🤖 Bot: Conectando com token termina em ...{token[-5:]}")
try:
- # 3. Constroi a Aplicação
self.app = ApplicationBuilder().token(token).build()
-
- # Handlers
self.app.add_handler(CommandHandler("start", self.cmd_start))
self.app.add_handler(CommandHandler("id", self.cmd_id))
self.app.add_handler(CallbackQueryHandler(self.handle_selection))
- # Inicializa
await self.app.initialize()
await self.app.start()
-
- # Inicia Polling (Limpa mensagens velhas acumuladas para não travar)
await self.app.updater.start_polling(drop_pending_updates=True)
self.is_connected = True
- print("✅ Bot Online e Rodando!")
+ print("✅ Bot Online!")
- # Tenta mandar um oi se tiver chat_id
- if chat_id:
- try:
- await self.app.bot.send_message(chat_id=chat_id, text="🚀 Clei-Flow: Conexão restabelecida!")
- except Exception as e:
- print(f"⚠️ Bot online, mas falhou ao enviar msg (Chat ID errado?): {e}")
-
except Exception as e:
- print(f"❌ Falha Crítica no Bot: {e}")
+ print(f"❌ Falha no Bot: {e}")
self.is_connected = False
async def stop(self):
+ """Para o Bot"""
if self.app:
try:
- await self.app.updater.stop()
- await self.app.stop()
- await self.app.shutdown()
+ if self.app.updater.running:
+ await self.app.updater.stop()
+ if self.app.running:
+ await self.app.stop()
+ await self.app.shutdown()
+ except Exception as e:
+ print(f"Erro ao parar bot: {e}")
+ finally:
self.is_connected = False
- except: pass
+ self.app = None
+
+ async def restart(self):
+ """Reinicia a conexão (Útil após mudar token)"""
+ print("🔄 Bot: Reiniciando serviço...")
+ await self.stop()
+ await asyncio.sleep(1)
+ await self.start()
# --- COMANDOS ---
async def cmd_start(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
chat_id = update.effective_chat.id
await update.message.reply_text(f"Olá! Configurado.\nSeu Chat ID é: `{chat_id}`", parse_mode='Markdown')
- # Opcional: Salvar o Chat ID automaticamente se o usuário mandar /start
- # AppConfig.set_val('telegram_chat_id', str(chat_id))
async def cmd_id(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
await update.message.reply_text(f"`{update.effective_chat.id}`", parse_mode='Markdown')
- # --- INTERAÇÃO (Renomeação) ---
+ async def send_test_msg(self):
+ """Envia mensagem de teste"""
+ if not self.is_connected: return False
+ chat_id = AppConfig.get_val('telegram_chat_id')
+ if not chat_id: return False
+ try:
+ await self.app.bot.send_message(chat_id=chat_id, text="🚀 Clei-Flow: Teste de conexão bem sucedido!")
+ return True
+ except:
+ return False
+
+ # --- INTERAÇÃO ---
async def ask_user_choice(self, filename, candidates):
- chat_id = AppConfig.get_val('telegram_chat_id') # Pega sempre o atual
- if not chat_id or not self.is_connected:
- print("❌ Bot não pode perguntar (Sem Chat ID ou Desconectado)")
- return None
+ chat_id = AppConfig.get_val('telegram_chat_id')
+ if not chat_id or not self.is_connected: return None
request_id = f"req_{filename}"
keyboard = []
for cand in candidates:
- # Texto do botão
text = f"{cand['title']} ({cand['year']})"
- # Dados (ID|Tipo)
callback_data = f"{request_id}|{cand['tmdb_id']}|{cand['type']}"
keyboard.append([InlineKeyboardButton(text, callback_data=callback_data)])
@@ -115,20 +115,17 @@ class TelegramManager:
try:
await self.app.bot.send_message(
chat_id=chat_id,
- text=f"🤔 Clei-Flow Precisa de Ajuda:\nArquivo: {filename}",
+ text=f"🤔 Clei-Flow:\nArquivo: {filename}",
reply_markup=reply_markup,
parse_mode='HTML'
)
- except Exception as e:
- print(f"Erro ao enviar pergunta: {e}")
- return None
+ except: return None
loop = asyncio.get_running_loop()
future = loop.create_future()
self.active_requests[request_id] = future
try:
- # Espera 12 horas
result = await asyncio.wait_for(future, timeout=43200)
return result
except asyncio.TimeoutError:
@@ -156,7 +153,7 @@ class TelegramManager:
future.set_result({'tmdb_id': int(tmdb_id), 'type': media_type})
del self.active_requests[req_id]
else:
- await query.edit_message_text(text="⚠️ Solicitação expirada.")
+ await query.edit_message_text(text="⚠️ Expirado.")
async def send_notification(self, message):
chat_id = AppConfig.get_val('telegram_chat_id')
diff --git a/app/database.py b/app/database.py
index 5aa5fd9..f43f12a 100755
--- a/app/database.py
+++ b/app/database.py
@@ -29,12 +29,10 @@ class AppConfig(BaseModel):
class Category(BaseModel):
name = CharField(unique=True)
target_path = CharField()
- match_keywords = CharField(null=True) # Mantido para legado, mas vamos priorizar os filtros abaixo
- content_type = CharField(default='mixed') # movie, series, mixed
-
- # NOVOS CAMPOS DE FILTRAGEM
- genre_filters = CharField(null=True) # Ex: "16,28,35" (IDs do TMDb)
- country_filters = CharField(null=True) # Ex: "JP,US,BR" (Siglas ISO)
+ match_keywords = CharField(null=True)
+ content_type = CharField(default='mixed')
+ genre_filters = CharField(null=True)
+ country_filters = CharField(null=True)
class FFmpegProfile(BaseModel):
name = CharField()
@@ -46,21 +44,20 @@ class FFmpegProfile(BaseModel):
is_active = BooleanField(default=False)
def init_db():
- db.connect()
+ # --- CORREÇÃO AQUI: Verifica se já está conectado ---
+ if db.is_closed():
+ db.connect()
+
db.create_tables([AppConfig, Category, FFmpegProfile], safe=True)
- # Migrações Seguras (Adiciona colunas se não existirem)
try: db.execute_sql('ALTER TABLE category ADD COLUMN content_type VARCHAR DEFAULT "mixed"')
except: pass
-
try: db.execute_sql('ALTER TABLE category ADD COLUMN genre_filters VARCHAR DEFAULT ""')
except: pass
-
try: db.execute_sql('ALTER TABLE category ADD COLUMN country_filters VARCHAR DEFAULT ""')
except: pass
- # Perfil padrão
if FFmpegProfile.select().count() == 0:
FFmpegProfile.create(name="Padrão VAAPI (Intel)", video_codec="h264_vaapi", is_active=True)
- db.close()
\ No newline at end of file
+ # Não fechamos a conexão aqui para manter o pool ativo no container
\ No newline at end of file
diff --git a/app/ui/__pycache__/manual_tools.cpython-311.pyc b/app/ui/__pycache__/manual_tools.cpython-311.pyc
index 969cd4e..4a7e344 100644
Binary files a/app/ui/__pycache__/manual_tools.cpython-311.pyc and b/app/ui/__pycache__/manual_tools.cpython-311.pyc differ
diff --git a/app/ui/__pycache__/settings.cpython-311.pyc b/app/ui/__pycache__/settings.cpython-311.pyc
index df80688..672b90c 100644
Binary files a/app/ui/__pycache__/settings.cpython-311.pyc and b/app/ui/__pycache__/settings.cpython-311.pyc differ
diff --git a/app/ui/manual_tools.py b/app/ui/manual_tools.py
index 182bb33..3878c3d 100755
--- a/app/ui/manual_tools.py
+++ b/app/ui/manual_tools.py
@@ -1,2 +1,450 @@
-from nicegui import ui
-# Abas antigas (Renomeador Manual, Encoder Manual) viram ferramentas aqui
+from nicegui import ui, run
+import os
+import shutil
+import asyncio
+import datetime
+import subprocess
+import json
+
+# Define a raiz como /downloads, mas permite navegar se o container tiver permissão
+ROOT_DIR = "/downloads"
+
+# --- UTILITÁRIOS ASSÍNCRONOS ---
+async def get_human_size(size):
+ for unit in ['B', 'KB', 'MB', 'GB']:
+ if size < 1024: return f"{size:.2f} {unit}"
+ size /= 1024
+ return f"{size:.2f} TB"
+
+async def get_media_info_async(filepath):
+ def _probe():
+ cmd = ["ffprobe", "-v", "quiet", "-print_format", "json", "-show_streams", "-show_format", filepath]
+ try:
+ res = subprocess.run(cmd, capture_output=True, text=True, timeout=10)
+ return json.loads(res.stdout)
+ except: return None
+
+ data = await run.io_bound(_probe)
+ if not data: return None
+
+ # Tratamento de erros se chaves não existirem
+ fmt = data.get('format', {})
+
+ info = {
+ "filename": os.path.basename(filepath),
+ "size": int(fmt.get('size', 0)),
+ "duration": float(fmt.get('duration', 0)),
+ "bitrate": int(fmt.get('bit_rate', 0)),
+ "video": [], "audio": [], "subtitle": []
+ }
+
+ for s in data.get('streams', []):
+ stype = s.get('codec_type')
+ lang = s.get('tags', {}).get('language', 'und').upper()
+ codec = s.get('codec_name', 'unknown').upper()
+
+ if stype == 'video':
+ info['video'].append({
+ "codec": codec,
+ "res": f"{s.get('width','?')}x{s.get('height','?')}",
+ "fps": s.get('r_frame_rate', '')
+ })
+ elif stype == 'audio':
+ info['audio'].append({
+ "lang": lang,
+ "codec": codec,
+ "ch": s.get('channels', 0),
+ "title": s.get('tags', {}).get('title', '')
+ })
+ elif stype == 'subtitle':
+ info['subtitle'].append({"lang": lang, "codec": codec})
+
+ return info
+
+# --- CLASSE GERENCIADORA ---
+class FileManager:
+ def __init__(self):
+ self.path = ROOT_DIR
+ self.view_mode = 'grid'
+ self.is_selecting = False
+ self.selected_items = set()
+ self.search_term = ""
+ self.refreshing = False
+
+ # Elementos UI
+ self.container_content = None
+ self.footer = None
+ self.lbl_selection_count = None
+ self.btn_select_mode = None
+ self.header_row = None
+
+ # --- NAVEGAÇÃO ---
+ async def navigate(self, path):
+ if os.path.exists(path) and os.path.isdir(path):
+ self.path = path
+ self.selected_items.clear()
+ self.is_selecting = False
+ self.search_term = ""
+ self.update_footer_state()
+
+ if self.header_row:
+ self.header_row.clear()
+ with self.header_row:
+ self.build_header_content()
+ await self.refresh()
+ else:
+ ui.notify('Caminho inválido.', type='negative')
+
+ async def navigate_up(self):
+ parent = os.path.dirname(self.path)
+ # Permite subir até a raiz do sistema se necessário, mas idealmente trava no ROOT
+ # Se quiser travar: if self.path != ROOT_DIR: await self.navigate(parent)
+ if parent and os.path.exists(parent):
+ await self.navigate(parent)
+
+ # --- UPLOAD ---
+ def open_upload_dialog(self):
+ with ui.dialog() as dialog, ui.card():
+ ui.label(f'Upload para: {os.path.basename(self.path)}').classes('font-bold')
+
+ async def handle(e):
+ try:
+ name = None
+ if hasattr(e, 'file'):
+ name = getattr(e.file, 'filename', None) or getattr(e.file, 'name', None)
+ if not name: name = getattr(e, 'name', 'arquivo_sem_nome')
+
+ content = b''
+ if hasattr(e, 'content'): content = e.content.read() # NiceGUI as vezes passa assim
+ elif hasattr(e, 'file'):
+ if hasattr(e.file, 'seek'): await e.file.seek(0)
+ if hasattr(e.file, 'read'): content = await e.file.read()
+
+ if not content:
+ ui.notify('Erro: Arquivo vazio', type='warning'); return
+
+ target = os.path.join(self.path, name)
+ await run.io_bound(self._save_file_bytes, target, content)
+ ui.notify(f'Sucesso: {name}', type='positive')
+ except Exception as ex:
+ ui.notify(f'Erro: {ex}', type='negative')
+
+ ui.upload(on_upload=handle, auto_upload=True, multiple=True).props('accept=*').classes('w-full')
+
+ async def close_and_refresh():
+ dialog.close(); await self.refresh()
+ ui.button('Fechar e Atualizar', on_click=close_and_refresh).props('flat w-full')
+ dialog.open()
+
+ def _save_file_bytes(self, target, content_bytes):
+ with open(target, 'wb') as f: f.write(content_bytes)
+
+ # --- SELEÇÃO ---
+ def toggle_select_mode(self):
+ self.is_selecting = not self.is_selecting
+ if not self.is_selecting: self.selected_items.clear()
+ self.update_footer_state()
+ self.refresh_ui_only()
+
+ def toggle_selection(self, item_path):
+ if item_path in self.selected_items: self.selected_items.remove(item_path)
+ else: self.selected_items.add(item_path)
+ self.update_footer_state()
+ self.refresh_ui_only() # Refresh leve
+
+ def select_all(self, entries):
+ current = {e.path for e in entries}
+ if self.selected_items.issuperset(current): self.selected_items.difference_update(current)
+ else: self.selected_items.update(current)
+ self.update_footer_state()
+ self.refresh_ui_only()
+
+ def update_footer_state(self):
+ if self.footer:
+ self.footer.set_visibility(self.is_selecting)
+ if self.lbl_selection_count:
+ self.lbl_selection_count.text = f'{len(self.selected_items)} item(s)'
+
+ # --- AÇÕES ---
+ async def delete_selected(self):
+ if not self.selected_items: return
+ with ui.dialog() as dialog, ui.card():
+ ui.label(f'Excluir {len(self.selected_items)} itens?').classes('font-bold text-red')
+ async def confirm():
+ dialog.close()
+ count = 0
+ items_copy = list(self.selected_items)
+ for item in items_copy:
+ try:
+ if os.path.isdir(item): await run.io_bound(shutil.rmtree, item)
+ else: await run.io_bound(os.remove, item)
+ count += 1
+ except: pass
+ ui.notify(f'{count} excluídos.', type='positive')
+ self.selected_items.clear()
+ self.update_footer_state()
+ await self.refresh()
+ with ui.row().classes('w-full justify-end'):
+ ui.button('Cancelar', on_click=dialog.close).props('flat')
+ ui.button('Confirmar', on_click=confirm).props('color=red')
+ dialog.open()
+
+ async def open_move_dialog(self, target_items=None):
+ items_to_move = target_items if target_items else list(self.selected_items)
+ if not items_to_move:
+ ui.notify('Nada para mover.', type='warning'); return
+
+ browser_path = ROOT_DIR
+ with ui.dialog() as dialog, ui.card().classes('w-96 h-[500px] flex flex-col p-4'):
+ ui.label(f'Mover {len(items_to_move)} item(s) para...').classes('font-bold')
+ lbl_path = ui.label(ROOT_DIR).classes('text-xs bg-gray-100 p-2 w-full truncate border rounded')
+ scroll = ui.scroll_area().classes('flex-grow border rounded p-1 bg-white')
+
+ async def load_folders(p):
+ nonlocal browser_path; browser_path = p
+ lbl_path.text = p; scroll.clear()
+ try:
+ entries = await run.io_bound(os.scandir, p)
+ sorted_e = sorted([e for e in entries if e.is_dir() and not e.name.startswith('.')], key=lambda e: e.name.lower())
+ with scroll:
+ parent = os.path.dirname(p)
+ if parent and os.path.exists(parent):
+ ui.button('..', on_click=lambda: load_folders(parent)).props('flat dense icon=arrow_upward w-full align=left')
+ for e in sorted_e:
+ ui.button(e.name, on_click=lambda path=e.path: load_folders(path)).props('flat dense w-full align=left icon=folder color=amber')
+ except: pass
+
+ ui.timer(0, lambda: load_folders(ROOT_DIR), once=True)
+
+ async def execute_move():
+ dialog.close(); ui.notify('Movendo...', type='info')
+ count = 0
+ for item in items_to_move:
+ try:
+ tgt = os.path.join(browser_path, os.path.basename(item))
+ if item != tgt:
+ await run.io_bound(shutil.move, item, tgt)
+ count += 1
+ except Exception as e: ui.notify(f"Erro: {e}", type='negative')
+
+ ui.notify(f'{count} movidos!', type='positive')
+ if not target_items: self.selected_items.clear()
+ self.update_footer_state()
+ await self.refresh()
+
+ with ui.row().classes('w-full justify-between mt-auto'):
+ ui.button('Cancelar', on_click=dialog.close).props('flat')
+ ui.button('Mover Aqui', on_click=execute_move).props('color=green icon=drive_file_move')
+ dialog.open()
+
+ # --- LAYOUT ---
+ def create_layout(self):
+ # Header
+ self.header_row = ui.row().classes('w-full items-center bg-gray-100 p-2 rounded-lg gap-2 sticky top-0 z-20 shadow-sm')
+ with self.header_row: self.build_header_content()
+
+ # Conteúdo
+ self.container_content = ui.column().classes('w-full gap-4 pb-24')
+
+ # Footer (Simulado)
+ with ui.row().classes('fixed bottom-0 left-0 w-full z-50 bg-white border-t p-2 justify-center gap-4 shadow-[0_-4px_10px_rgba(0,0,0,0.1)]') as f:
+ self.footer = f
+ self.lbl_selection_count = ui.label('0 selecionados').classes('font-bold self-center')
+ ui.button('Mover', on_click=lambda: self.open_move_dialog(None)).props('color=amber icon=drive_file_move dense')
+ ui.button('Excluir', on_click=self.delete_selected).props('color=red icon=delete dense')
+ ui.button(icon='close', on_click=self.toggle_select_mode).props('flat round dense')
+ self.footer.visible = False
+
+ def build_header_content(self):
+ if self.path != ROOT_DIR:
+ ui.button(icon='arrow_upward', on_click=self.navigate_up).props('flat round dense')
+ else:
+ ui.button(icon='home').props('flat round dense disabled text-color=grey')
+
+ # Breadcrumbs
+ rel = os.path.relpath(self.path, ROOT_DIR) if self.path.startswith(ROOT_DIR) else self.path
+ parts = rel.split(os.sep) if rel != '.' else []
+
+ with ui.row().classes('items-center gap-0 overflow-hidden flex-grow'):
+ ui.button('root', on_click=lambda: self.navigate(ROOT_DIR)).props('flat dense no-caps min-w-0 px-1 text-xs')
+ acc = ROOT_DIR
+ for part in parts:
+ acc = os.path.join(acc, part)
+ ui.label('/').classes('text-gray-400')
+ ui.button(part, on_click=lambda p=acc: self.navigate(p)).props('flat dense no-caps min-w-0 px-1 text-xs')
+
+ # Toolbar
+ self.btn_select_mode = ui.button(icon='check_box', on_click=self.toggle_select_mode).props('flat round dense')
+ ui.button(icon='create_new_folder', on_click=self.open_create_folder).props('flat round dense')
+ ui.button(icon='cloud_upload', on_click=self.open_upload_dialog).props('flat round dense')
+ ui.button(icon='view_list', on_click=self.toggle_view).props('flat round dense')
+ ui.button(icon='refresh', on_click=self.refresh).props('flat round dense')
+
+ def toggle_view(self):
+ self.view_mode = 'list' if self.view_mode == 'grid' else 'grid'
+ self.refresh_ui_only()
+
+ def open_create_folder(self):
+ with ui.dialog() as dialog, ui.card():
+ ui.label('Nova Pasta'); name = ui.input('Nome')
+ async def create():
+ try:
+ await run.io_bound(os.makedirs, os.path.join(self.path, name.value))
+ dialog.close(); await self.refresh()
+ except Exception as e: ui.notify(str(e), type='negative')
+ ui.button('Criar', on_click=create)
+ dialog.open()
+
+ def open_rename_dialog(self, path):
+ with ui.dialog() as dialog, ui.card():
+ ui.label('Renomear'); name = ui.input('Novo Nome', value=os.path.basename(path)).classes('w-full')
+ async def save():
+ try:
+ new_path = os.path.join(os.path.dirname(path), name.value)
+ await run.io_bound(os.rename, path, new_path)
+ dialog.close(); await self.refresh()
+ except Exception as e: ui.notify(str(e), type='negative')
+ ui.button('Salvar', on_click=save)
+ dialog.open()
+
+ # --- RENDERIZAÇÃO ---
+ async def refresh(self):
+ if self.refreshing: return
+ self.refreshing = True
+
+ # Recarrega arquivos do disco
+ try:
+ entries = await run.io_bound(os.scandir, self.path)
+ self.cached_entries = sorted(entries, key=lambda e: (not e.is_dir(), e.name.lower()))
+ except Exception as e:
+ self.cached_entries = []
+ ui.notify(f"Erro leitura: {e}", type='negative')
+
+ self.refresh_ui_only()
+ self.refreshing = False
+
+ def refresh_ui_only(self):
+ if self.container_content:
+ self.container_content.clear()
+
+ color = 'green' if self.is_selecting else 'grey'
+ if self.btn_select_mode: self.btn_select_mode.props(f'text-color={color}')
+
+ # Select All Bar
+ if self.is_selecting:
+ with self.container_content:
+ with ui.row().classes('w-full px-2 py-1 bg-green-50 items-center justify-between text-xs text-green-800 rounded border border-green-200'):
+ ui.checkbox(f'Todos ({len(self.cached_entries)})', on_change=lambda: self.select_all(self.cached_entries)).props('dense size=xs')
+
+ # Content
+ with self.container_content:
+ if not self.cached_entries:
+ ui.label('Pasta Vazia').classes('w-full text-center text-gray-400 mt-10')
+ elif self.view_mode == 'grid':
+ with ui.grid().classes('w-full grid-cols-2 sm:grid-cols-3 md:grid-cols-4 lg:grid-cols-6 gap-3'):
+ for entry in self.cached_entries: self.render_card(entry)
+ else:
+ with ui.column().classes('w-full gap-0'):
+ for entry in self.cached_entries: self.render_list_item(entry)
+
+ def render_card(self, entry):
+ is_sel = entry.path in self.selected_items
+ is_dir = entry.is_dir()
+ icon = 'folder' if is_dir else 'description'
+ if not is_dir and entry.name.lower().endswith(('.mkv', '.mp4', '.avi')): icon = 'movie'
+
+ color = 'amber-8' if is_dir else ('purple-6' if icon=='movie' else 'blue-grey')
+ bg = 'bg-green-100 ring-2 ring-green-500' if is_sel else 'bg-white hover:shadow-md'
+
+ with ui.card().classes(f'w-full aspect-square p-2 items-center justify-center relative group cursor-pointer select-none {bg}') as card:
+ self.bind_context_menu(card, entry)
+ if self.is_selecting:
+ card.on('click', lambda: self.toggle_selection(entry.path))
+ ui.checkbox(value=is_sel, on_change=lambda: self.toggle_selection(entry.path)).props('dense').classes('absolute top-1 left-1 z-10').on('click', lambda e: e.stop_propagation())
+ else:
+ if is_dir: card.on('click', lambda e, p=entry.path: self.navigate(p))
+ elif icon == 'movie': card.on('click', lambda e, p=entry.path: self.open_inspector(p))
+
+ ui.icon(icon, size='3rem', color=color)
+ ui.label(entry.name).classes('text-xs text-center leading-tight line-clamp-2 w-full break-all mt-2')
+
+ def render_list_item(self, entry):
+ is_sel = entry.path in self.selected_items
+ bg = 'bg-green-100' if is_sel else 'hover:bg-gray-50'
+ icon = 'folder' if entry.is_dir() else 'description'
+
+ with ui.row().classes(f'w-full items-center px-2 py-2 border-b cursor-pointer group {bg}') as row:
+ if self.is_selecting:
+ ui.checkbox(value=is_sel, on_change=lambda: self.toggle_selection(entry.path)).props('dense')
+ row.on('click', lambda: self.toggle_selection(entry.path))
+ else:
+ if entry.is_dir(): row.on('click', lambda p=entry.path: self.navigate(p))
+
+ self.bind_context_menu(row, entry)
+ ui.icon(icon, color='amber' if entry.is_dir() else 'grey')
+ ui.label(entry.name).classes('text-sm font-medium flex-grow')
+
+ def bind_context_menu(self, element, entry):
+ with ui.menu() as m:
+ if not entry.is_dir() and entry.name.lower().endswith(('.mkv', '.mp4', '.avi')):
+ ui.menu_item('Media Info', on_click=lambda: self.open_inspector(entry.path))
+ ui.menu_item('Renomear', on_click=lambda: self.open_rename_dialog(entry.path))
+ ui.menu_item('Mover Para...', on_click=lambda: self.open_move_dialog([entry.path]))
+
+ async def delete_single():
+ try:
+ if entry.is_dir(): await run.io_bound(shutil.rmtree, entry.path)
+ else: await run.io_bound(os.remove, entry.path)
+ await self.refresh(); ui.notify('Excluído.', type='positive')
+ except Exception as e: ui.notify(str(e), type='negative')
+ ui.menu_item('Excluir', on_click=delete_single).props('text-color=red')
+ element.on('contextmenu.prevent', lambda: m.open())
+
+ # --- INSPECTOR ---
+ async def open_inspector(self, path):
+ dialog = ui.dialog()
+ with dialog, ui.card().classes('w-full max-w-3xl'):
+ with ui.row().classes('w-full justify-between items-start'):
+ ui.label(os.path.basename(path)).classes('text-lg font-bold break-all w-10/12 text-blue-800')
+ ui.button(icon='close', on_click=dialog.close).props('flat dense round')
+
+ content = ui.column().classes('w-full')
+ with content: ui.spinner('dots').classes('self-center')
+ dialog.open()
+
+ info = await get_media_info_async(path)
+ content.clear()
+
+ if not info:
+ with content: ui.label('Erro ao ler metadados.').classes('text-red'); return
+
+ with content:
+ with ui.row().classes('w-full bg-blue-50 p-2 rounded mb-4 gap-4'):
+ ui.label(f"⏱️ {datetime.timedelta(seconds=int(info['duration']))}")
+ ui.label(f"📦 {await get_human_size(info['size'])}")
+ ui.label(f"🚀 {int(info['bitrate']/1000)} kbps")
+
+ ui.label('Vídeo').classes('text-xs font-bold text-gray-500 uppercase mt-2')
+ for v in info['video']:
+ with ui.card().classes('w-full p-2 bg-gray-50 border-l-4 border-blue-500'):
+ ui.label(f"{v['codec']} • {v['res']} • {v['fps']} fps").classes('font-bold')
+
+ ui.label('Áudio').classes('text-xs font-bold text-gray-500 uppercase mt-4')
+ if info['audio']:
+ with ui.grid().classes('grid-cols-[auto_1fr_auto] w-full gap-2 text-sm'):
+ for a in info['audio']:
+ ui.label(a['lang']).classes('font-bold bg-gray-200 px-2 rounded text-center')
+ ui.label(f"{a['codec']} - {a['title']}")
+ ui.label(str(a['ch'])).classes('text-gray-500')
+
+ ui.label('Legendas').classes('text-xs font-bold text-gray-500 uppercase mt-4')
+ if info['subtitle']:
+ with ui.row().classes('w-full gap-2'):
+ for s in info['subtitle']:
+ color = 'green' if s['codec'] in ['subrip', 'ass'] else 'grey'
+ ui.chip(f"{s['lang']} ({s['codec']})", color=color).props('dense icon=subtitles')
+
+def show():
+ fm = FileManager()
+ fm.create_layout()
+ ui.timer(0, fm.refresh, once=True)
\ No newline at end of file
diff --git a/app/ui/settings.py b/app/ui/settings.py
index 92023ab..a08a6c4 100755
--- a/app/ui/settings.py
+++ b/app/ui/settings.py
@@ -1,10 +1,10 @@
-from nicegui import ui
+from nicegui import ui, app
import os
+import asyncio
from database import Category, FFmpegProfile, AppConfig
+from core.state import state # <--- ACESSAMOS O BOT POR AQUI AGORA
-# --- CONSTANTES DE DADOS ---
-
-# 1. Gêneros TMDb (Para as Categorias)
+# --- CONSTANTES ---
TMDB_GENRES = {
'28': 'Ação (Action)', '12': 'Aventura (Adventure)', '16': 'Animação (Animation)',
'35': 'Comédia (Comedy)', '80': 'Crime', '99': 'Documentário', '18': 'Drama',
@@ -14,15 +14,11 @@ TMDB_GENRES = {
'53': 'Thriller', '10752': 'Guerra (War)', '37': 'Faroeste (Western)',
'10759': 'Action & Adventure (TV)', '10762': 'Kids (TV)', '10765': 'Sci-Fi & Fantasy (TV)'
}
-
-# 2. Países Comuns (Para as Categorias)
COMMON_COUNTRIES = {
'JP': '🇯🇵 Japão (Anime)', 'US': '🇺🇸 EUA (Hollywood)', 'BR': '🇧🇷 Brasil',
'KR': '🇰🇷 Coreia do Sul (Dorama)', 'GB': '🇬🇧 Reino Unido', 'ES': '🇪🇸 Espanha',
'FR': '🇫🇷 França', 'CN': '🇨🇳 China', 'IN': '🇮🇳 Índia'
}
-
-# 3. Idiomas ISO-639-2 (Para o FFmpeg - ESTAVA FALTANDO ISSO)
ISO_LANGS = {
'por': 'Português (por)', 'eng': 'Inglês (eng)', 'jpn': 'Japonês (jpn)',
'spa': 'Espanhol (spa)', 'fra': 'Francês (fra)', 'ger': 'Alemão (ger)',
@@ -30,41 +26,78 @@ ISO_LANGS = {
'zho': 'Chinês (zho)', 'und': 'Indefinido (und)'
}
-# --- SELETOR DE PASTAS ---
-async def pick_folder_dialog(start_path='/media'):
- ALLOWED_ROOT = '/media'
- if not start_path or not start_path.startswith(ALLOWED_ROOT): start_path = ALLOWED_ROOT
+# --- SELETOR DE PASTAS FLEXÍVEL ---
+async def pick_folder_dialog(start_path='/', allowed_root='/'):
+ """
+ allowed_root: '/' permite tudo, '/media' restringe.
+ """
+ # Se start_path for vazio ou None, começa na raiz permitida
+ if not start_path: start_path = allowed_root
+
+ # Se o caminho salvo não começar com a raiz permitida, reseta
+ if allowed_root != '/' and not start_path.startswith(allowed_root):
+ start_path = allowed_root
+
result = {'path': None}
+
with ui.dialog() as dialog, ui.card().classes('w-96 h-[500px] flex flex-col'):
- ui.label('Selecionar Pasta (/media)').classes('font-bold text-lg mb-2')
+ ui.label(f'Selecionar Pasta ({allowed_root})').classes('font-bold text-lg mb-2')
path_label = ui.label(start_path).classes('text-xs bg-gray-100 p-2 border rounded w-full break-all font-mono')
+
scroll = ui.scroll_area().classes('flex-grow border rounded p-1 mt-2 bg-white')
+
async def load_dir(path):
- if not path.startswith(ALLOWED_ROOT): path = ALLOWED_ROOT
+ # Segurança: Garante que não saiu da jaula (se houver jaula)
+ if allowed_root != '/' and not path.startswith(allowed_root):
+ path = allowed_root
+
path_label.text = path
scroll.clear()
+
try:
- if path != ALLOWED_ROOT:
+ # Botão Voltar (..)
+ # Só mostra se não estiver na raiz do sistema OU na raiz permitida
+ show_back = True
+ if path == '/': show_back = False
+ if allowed_root != '/' and path == allowed_root: show_back = False
+
+ if show_back:
parent = os.path.dirname(path)
- if not parent.startswith(ALLOWED_ROOT): parent = ALLOWED_ROOT
- with scroll: ui.button('.. (Voltar)', on_click=lambda: load_dir(parent)).props('flat dense icon=arrow_upward align=left w-full')
+ with scroll:
+ ui.button('.. (Voltar)', on_click=lambda: load_dir(parent)).props('flat dense icon=arrow_upward align=left w-full')
+
+ # Lista Pastas
with scroll:
if os.path.exists(path):
- for entry in sorted([e for e in os.scandir(path) if e.is_dir()], key=lambda x: x.name.lower()):
+ entries = sorted([e for e in os.scandir(path) if e.is_dir()], key=lambda x: x.name.lower())
+ for entry in entries:
ui.button(entry.name, on_click=lambda p=entry.path: load_dir(p)).props('flat dense icon=folder align=left w-full color=amber-8')
- else: ui.label('Pasta nova').classes('text-gray-400 italic p-2')
+ else:
+ ui.label('Caminho não encontrado.').classes('text-gray-400 italic p-2')
+
except Exception as e:
- with scroll: ui.label(f'Erro: {e}').classes('text-red text-xs')
- def select_this(): result['path'] = path_label.text; dialog.close()
+ with scroll: ui.label(f'Acesso negado: {e}').classes('text-red text-xs')
+
+ def select_this():
+ result['path'] = path_label.text
+ dialog.close()
+
with ui.row().classes('w-full justify-between mt-auto pt-2'):
ui.button('Cancelar', on_click=dialog.close).props('flat color=grey')
ui.button('Selecionar Esta', on_click=select_this).props('flat icon=check color=green')
+
await load_dir(start_path)
await dialog
return result['path']
# --- TELA PRINCIPAL ---
def show():
+ # Helper para pegar o bot de forma segura via State
+ def get_bot():
+ if state.watcher and state.watcher.bot:
+ return state.watcher.bot
+ return None
+
with ui.column().classes('w-full p-6'):
ui.label('Configurações').classes('text-3xl font-light text-gray-800 mb-4')
@@ -82,41 +115,36 @@ def show():
with ui.card().classes('w-full max-w-2xl mx-auto p-6'):
ui.label('🔍 Configuração do Identificador').classes('text-2xl font-bold mb-4 text-indigo-600')
tmdb_key = AppConfig.get_val('tmdb_api_key', '')
- lang = AppConfig.get_val('tmdb_language', 'pt-BR')
- confidence = AppConfig.get_val('min_confidence', '90')
ui.input('API Key do TMDb', value=tmdb_key, password=True, on_change=lambda e: AppConfig.set_val('tmdb_api_key', e.value)).classes('w-full mb-4')
with ui.grid(columns=2).classes('w-full gap-4'):
- ui.input('Idioma', value=lang, on_change=lambda e: AppConfig.set_val('tmdb_language', e.value))
- ui.number('Confiança Auto (%)', value=int(confidence), min=50, max=100, on_change=lambda e: AppConfig.set_val('min_confidence', str(int(e.value))))
- ui.notify('As alterações são salvas automaticamente.', type='info', position='bottom')
+ ui.input('Idioma', value=AppConfig.get_val('tmdb_language', 'pt-BR'), on_change=lambda e: AppConfig.set_val('tmdb_language', e.value))
+ ui.number('Confiança Auto (%)', value=int(AppConfig.get_val('min_confidence', '90')), min=50, max=100, on_change=lambda e: AppConfig.set_val('min_confidence', str(int(e.value))))
- # --- ABA 2: CATEGORIAS ---
+ # --- ABA 2: CATEGORIAS (COM EDIÇÃO) ---
with ui.tab_panel(tab_cats):
ui.label('Regras de Organização').classes('text-xl text-gray-700 mb-2')
+
+ editing_id = {'val': None}
cats_container = ui.column().classes('w-full gap-2')
+ form_card = ui.card().classes('w-full mb-4 bg-gray-100 p-4')
def load_cats():
cats_container.clear()
cats = list(Category.select())
- if not cats: ui.label('Nenhuma categoria criada.').classes('text-gray-400')
for cat in cats:
- # Processa filtros para exibição
g_ids = cat.genre_filters.split(',') if cat.genre_filters else []
c_codes = cat.country_filters.split(',') if cat.country_filters else []
-
g_names = [TMDB_GENRES.get(gid, gid) for gid in g_ids if gid]
c_names = [COMMON_COUNTRIES.get(cc, cc) for cc in c_codes if cc]
- desc_text = f"Tipo: {cat.content_type.upper()}"
- if g_names: desc_text += f" | Gêneros: {', '.join(g_names)}"
- if c_names: desc_text += f" | Países: {', '.join(c_names)}"
- if not g_names and not c_names: desc_text += " | (Genérico/Padrão)"
+ desc = f"{cat.content_type.upper()}"
+ if g_names: desc += f" | {', '.join(g_names)}"
+ if c_names: desc += f" | {', '.join(c_names)}"
color_cls = 'bg-white border-gray-200'
- if 'Animação' in str(g_names) and 'Japão' in str(c_names): color_cls = 'bg-purple-50 border-purple-200' # Anime visual
- elif 'Animação' in str(g_names): color_cls = 'bg-yellow-50 border-yellow-200' # Desenho visual
+ if 'Animação' in str(g_names) and 'Japão' in str(c_names): color_cls = 'bg-purple-50 border-purple-200'
with cats_container, ui.card().classes(f'w-full flex-row items-center justify-between p-3 border {color_cls}'):
with ui.row().classes('items-center gap-4'):
@@ -124,67 +152,103 @@ def show():
ui.icon(icon).classes('text-gray-600')
with ui.column().classes('gap-0'):
ui.label(cat.name).classes('font-bold text-lg')
- ui.label(desc_text).classes('text-xs text-gray-500')
- ui.button(icon='delete', color='red', on_click=lambda c=cat: delete_cat(c)).props('flat dense')
+ ui.label(desc).classes('text-xs text-gray-500')
+
+ with ui.row():
+ ui.button(icon='edit', on_click=lambda c=cat: start_edit(c)).props('flat dense color=blue')
+ ui.button(icon='delete', on_click=lambda c=cat: delete_cat(c)).props('flat dense color=red')
- def add_cat():
+ def start_edit(cat):
+ editing_id['val'] = cat.id
+ name_input.value = cat.name
+ type_select.value = cat.content_type
+ genre_select.value = cat.genre_filters.split(',') if cat.genre_filters else []
+ country_select.value = cat.country_filters.split(',') if cat.country_filters else []
+
+ form_label.text = f"Editando: {cat.name}"
+ btn_save.text = "Atualizar Regra"
+ btn_save.props('color=blue icon=save')
+ btn_cancel.classes(remove='hidden')
+ form_card.classes(replace='bg-blue-50')
+
+ def cancel_edit():
+ editing_id['val'] = None
+ name_input.value = ''
+ genre_select.value = []
+ country_select.value = []
+ form_label.text = "Nova Biblioteca"
+ btn_save.text = "Adicionar Regra"
+ btn_save.props('color=green icon=add')
+ btn_cancel.classes(add='hidden')
+ form_card.classes(replace='bg-gray-100')
+
+ def save_cat():
if not name_input.value: return
- # Converte listas para string CSV
g_str = ",".join(genre_select.value) if genre_select.value else ""
c_str = ",".join(country_select.value) if country_select.value else ""
try:
- Category.create(
- name=name_input.value,
- target_path=f"/media/{name_input.value}",
- content_type=type_select.value,
- genre_filters=g_str,
- country_filters=c_str
- )
- name_input.value = ''; genre_select.value = []; country_select.value = []
- ui.notify('Categoria criada!', type='positive')
+ if editing_id['val']:
+ Category.update(
+ name=name_input.value,
+ content_type=type_select.value,
+ genre_filters=g_str,
+ country_filters=c_str
+ ).where(Category.id == editing_id['val']).execute()
+ ui.notify('Categoria atualizada!', type='positive')
+ else:
+ Category.create(
+ name=name_input.value,
+ target_path=f"/media/{name_input.value}",
+ content_type=type_select.value,
+ genre_filters=g_str,
+ country_filters=c_str
+ )
+ ui.notify('Categoria criada!', type='positive')
+ cancel_edit()
load_cats()
except Exception as e: ui.notify(f'Erro: {e}', type='negative')
- def delete_cat(cat):
+ def delete_cat(cat):
cat.delete_instance()
+ if editing_id['val'] == cat.id: cancel_edit()
load_cats()
- # Formulário de Criação Inteligente
- with ui.card().classes('w-full mb-4 bg-gray-100 p-4'):
- ui.label('Nova Biblioteca').classes('font-bold text-gray-700')
- with ui.grid(columns=4).classes('w-full gap-2'):
- name_input = ui.input('Nome (ex: Animes)')
-
- type_select = ui.select({
- 'mixed': 'Misto (Filmes e Séries)',
- 'movie': 'Apenas Filmes',
- 'series': 'Apenas Séries'
- }, value='mixed', label='Tipo de Conteúdo')
+ with form_card:
+ with ui.row().classes('w-full justify-between items-center'):
+ form_label = ui.label('Nova Biblioteca').classes('font-bold text-gray-700')
+ btn_cancel = ui.button('Cancelar', on_click=cancel_edit).props('flat dense color=red').classes('hidden')
- # Novos Seletores Múltiplos
- genre_select = ui.select(TMDB_GENRES, multiple=True, label='Filtrar Gêneros (Opcional)').props('use-chips')
- country_select = ui.select(COMMON_COUNTRIES, multiple=True, label='Filtrar País (Opcional)').props('use-chips')
+ with ui.grid(columns=4).classes('w-full gap-2'):
+ name_input = ui.input('Nome')
+ type_select = ui.select({'mixed': 'Misto', 'movie': 'Só Filmes', 'series': 'Só Séries'}, value='mixed', label='Tipo')
+ genre_select = ui.select(TMDB_GENRES, multiple=True, label='Gêneros').props('use-chips')
+ country_select = ui.select(COMMON_COUNTRIES, multiple=True, label='Países').props('use-chips')
- ui.button('Adicionar Regra', on_click=add_cat).props('icon=add color=green').classes('mt-2 w-full')
+ btn_save = ui.button('Adicionar Regra', on_click=save_cat).props('icon=add color=green').classes('mt-2 w-full')
load_cats()
- # --- ABA 3: DEPLOY ---
+ # --- ABA 3: DEPLOY (DESTINOS VS ORIGEM) ---
with ui.tab_panel(tab_deploy):
- async def open_picker(input_element):
- start = input_element.value if input_element.value else '/'
- selected = await pick_folder_dialog(start)
- if selected: input_element.value = selected
-
+
+ # ORIGEM (Permite Raiz /)
with ui.card().classes('w-full mb-6 border-l-4 border-amber-500 bg-amber-50'):
ui.label('📡 Origem dos Arquivos').classes('text-lg font-bold mb-2 text-amber-900')
monitor_path = AppConfig.get_val('monitor_path', '/downloads')
+
+ async def pick_source():
+ # AQUI: allowed_root='/' permite navegar em tudo (incluindo /downloads)
+ p = await pick_folder_dialog(mon_input.value, allowed_root='/')
+ if p: mon_input.value = p
+
with ui.row().classes('w-full items-center gap-2'):
mon_input = ui.input('Pasta Monitorada', value=monitor_path).classes('flex-grow font-mono bg-white rounded px-2')
- ui.button(icon='folder', on_click=lambda i=mon_input: open_picker(i)).props('flat dense color=amber-9')
+ ui.button(icon='folder', on_click=pick_source).props('flat dense color=amber-9')
+
ui.button('Salvar Origem', on_click=lambda: (AppConfig.set_val('monitor_path', mon_input.value), ui.notify('Salvo!'))).classes('mt-2 bg-amber-600 text-white')
+ # DESTINOS (Restrito a /media para organização)
ui.label('📂 Mapeamento de Destinos (/media)').classes('text-xl text-gray-700 mt-4')
paths_container = ui.column().classes('w-full gap-2')
@@ -194,52 +258,42 @@ def show():
with paths_container, ui.card().classes('w-full p-4 flex-row items-center gap-4'):
ui.label(cat.name).classes('font-bold w-32')
path_input = ui.input(value=cat.target_path).classes('flex-grow font-mono')
- ui.button(icon='folder', on_click=lambda i=path_input: open_picker(i)).props('flat dense color=amber-8')
+
+ async def pick_dest(i=path_input):
+ # AQUI: allowed_root='/media' trava a navegação
+ p = await pick_folder_dialog(i.value, allowed_root='/media')
+ if p: i.value = p
+
+ ui.button(icon='folder', on_click=pick_dest).props('flat dense color=amber-8')
ui.button(icon='save', on_click=lambda c=cat, p=path_input: (setattr(c, 'target_path', p.value), c.save(), ui.notify('Salvo!'))).props('flat round color=green')
load_deploy_paths()
- # --- ABA 4: FFMPEG (AGORA COM ISO_LANGS CORRIGIDO) ---
+ # --- ABA 4: FFMPEG ---
with ui.tab_panel(tab_ffmpeg):
ui.label('Perfis de Conversão').classes('text-xl text-gray-700')
-
profiles_query = list(FFmpegProfile.select())
profiles_dict = {p.id: p.name for p in profiles_query}
-
active_profile = next((p for p in profiles_query if p.is_active), None)
- initial_val = active_profile.id if active_profile else None
def set_active_profile(e):
if not e.value: return
FFmpegProfile.update(is_active=False).execute()
FFmpegProfile.update(is_active=True).where(FFmpegProfile.id == int(e.value)).execute()
- ui.notify(f'Perfil "{profiles_dict[e.value]}" ativado!', type='positive')
+ ui.notify(f'Perfil Ativado!', type='positive')
- select_profile = ui.select(profiles_dict, value=initial_val, label='Perfil Ativo', on_change=set_active_profile).classes('w-64 mb-6')
-
- ui.separator()
+ select_profile = ui.select(profiles_dict, value=active_profile.id if active_profile else None, label='Perfil Ativo', on_change=set_active_profile).classes('w-64 mb-6')
for p in profiles_query:
with ui.expansion(f"{p.name}", icon='tune').classes('w-full bg-white mb-2 border rounded'):
with ui.column().classes('p-4 w-full'):
with ui.grid(columns=2).classes('w-full gap-4'):
c_name = ui.input('Nome', value=p.name)
- c_codec = ui.select({
- 'h264_vaapi': 'Intel VAAPI (Linux/Docker)',
- 'h264_qsv': 'Intel QuickSync',
- 'h264_nvenc': 'Nvidia NVENC',
- 'libx264': 'CPU',
- 'copy': 'Copy'
- }, value=p.video_codec, label='Codec')
+ c_codec = ui.select({'h264_vaapi':'VAAPI','libx264':'CPU','copy':'Copy'}, value=p.video_codec, label='Codec')
c_preset = ui.select(['fast', 'medium', 'slow'], value=p.preset, label='Preset')
- c_crf = ui.number('CRF/QP', value=p.crf, min=0, max=51)
-
- audios = p.audio_langs.split(',') if p.audio_langs else []
- subs = p.subtitle_langs.split(',') if p.subtitle_langs else []
-
- # AQUI ESTAVA O ERRO (ISO_LANGS faltando)
- c_audio = ui.select(ISO_LANGS, value=audios, multiple=True, label='Áudios').props('use-chips')
- c_sub = ui.select(ISO_LANGS, value=subs, multiple=True, label='Legendas').props('use-chips')
+ c_crf = ui.number('CRF', value=p.crf)
+ c_audio = ui.select(ISO_LANGS, value=p.audio_langs.split(','), multiple=True, label='Áudios').props('use-chips')
+ c_sub = ui.select(ISO_LANGS, value=p.subtitle_langs.split(','), multiple=True, label='Legendas').props('use-chips')
def save_profile(prof=p, n=c_name, c=c_codec, pr=c_preset, cr=c_crf, a=c_audio, s=c_sub):
prof.name = n.value; prof.video_codec = c.value; prof.preset = pr.value
@@ -249,19 +303,38 @@ def show():
select_profile.options = profiles_dict
select_profile.update()
ui.notify('Salvo!')
-
ui.button('Salvar', on_click=save_profile).classes('mt-2')
# --- ABA 5: TELEGRAM ---
with ui.tab_panel(tab_telegram):
with ui.card().classes('w-full max-w-lg mx-auto p-6'):
ui.label('🤖 Integração Telegram').classes('text-2xl font-bold mb-4 text-blue-600')
- t_token = AppConfig.get_val('telegram_token', '')
- t_chat = AppConfig.get_val('telegram_chat_id', '')
- token_input = ui.input('Bot Token', value=t_token).props('password').classes('w-full mb-2')
- chat_input = ui.input('Chat ID', value=t_chat).classes('w-full mb-6')
- def save_telegram():
+
+ token_input = ui.input('Bot Token', value=AppConfig.get_val('telegram_token', '')).props('password').classes('w-full mb-2')
+ chat_input = ui.input('Chat ID', value=AppConfig.get_val('telegram_chat_id', '')).classes('w-full mb-6')
+
+ async def save_and_restart():
AppConfig.set_val('telegram_token', token_input.value)
AppConfig.set_val('telegram_chat_id', chat_input.value)
- ui.notify('Salvo!', type='positive')
- ui.button('Salvar', on_click=save_telegram).props('icon=save color=blue').classes('w-full')
\ No newline at end of file
+ ui.notify('Salvando...', type='warning')
+
+ bot = get_bot()
+ if bot:
+ await bot.restart()
+ ui.notify('Bot Reiniciado!', type='positive')
+ else:
+ ui.notify('Salvo, mas Bot não está rodando.', type='warning')
+
+ async def test_connection():
+ bot = get_bot()
+ if bot:
+ ui.notify('Enviando mensagem...')
+ success = await bot.send_test_msg()
+ if success: ui.notify('Mensagem enviada!', type='positive')
+ else: ui.notify('Falha no envio.', type='negative')
+ else:
+ ui.notify('Bot offline.', type='negative')
+
+ with ui.row().classes('w-full gap-2'):
+ ui.button('Salvar e Reiniciar', on_click=save_and_restart).props('icon=save color=blue').classes('flex-grow')
+ ui.button('Testar', on_click=test_connection).props('icon=send color=green')
\ No newline at end of file