|
|
from dataclasses import dataclass |
|
|
from typing import Optional |
|
|
from datetime import datetime |
|
|
|
|
|
@dataclass |
|
|
class Source: |
|
|
"""Source model representing an RSS source.""" |
|
|
id: str |
|
|
user_id: str |
|
|
source: str |
|
|
category: Optional[str] = None |
|
|
last_update: Optional[datetime] = None |
|
|
created_at: Optional[datetime] = None |
|
|
|
|
|
@classmethod |
|
|
def from_dict(cls, data: dict): |
|
|
"""Create a Source instance from a dictionary.""" |
|
|
return cls( |
|
|
id=data['id'], |
|
|
user_id=data['user_id'], |
|
|
source=data['source'], |
|
|
category=data.get('category'), |
|
|
last_update=datetime.fromisoformat(data['last_update'].replace('Z', '+00:00')) if data.get('last_update') else None, |
|
|
created_at=datetime.fromisoformat(data['created_at'].replace('Z', '+00:00')) if data.get('created_at') else None |
|
|
) |
|
|
|
|
|
def to_dict(self): |
|
|
"""Convert Source instance to dictionary.""" |
|
|
return { |
|
|
'id': self.id, |
|
|
'user_id': self.user_id, |
|
|
'source': self.source, |
|
|
'category': self.category, |
|
|
'last_update': self.last_update.isoformat() if self.last_update else None, |
|
|
'created_at': self.created_at.isoformat() if self.created_at else None |
|
|
} |
