Variet Agent
c55203474b
- nc_files.py: WebDAV SEARCH 501 시 PROPFIND depth=99 + 로컬 필터 폴백 - subtitle_downloader.py: download_file에 기존 파일 존재 체크 추가 (이미 있으면 스킵)
263 lines
9.5 KiB
Python
263 lines
9.5 KiB
Python
r"""Nextcloud Files 검색/관리 모듈.
|
|
|
|
파일 검색, 최근 파일 조회, 공유 링크 생성 등.
|
|
업로드는 NC 앱에서 직접 수행 — 봇은 검색/관리/링크 역할.
|
|
"""
|
|
|
|
import asyncio
|
|
import logging
|
|
import os
|
|
import sys
|
|
from dataclasses import dataclass
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
|
|
import config
|
|
from tools.nextcloud_client import NextcloudClient
|
|
|
|
logger = logging.getLogger("variet.tools.nc_files")
|
|
|
|
|
|
@dataclass
|
|
class NCFile:
|
|
"""Nextcloud 파일 정보."""
|
|
name: str
|
|
path: str # 사용자 상대 경로 (예: Documents/세금/...)
|
|
size: int = 0
|
|
lastmod: str = ""
|
|
content_type: str = ""
|
|
is_dir: bool = False
|
|
|
|
@property
|
|
def size_human(self) -> str:
|
|
"""사람이 읽기 쉬운 크기 표시."""
|
|
if self.size < 1024:
|
|
return f"{self.size}B"
|
|
elif self.size < 1024 * 1024:
|
|
return f"{self.size / 1024:.1f}KB"
|
|
elif self.size < 1024 * 1024 * 1024:
|
|
return f"{self.size / (1024 * 1024):.1f}MB"
|
|
else:
|
|
return f"{self.size / (1024 * 1024 * 1024):.1f}GB"
|
|
|
|
@property
|
|
def icon(self) -> str:
|
|
if self.is_dir:
|
|
return "📁"
|
|
ct = self.content_type or ""
|
|
if "image" in ct:
|
|
return "🖼️"
|
|
elif "pdf" in ct:
|
|
return "📕"
|
|
elif "video" in ct:
|
|
return "🎬"
|
|
elif "audio" in ct:
|
|
return "🎵"
|
|
elif "zip" in ct or "tar" in ct or "rar" in ct:
|
|
return "📦"
|
|
elif "spreadsheet" in ct or "excel" in ct:
|
|
return "📊"
|
|
elif "document" in ct or "word" in ct:
|
|
return "📝"
|
|
return "📄"
|
|
|
|
|
|
class NCFilesClient:
|
|
"""Nextcloud Files 검색/관리 클라이언트."""
|
|
|
|
def __init__(self, nc_client: NextcloudClient = None):
|
|
self.nc = nc_client or NextcloudClient()
|
|
|
|
def _href_to_user_path(self, href: str) -> str:
|
|
"""WebDAV href → 사용자 상대 경로."""
|
|
# href: /remote.php/dav/files/username/Documents/...
|
|
marker = f"/files/{self.nc.username}/"
|
|
idx = href.find(marker)
|
|
if idx >= 0:
|
|
return href[idx + len(marker):].rstrip("/")
|
|
return href.rstrip("/").split("/")[-1]
|
|
|
|
def _to_ncfile(self, item: dict) -> NCFile:
|
|
"""propfind 결과 dict → NCFile."""
|
|
return NCFile(
|
|
name=item.get("name", ""),
|
|
path=self._href_to_user_path(item.get("href", "")),
|
|
size=item.get("size", 0),
|
|
lastmod=item.get("lastmod", ""),
|
|
content_type=item.get("content_type", ""),
|
|
is_dir=item.get("is_dir", False),
|
|
)
|
|
|
|
# ──────────────────────────────────────
|
|
# 검색
|
|
# ──────────────────────────────────────
|
|
|
|
async def search(self, query: str) -> list[NCFile]:
|
|
"""파일 검색 (이름 기준).
|
|
|
|
Args:
|
|
query: 검색어 (부분 일치)
|
|
|
|
Returns:
|
|
매칭된 NCFile 리스트
|
|
"""
|
|
# 1차: WebDAV SEARCH 시도
|
|
results = await self.nc.webdav_search(query)
|
|
|
|
# 2차: SEARCH 실패 시 (501 등) PROPFIND로 전체 목록 → 로컬 필터
|
|
if not results:
|
|
logger.info(f"WebDAV SEARCH 미지원, PROPFIND 폴백 사용: '{query}'")
|
|
try:
|
|
dav_path = f"files/{self.nc.username}/"
|
|
all_files = await self.nc.webdav_propfind(dav_path, depth=99)
|
|
query_lower = query.lower().lstrip(".")
|
|
results = [
|
|
f for f in all_files[1:] # 첫 번째는 루트
|
|
if query_lower in f.get("name", "").lower()
|
|
or query_lower in f.get("content_type", "").lower()
|
|
]
|
|
except Exception as e:
|
|
logger.warning(f"PROPFIND 폴백 실패: {e}")
|
|
|
|
files = [self._to_ncfile(r) for r in results]
|
|
# 디렉토리보다 파일 우선, 최신순
|
|
files.sort(key=lambda f: (f.is_dir, f.name))
|
|
logger.info(f"파일 검색 '{query}': {len(files)}건")
|
|
return files
|
|
|
|
# ──────────────────────────────────────
|
|
# 목록
|
|
# ──────────────────────────────────────
|
|
|
|
async def list_dir(self, path: str = "") -> list[NCFile]:
|
|
"""디렉토리 목록 조회.
|
|
|
|
Args:
|
|
path: 사용자 상대 경로 (빈 문자열 = 루트)
|
|
"""
|
|
dav_path = f"files/{self.nc.username}/{path.lstrip('/')}"
|
|
results = await self.nc.webdav_propfind(dav_path)
|
|
# 첫 번째는 자기 자신 → 제외
|
|
files = [self._to_ncfile(r) for r in results[1:]]
|
|
files.sort(key=lambda f: (not f.is_dir, f.name.lower()))
|
|
return files
|
|
|
|
async def list_recent(self, limit: int = 10) -> list[NCFile]:
|
|
"""최근 수정된 파일 목록.
|
|
|
|
PROPFIND depth=infinity는 부하가 크므로
|
|
OCS activity API나 search를 활용.
|
|
"""
|
|
# 대안: WebDAV SEARCH로 최근 수정 파일 검색
|
|
# 현재는 루트 1단계만 조회하여 lastmod 정렬
|
|
dav_path = f"files/{self.nc.username}"
|
|
# depth=infinity는 서버 부하 → 대신 OCS favorite 또는 search 활용
|
|
# 간단하게: search로 '*' 전체 검색 후 정렬
|
|
results = await self.nc.webdav_search("*")
|
|
files = [self._to_ncfile(r) for r in results if not r.get("is_dir")]
|
|
# lastmod 기준 정렬 (최신 먼저)
|
|
files.sort(key=lambda f: f.lastmod, reverse=True)
|
|
return files[:limit]
|
|
|
|
# ──────────────────────────────────────
|
|
# 공유 링크
|
|
# ──────────────────────────────────────
|
|
|
|
async def create_link(self, path: str, expire_days: int = 7) -> Optional[str]:
|
|
"""파일/폴더의 공유 링크 생성.
|
|
|
|
Args:
|
|
path: 사용자 상대 경로
|
|
expire_days: 만료 일수 (0 = 만료 없음)
|
|
|
|
Returns:
|
|
공유 URL 또는 None
|
|
"""
|
|
# OCS Share API는 / 시작 경로 필요
|
|
share_path = f"/{path.lstrip('/')}"
|
|
return await self.nc.create_share_link(share_path, expire_days)
|
|
|
|
# ──────────────────────────────────────
|
|
# 용량 분석
|
|
# ──────────────────────────────────────
|
|
|
|
async def get_quota(self) -> dict:
|
|
"""사용자 스토리지 용량 정보.
|
|
|
|
Returns:
|
|
{used, total, free, percent} (바이트 기준)
|
|
"""
|
|
try:
|
|
dav_path = f"files/{self.nc.username}"
|
|
results = await self.nc.webdav_propfind(
|
|
dav_path,
|
|
props=["d:quota-used-bytes", "d:quota-available-bytes"],
|
|
depth=0,
|
|
)
|
|
if results:
|
|
prop = results[0]
|
|
# propfind 커스텀 필드는 직접 파싱 필요
|
|
# 기본 구현에서는 OCS 활용
|
|
return {}
|
|
except Exception as e:
|
|
logger.warning(f"용량 조회 실패: {e}")
|
|
return {}
|
|
|
|
|
|
# ──────────────────────────────────────
|
|
# CLI
|
|
# ──────────────────────────────────────
|
|
|
|
async def _cli():
|
|
import io
|
|
if sys.stdout.encoding != "utf-8":
|
|
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8", errors="replace")
|
|
|
|
args = sys.argv[1:]
|
|
client = NCFilesClient()
|
|
|
|
if not args:
|
|
print("사용법:")
|
|
print(" nc_files.py search <검색어>")
|
|
print(" nc_files.py ls [경로]")
|
|
print(" nc_files.py recent [갯수]")
|
|
print(" nc_files.py link <경로>")
|
|
return
|
|
|
|
if args[0] == "search" and len(args) > 1:
|
|
query = " ".join(args[1:])
|
|
files = await client.search(query)
|
|
print(f"🔍 '{query}' 검색 결과: {len(files)}건\n")
|
|
for i, f in enumerate(files, 1):
|
|
print(f" {i}. {f.icon} {f.name} ({f.size_human})")
|
|
print(f" 📂 {f.path}")
|
|
|
|
elif args[0] == "ls":
|
|
path = args[1] if len(args) > 1 else ""
|
|
files = await client.list_dir(path)
|
|
print(f"📂 /{path or '(루트)'} — {len(files)}건\n")
|
|
for f in files:
|
|
size = f" ({f.size_human})" if not f.is_dir else ""
|
|
print(f" {f.icon} {f.name}{size}")
|
|
|
|
elif args[0] == "recent":
|
|
limit = int(args[1]) if len(args) > 1 else 10
|
|
files = await client.list_recent(limit)
|
|
print(f"🕐 최근 파일 {len(files)}건:\n")
|
|
for i, f in enumerate(files, 1):
|
|
print(f" {i}. {f.icon} {f.name} ({f.size_human})")
|
|
print(f" 📂 {f.path} | 📅 {f.lastmod}")
|
|
|
|
elif args[0] == "link" and len(args) > 1:
|
|
path = args[1]
|
|
url = await client.create_link(path)
|
|
if url:
|
|
print(f"🔗 {url}")
|
|
else:
|
|
print("❌ 공유 링크 생성 실패")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(_cli())
|