87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
|
"""
|
||
|
|
NAS Sharing Session Management
|
||
|
|
|
||
|
|
Handles session checking and credential management for Synology DSM sharing links.
|
||
|
|
Chrome profile handles cookie persistence automatically - no JSON files needed.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
import time
|
||
|
|
from typing import Optional, Tuple
|
||
|
|
from selenium import webdriver
|
||
|
|
from selenium.webdriver.common.by import By
|
||
|
|
|
||
|
|
|
||
|
|
def get_username_password() -> Tuple[str, str]:
|
||
|
|
"""
|
||
|
|
Get credentials from environment variables.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
Tuple of (username, password)
|
||
|
|
|
||
|
|
Raises:
|
||
|
|
ValueError: If credentials not found in environment
|
||
|
|
"""
|
||
|
|
username = os.getenv("NAS_USERNAME")
|
||
|
|
password = os.getenv("NAS_PASSWORD")
|
||
|
|
|
||
|
|
if not username or not password:
|
||
|
|
raise ValueError("NAS_USERNAME and NAS_PASSWORD must be set in .env.local")
|
||
|
|
|
||
|
|
return username, password
|
||
|
|
|
||
|
|
|
||
|
|
class SharingSessionManager:
|
||
|
|
"""
|
||
|
|
Manages DSM session check for sharing links.
|
||
|
|
Chrome profile handles cookies automatically - no JSON files needed.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, driver: webdriver.Chrome):
|
||
|
|
"""
|
||
|
|
Initialize session manager.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
driver: Selenium Chrome WebDriver instance
|
||
|
|
"""
|
||
|
|
self.driver = driver
|
||
|
|
|
||
|
|
def ensure_logged_in_page(self) -> None:
|
||
|
|
"""
|
||
|
|
Navigate to DSM URL to trigger Chrome profile cookie loading.
|
||
|
|
Chrome profile handles cookies automatically.
|
||
|
|
|
||
|
|
Raises:
|
||
|
|
RuntimeError: If driver not initialized
|
||
|
|
ValueError: If NAS_DSM_URL not set in environment
|
||
|
|
"""
|
||
|
|
if not self.driver:
|
||
|
|
raise RuntimeError("[Session] ❌ Driver not initialized")
|
||
|
|
|
||
|
|
dsm_url = os.getenv("NAS_DSM_URL")
|
||
|
|
if not dsm_url:
|
||
|
|
raise ValueError("NAS_DSM_URL must be set in .env.local")
|
||
|
|
|
||
|
|
print("[Session] 🌐 Navigating to DSM URL...")
|
||
|
|
self.driver.get(dsm_url)
|
||
|
|
time.sleep(2) # Wait for cookies to load from profile
|
||
|
|
print("[Session] ✅ Chrome profile cookies loaded automatically")
|
||
|
|
|
||
|
|
def is_logged_in(self) -> bool:
|
||
|
|
"""
|
||
|
|
Check if already logged in to DSM.
|
||
|
|
Chrome profile cookies handle authentication automatically.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
True if logged in, False otherwise
|
||
|
|
"""
|
||
|
|
if not self.driver:
|
||
|
|
return False
|
||
|
|
|
||
|
|
try:
|
||
|
|
# Look for DSM-specific elements that indicate logged-in state
|
||
|
|
elem = self.driver.find_element(By.XPATH, "//div[contains(text(), 'Synology Drive')]")
|
||
|
|
return elem is not None
|
||
|
|
except:
|
||
|
|
return False
|