ALwrity Version 0.5.0 (Fastapi + React )
This commit is contained in:
ajaysi
184
backend/api/content_planning/services/calendar_service.py
Normal file
184
backend/api/content_planning/services/calendar_service.py
Normal file
@@ -0,0 +1,184 @@
|
||||
"""
|
||||
Calendar Service for Content Planning API
|
||||
Extracted business logic from the calendar events route for better separation of concerns.
|
||||
"""
|
||||
|
||||
from typing import Dict, Any, List, Optional
|
||||
from datetime import datetime
|
||||
from loguru import logger
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
# Import database service
|
||||
from services.content_planning_db import ContentPlanningDBService
|
||||
|
||||
# Import utilities
|
||||
from ..utils.error_handlers import ContentPlanningErrorHandler
|
||||
from ..utils.response_builders import ResponseBuilder
|
||||
from ..utils.constants import ERROR_MESSAGES, SUCCESS_MESSAGES
|
||||
|
||||
class CalendarService:
|
||||
"""Service class for calendar event operations."""
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
async def create_calendar_event(self, event_data: Dict[str, Any], db: Session) -> Dict[str, Any]:
|
||||
"""Create a new calendar event."""
|
||||
try:
|
||||
logger.info(f"Creating calendar event: {event_data.get('title', 'Unknown')}")
|
||||
|
||||
db_service = ContentPlanningDBService(db)
|
||||
created_event = await db_service.create_calendar_event(event_data)
|
||||
|
||||
if created_event:
|
||||
logger.info(f"Calendar event created successfully: {created_event.id}")
|
||||
return created_event.to_dict()
|
||||
else:
|
||||
raise Exception("Failed to create calendar event")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error creating calendar event: {str(e)}")
|
||||
raise ContentPlanningErrorHandler.handle_general_error(e, "create_calendar_event")
|
||||
|
||||
async def get_calendar_events(self, strategy_id: Optional[int] = None, db: Session = None) -> List[Dict[str, Any]]:
|
||||
"""Get calendar events, optionally filtered by strategy."""
|
||||
try:
|
||||
logger.info("Fetching calendar events")
|
||||
|
||||
db_service = ContentPlanningDBService(db)
|
||||
|
||||
if strategy_id:
|
||||
events = await db_service.get_strategy_calendar_events(strategy_id)
|
||||
else:
|
||||
# TODO: Implement get_all_calendar_events method
|
||||
events = []
|
||||
|
||||
return [event.to_dict() for event in events]
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error getting calendar events: {str(e)}")
|
||||
raise ContentPlanningErrorHandler.handle_general_error(e, "get_calendar_events")
|
||||
|
||||
async def get_calendar_event_by_id(self, event_id: int, db: Session) -> Dict[str, Any]:
|
||||
"""Get a specific calendar event by ID."""
|
||||
try:
|
||||
logger.info(f"Fetching calendar event: {event_id}")
|
||||
|
||||
db_service = ContentPlanningDBService(db)
|
||||
event = await db_service.get_calendar_event(event_id)
|
||||
|
||||
if event:
|
||||
return event.to_dict()
|
||||
else:
|
||||
raise ContentPlanningErrorHandler.handle_not_found_error("Calendar event", event_id)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error getting calendar event: {str(e)}")
|
||||
raise ContentPlanningErrorHandler.handle_general_error(e, "get_calendar_event_by_id")
|
||||
|
||||
async def update_calendar_event(self, event_id: int, update_data: Dict[str, Any], db: Session) -> Dict[str, Any]:
|
||||
"""Update a calendar event."""
|
||||
try:
|
||||
logger.info(f"Updating calendar event: {event_id}")
|
||||
|
||||
db_service = ContentPlanningDBService(db)
|
||||
updated_event = await db_service.update_calendar_event(event_id, update_data)
|
||||
|
||||
if updated_event:
|
||||
return updated_event.to_dict()
|
||||
else:
|
||||
raise ContentPlanningErrorHandler.handle_not_found_error("Calendar event", event_id)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error updating calendar event: {str(e)}")
|
||||
raise ContentPlanningErrorHandler.handle_general_error(e, "update_calendar_event")
|
||||
|
||||
async def delete_calendar_event(self, event_id: int, db: Session) -> bool:
|
||||
"""Delete a calendar event."""
|
||||
try:
|
||||
logger.info(f"Deleting calendar event: {event_id}")
|
||||
|
||||
db_service = ContentPlanningDBService(db)
|
||||
deleted = await db_service.delete_calendar_event(event_id)
|
||||
|
||||
if deleted:
|
||||
return True
|
||||
else:
|
||||
raise ContentPlanningErrorHandler.handle_not_found_error("Calendar event", event_id)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error deleting calendar event: {str(e)}")
|
||||
raise ContentPlanningErrorHandler.handle_general_error(e, "delete_calendar_event")
|
||||
|
||||
async def get_events_by_status(self, strategy_id: int, status: str, db: Session) -> List[Dict[str, Any]]:
|
||||
"""Get calendar events by status for a specific strategy."""
|
||||
try:
|
||||
logger.info(f"Fetching events for strategy {strategy_id} with status {status}")
|
||||
|
||||
db_service = ContentPlanningDBService(db)
|
||||
events = await db_service.get_events_by_status(strategy_id, status)
|
||||
|
||||
return [event.to_dict() for event in events]
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error getting events by status: {str(e)}")
|
||||
raise ContentPlanningErrorHandler.handle_general_error(e, "get_events_by_status")
|
||||
|
||||
async def get_strategy_events(self, strategy_id: int, db: Session) -> Dict[str, Any]:
|
||||
"""Get calendar events for a specific strategy."""
|
||||
try:
|
||||
logger.info(f"Fetching events for strategy: {strategy_id}")
|
||||
|
||||
db_service = ContentPlanningDBService(db)
|
||||
events = await db_service.get_strategy_calendar_events(strategy_id)
|
||||
|
||||
return {
|
||||
'strategy_id': strategy_id,
|
||||
'events_count': len(events),
|
||||
'events': [event.to_dict() for event in events]
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error getting strategy events: {str(e)}")
|
||||
raise ContentPlanningErrorHandler.handle_general_error(e, "get_strategy_events")
|
||||
|
||||
async def schedule_event(self, event_data: Dict[str, Any], db: Session) -> Dict[str, Any]:
|
||||
"""Schedule a calendar event with conflict checking."""
|
||||
try:
|
||||
logger.info(f"Scheduling calendar event: {event_data.get('title', 'Unknown')}")
|
||||
|
||||
# Check for scheduling conflicts
|
||||
conflicts = await self._check_scheduling_conflicts(event_data, db)
|
||||
|
||||
if conflicts:
|
||||
logger.warning(f"Scheduling conflicts found: {conflicts}")
|
||||
return {
|
||||
"status": "conflict",
|
||||
"message": "Scheduling conflicts detected",
|
||||
"conflicts": conflicts,
|
||||
"event_data": event_data
|
||||
}
|
||||
|
||||
# Create the event
|
||||
created_event = await self.create_calendar_event(event_data, db)
|
||||
|
||||
return {
|
||||
"status": "success",
|
||||
"message": "Calendar event scheduled successfully",
|
||||
"event": created_event
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error scheduling calendar event: {str(e)}")
|
||||
raise ContentPlanningErrorHandler.handle_general_error(e, "schedule_event")
|
||||
|
||||
async def _check_scheduling_conflicts(self, event_data: Dict[str, Any], db: Session) -> List[Dict[str, Any]]:
|
||||
"""Check for scheduling conflicts with existing events."""
|
||||
try:
|
||||
# This is a placeholder for conflict checking logic
|
||||
# In a real implementation, you would check for overlapping times, etc.
|
||||
return []
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error checking scheduling conflicts: {str(e)}")
|
||||
return []
|
||||
Reference in New Issue
Block a user