48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
"""Torrent application DTOs."""
|
|
from dataclasses import dataclass
|
|
from typing import Optional, List, Dict, Any
|
|
|
|
|
|
@dataclass
|
|
class SearchTorrentsResponse:
|
|
"""Response from searching for torrents."""
|
|
status: str
|
|
torrents: Optional[List[Dict[str, Any]]] = None
|
|
count: Optional[int] = None
|
|
error: Optional[str] = None
|
|
message: Optional[str] = None
|
|
|
|
def to_dict(self):
|
|
"""Convert to dict for agent compatibility."""
|
|
result = {"status": self.status}
|
|
|
|
if self.error:
|
|
result["error"] = self.error
|
|
result["message"] = self.message
|
|
else:
|
|
if self.torrents is not None:
|
|
result["torrents"] = self.torrents
|
|
if self.count is not None:
|
|
result["count"] = self.count
|
|
|
|
return result
|
|
|
|
|
|
@dataclass
|
|
class AddTorrentResponse:
|
|
"""Response from adding a torrent."""
|
|
status: str
|
|
message: Optional[str] = None
|
|
error: Optional[str] = None
|
|
|
|
def to_dict(self):
|
|
"""Convert to dict for agent compatibility."""
|
|
result = {"status": self.status}
|
|
|
|
if self.error:
|
|
result["error"] = self.error
|
|
if self.message:
|
|
result["message"] = self.message
|
|
|
|
return result
|