tab_manager.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Менеджер вкладок браузера
"""

import time
from typing import List, Optional


class TabInfo:
    """Информация о вкладке"""
    
    def __init__(self, tab):
        self.tab = tab
        self.created_at = time.time()
        self.last_active = time.time()
        self.session_killed = False
    
    def update_activity(self):
        """Обновить время последней активности"""
        self.last_active = time.time()
    
    def get_inactive_time(self) -> float:
        """Получить время неактивности в секундах"""
        return time.time() - self.last_active
    
    def get_inactive_minutes(self) -> float:
        """Получить время неактивности в минутах"""
        return self.get_inactive_time() / 60
    
    def should_kill_session(self, inactive_threshold: int = 1800) -> bool:
        """
        Проверить, нужно ли убить сессию
        inactive_threshold: порог неактивности в секундах (по умолчанию 1800 = 30 минут)
        """
        return (
            not self.session_killed and 
            self.get_inactive_time() > inactive_threshold
        )


class TabManager:
    """Менеджер вкладок"""
    
    def __init__(self):
        self.tabs: List[TabInfo] = []
    
    def add_tab(self, tab) -> TabInfo:
        """Добавить вкладку"""
        tab_info = TabInfo(tab)
        self.tabs.append(tab_info)
        return tab_info
    
    def remove_tab(self, tab) -> bool:
        """Удалить вкладку"""
        for tab_info in self.tabs:
            if tab_info.tab == tab:
                self.tabs.remove(tab_info)
                return True
        return False
    
    def get_tab_info(self, tab) -> Optional[TabInfo]:
        """Получить информацию о вкладке"""
        for tab_info in self.tabs:
            if tab_info.tab == tab:
                return tab_info
        return None
    
    def update_tab_activity(self, tab):
        """Обновить активность вкладки"""
        tab_info = self.get_tab_info(tab)
        if tab_info:
            tab_info.update_activity()
    
    def get_inactive_tabs(self, inactive_threshold: int = 1800) -> List[TabInfo]:
        """
        Получить список неактивных вкладок
        inactive_threshold: порог неактивности в секундах (по умолчанию 1800 = 30 минут)
        """
        return [
            tab_info for tab_info in self.tabs
            if tab_info.should_kill_session(inactive_threshold)
        ]
    
    def kill_inactive_sessions(self, inactive_threshold: int = 1800) -> int:
        """
        Убить сессии неактивных вкладок
        Возвращает количество убитых сессий
        """
        inactive_tabs = self.get_inactive_tabs(inactive_threshold)
        killed_count = 0
        
        for tab_info in inactive_tabs:
            if not tab_info.session_killed:
                tab_info.tab.kill_session()
                tab_info.session_killed = True
                killed_count += 1
        
        return killed_count
    
    def get_stats(self) -> dict:
        """Получить статистику по вкладкам"""
        if not self.tabs:
            return {
                'total_tabs': 0,
                'active_tabs': 0,
                'inactive_tabs': 0,
                'killed_sessions': 0
            }
        
        inactive_count = len(self.get_inactive_tabs())
        killed_count = sum(1 for tab_info in self.tabs if tab_info.session_killed)
        
        return {
            'total_tabs': len(self.tabs),
            'active_tabs': len(self.tabs) - inactive_count,
            'inactive_tabs': inactive_count,
            'killed_sessions': killed_count
        }
    
    def get_all_tabs_info(self) -> List[dict]:
        """Получить информацию обо всех вкладках"""
        return [
            {
                'url': tab_info.tab.get_url(),
                'title': tab_info.tab.get_title(),
                'inactive_minutes': round(tab_info.get_inactive_minutes(), 2),
                'session_killed': tab_info.session_killed,
                'created_at': tab_info.created_at
            }
            for tab_info in self.tabs
        ]