from nicegui import ui, app import os import shutil import datetime import subprocess import json ROOT_DIR = "/downloads" # --- UTILITÁRIOS --- 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" def get_subfolders(root): folders = [root] try: for r, d, f in os.walk(root): if "finalizados" in r or "temp" in r: continue for folder in d: if not folder.startswith('.'): folders.append(os.path.join(r, folder)) except: pass return sorted(folders) # --- LEITOR DE METADADOS (FFPROBE) --- def get_media_info(filepath): """Lê as faixas de áudio e legenda do arquivo""" cmd = ["ffprobe", "-v", "quiet", "-print_format", "json", "-show_streams", "-show_format", filepath] try: res = subprocess.run(cmd, capture_output=True, text=True) data = json.loads(res.stdout) info = { "duration": float(data['format'].get('duration', 0)), "bitrate": int(data['format'].get('bit_rate', 0)), "video": [], "audio": [], "subtitle": [] } for s in data.get('streams', []): type = s['codec_type'] lang = s.get('tags', {}).get('language', 'und') title = s.get('tags', {}).get('title', '') codec = s.get('codec_name', 'unknown') desc = f"[{lang.upper()}] {codec}" if title: desc += f" - {title}" if type == 'video': w = s.get('width', 0) h = s.get('height', 0) info['video'].append(f"{codec.upper()} ({w}x{h})") elif type == 'audio': ch = s.get('channels', 0) info['audio'].append(f"{desc} ({ch}ch)") elif type == 'subtitle': info['subtitle'].append(desc) return info except: return None # --- CLASSE GERENCIADORA --- class FileManager: def __init__(self): self.path = ROOT_DIR self.view_mode = 'grid' self.container = None def navigate(self, path): if os.path.exists(path) and os.path.isdir(path): self.path = path self.refresh() else: ui.notify('Caminho inválido', type='negative') def navigate_up(self): parent = os.path.dirname(self.path) if self.path != ROOT_DIR: self.navigate(parent) def toggle_view(self): self.view_mode = 'list' if self.view_mode == 'grid' else 'grid' self.refresh() def refresh(self): if self.container: self.container.clear() with self.container: self.render_header() self.render_content() # --- PLAYER DE VÍDEO --- def open_player(self, path): filename = os.path.basename(path) # Converte caminho local (/downloads/pasta/video.mkv) para URL (/files/pasta/video.mkv) # O prefixo /files foi configurado no main.py rel_path = os.path.relpath(path, ROOT_DIR) video_url = f"/files/{rel_path}" # Pega dados técnicos info = get_media_info(path) with ui.dialog() as dialog, ui.card().classes('w-full max-w-4xl h-[80vh] p-0 gap-0'): # Header with ui.row().classes('w-full bg-gray-100 p-2 justify-between items-center'): ui.label(filename).classes('font-bold text-lg truncate') ui.button(icon='close', on_click=dialog.close).props('flat round dense') with ui.row().classes('w-full h-full'): # Coluna Esquerda: Player with ui.column().classes('w-2/3 h-full bg-black justify-center'): # Player HTML5 Nativo ui.video(video_url).classes('w-full max-h-full') ui.label('Nota: Áudios AC3/DTS podem ficar mudos no navegador.').classes('text-gray-500 text-xs text-center w-full') # Coluna Direita: Informações with ui.column().classes('w-1/3 h-full p-4 overflow-y-auto bg-white border-l'): ui.label('📋 Detalhes do Arquivo').classes('text-lg font-bold mb-4 text-blue-600') if info: # Vídeo ui.label('Vídeo').classes('font-bold text-xs text-gray-500 uppercase') for v in info['video']: ui.label(f"📺 {v}").classes('ml-2 text-sm') ui.separator().classes('my-2') # Áudio ui.label('Áudio').classes('font-bold text-xs text-gray-500 uppercase') if info['audio']: for a in info['audio']: ui.label(f"🔊 {a}").classes('ml-2 text-sm') else: ui.label("Sem áudio").classes('ml-2 text-sm text-gray-400') ui.separator().classes('my-2') # Legenda ui.label('Legendas').classes('font-bold text-xs text-gray-500 uppercase') if info['subtitle']: for s in info['subtitle']: ui.label(f"💬 {s}").classes('ml-2 text-sm') else: ui.label("Sem legendas").classes('ml-2 text-sm text-gray-400') else: ui.label('Não foi possível ler os metadados.').classes('text-red') dialog.open() # --- DIÁLOGOS DE AÇÃO --- def open_delete_dialog(self, path): with ui.dialog() as dialog, ui.card(): ui.label('Excluir item?').classes('font-bold') ui.label(os.path.basename(path)) with ui.row().classes('w-full justify-end'): ui.button('Cancelar', on_click=dialog.close).props('flat') def confirm(): try: if os.path.isdir(path): shutil.rmtree(path) else: os.remove(path) dialog.close() self.refresh() ui.notify('Excluído!') except Exception as e: ui.notify(str(e), type='negative') ui.button('Excluir', on_click=confirm).props('color=red') 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') def save(): try: os.rename(path, os.path.join(os.path.dirname(path), name.value)) dialog.close() self.refresh() ui.notify('Renomeado!') except Exception as e: ui.notify(str(e), type='negative') ui.button('Salvar', on_click=save) dialog.open() def open_move_dialog(self, path): folders = get_subfolders(ROOT_DIR) if os.path.isdir(path) and path in folders: folders.remove(path) opts = {f: f.replace(ROOT_DIR, "Raiz") if f != ROOT_DIR else "Raiz" for f in folders} with ui.dialog() as dialog, ui.card().classes('w-96'): ui.label('Mover Para') target = ui.select(opts, value=ROOT_DIR, with_input=True).classes('w-full') def confirm(): try: shutil.move(path, target.value) dialog.close() self.refresh() ui.notify('Movido!') except Exception as e: ui.notify(str(e), type='negative') ui.button('Mover', on_click=confirm) dialog.open() def open_create_folder(self): with ui.dialog() as dialog, ui.card(): ui.label('Nova Pasta') name = ui.input('Nome') def create(): try: os.makedirs(os.path.join(self.path, name.value)) dialog.close() self.refresh() except Exception as e: ui.notify(str(e), type='negative') ui.button('Criar', on_click=create) dialog.open() # --- MENU DE CONTEXTO --- 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('▶️ Reproduzir / Detalhes', on_click=lambda: self.open_player(entry.path)) ui.separator() 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)) ui.separator() ui.menu_item('Excluir', on_click=lambda: self.open_delete_dialog(entry.path)).props('text-color=red') element.on('contextmenu.prevent', lambda: m.open()) # --- RENDERIZADORES --- def render_header(self): with ui.row().classes('w-full items-center bg-gray-100 p-2 rounded-lg gap-2'): if self.path != ROOT_DIR: ui.button(icon='arrow_upward', on_click=self.navigate_up).props('flat round dense').tooltip('Subir') else: ui.button(icon='home').props('flat round dense disabled text-color=grey') rel = os.path.relpath(self.path, ROOT_DIR) parts = rel.split(os.sep) if rel != '.' else [] with ui.row().classes('items-center gap-0'): ui.button('/', on_click=lambda: self.navigate(ROOT_DIR)).props('flat dense no-caps min-w-0 px-2') acc = ROOT_DIR for part in parts: acc = os.path.join(acc, part) ui.label('/') ui.button(part, on_click=lambda p=acc: self.navigate(p)).props('flat dense no-caps min-w-0 px-2') ui.space() ui.button(icon='create_new_folder', on_click=self.open_create_folder).props('flat round dense') ui.button(icon='view_list' if self.view_mode == 'grid' else 'grid_view', on_click=self.toggle_view).props('flat round dense') ui.button(icon='refresh', on_click=self.refresh).props('flat round dense') def render_content(self): try: entries = sorted(os.scandir(self.path), key=lambda e: (not e.is_dir(), e.name.lower())) except: return if not entries: ui.label('Pasta vazia').classes('w-full text-center text-gray-400 mt-10') return if 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 entries: 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 'blue-grey' if icon == 'movie': color = 'purple-6' with ui.card().classes('w-full aspect-square p-2 items-center justify-center relative group hover:shadow-md cursor-pointer select-none') as card: if is_dir: card.on('click', lambda p=entry.path: self.navigate(p)) elif icon == 'movie': # Duplo clique no vídeo abre o player card.on('dblclick', lambda p=entry.path: self.open_player(p)) self.bind_context_menu(card, entry) 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') if not is_dir: ui.label(get_human_size(entry.stat().st_size)).classes('text-[10px] text-gray-400') with ui.button(icon='more_vert').props('flat round dense size=sm').classes('absolute top-1 right-1 opacity-0 group-hover:opacity-100 transition-opacity bg-white/90'): with ui.menu(): if icon == 'movie': ui.menu_item('▶️ Play', on_click=lambda p=entry.path: self.open_player(p)) ui.separator() ui.menu_item('Renomear', on_click=lambda p=entry.path: self.open_rename_dialog(p)) ui.menu_item('Mover', on_click=lambda p=entry.path: self.open_move_dialog(p)) ui.separator() ui.menu_item('Excluir', on_click=lambda p=entry.path: self.open_delete_dialog(p)).props('text-color=red') else: with ui.column().classes('w-full gap-0'): for entry in entries: is_dir = entry.is_dir() icon = 'folder' if is_dir else 'description' color = 'amber-8' if is_dir else 'blue-grey' with ui.row().classes('w-full items-center px-2 py-2 border-b hover:bg-blue-50 cursor-pointer group') as row: if is_dir: row.on('click', lambda p=entry.path: self.navigate(p)) elif entry.name.lower().endswith(('.mkv', '.mp4')): row.on('dblclick', lambda p=entry.path: self.open_player(p)) self.bind_context_menu(row, entry) ui.icon(icon, color=color).classes('mr-2') with ui.column().classes('flex-grow gap-0'): ui.label(entry.name).classes('text-sm font-medium break-all') if not is_dir: ui.label(get_human_size(entry.stat().st_size)).classes('text-xs text-gray-500 mr-4') with ui.button(icon='more_vert').props('flat round dense size=sm').classes('sm:opacity-0 group-hover:opacity-100'): with ui.menu(): if not is_dir: ui.menu_item('▶️ Play', on_click=lambda p=entry.path: self.open_player(p)) ui.menu_item('Renomear', on_click=lambda p=entry.path: self.open_rename_dialog(p)) ui.menu_item('Mover', on_click=lambda p=entry.path: self.open_move_dialog(p)) ui.menu_item('Excluir', on_click=lambda p=entry.path: self.open_delete_dialog(p)) def create_ui(): fm = FileManager() fm.container = ui.column().classes('w-full h-full p-2 md:p-4 gap-4') fm.refresh()