chore(backend): refactor mono cli file into package
This commit is contained in:
88
backend/cli/commands/services.py
Normal file
88
backend/cli/commands/services.py
Normal file
@@ -0,0 +1,88 @@
|
||||
"""
|
||||
Service-related CLI commands.
|
||||
"""
|
||||
|
||||
import argparse
|
||||
from datetime import date
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from backend.cli.base import NimbusFlowCLI
|
||||
|
||||
from backend.models.enums import ScheduleStatus
|
||||
|
||||
|
||||
def cmd_services_list(cli: "NimbusFlowCLI", args) -> None:
|
||||
"""List services with optional filters."""
|
||||
print("Listing services...")
|
||||
|
||||
# Apply filters
|
||||
if args.upcoming:
|
||||
services = cli.service_repo.upcoming(limit=args.limit or 50)
|
||||
elif args.date:
|
||||
try:
|
||||
target_date = date.fromisoformat(args.date)
|
||||
# Get services for specific date
|
||||
all_services = cli.service_repo.list_all()
|
||||
services = [s for s in all_services if s.ServiceDate == target_date]
|
||||
except ValueError:
|
||||
print(f"❌ Invalid date format '{args.date}'. Use YYYY-MM-DD")
|
||||
return
|
||||
else:
|
||||
services = cli.service_repo.list_all()
|
||||
if args.limit:
|
||||
services = services[:args.limit]
|
||||
|
||||
if not services:
|
||||
print("No services found.")
|
||||
return
|
||||
|
||||
# Get service type names for display
|
||||
service_type_map = {st.ServiceTypeId: st.TypeName for st in cli.service_type_repo.list_all()}
|
||||
|
||||
# Get schedule counts for each service
|
||||
schedule_counts = {}
|
||||
for service in services:
|
||||
schedules = cli.schedule_repo.list_all()
|
||||
service_schedules = [s for s in schedules if s.ServiceId == service.ServiceId]
|
||||
|
||||
pending_count = len([s for s in service_schedules if s.Status == ScheduleStatus.PENDING.value])
|
||||
accepted_count = len([s for s in service_schedules if s.Status == ScheduleStatus.ACCEPTED.value])
|
||||
declined_count = len([s for s in service_schedules if s.Status == ScheduleStatus.DECLINED.value])
|
||||
|
||||
schedule_counts[service.ServiceId] = {
|
||||
'pending': pending_count,
|
||||
'accepted': accepted_count,
|
||||
'declined': declined_count,
|
||||
'total': len(service_schedules)
|
||||
}
|
||||
|
||||
# Print header
|
||||
print(f"\n{'ID':<3} | {'Date':<12} | {'Service Type':<12} | {'Total':<5} | {'Pending':<7} | {'Accepted':<8} | {'Declined'}")
|
||||
print("-" * 85)
|
||||
|
||||
# Print services
|
||||
for service in sorted(services, key=lambda s: s.ServiceDate):
|
||||
type_name = service_type_map.get(service.ServiceTypeId, "Unknown")
|
||||
counts = schedule_counts.get(service.ServiceId, {'total': 0, 'pending': 0, 'accepted': 0, 'declined': 0})
|
||||
|
||||
# Format date properly
|
||||
date_str = str(service.ServiceDate) if service.ServiceDate else "N/A"
|
||||
|
||||
print(f"{service.ServiceId:<3} | {date_str:<12} | {type_name:<12} | "
|
||||
f"{counts['total']:<5} | {counts['pending']:<7} | {counts['accepted']:<8} | {counts['declined']}")
|
||||
|
||||
print(f"\nTotal: {len(services)} services")
|
||||
|
||||
|
||||
def setup_services_parser(subparsers) -> None:
|
||||
"""Set up service-related command parsers."""
|
||||
# Services commands
|
||||
services_parser = subparsers.add_parser("services", help="Manage services")
|
||||
services_subparsers = services_parser.add_subparsers(dest="services_action", help="Services actions")
|
||||
|
||||
# services list
|
||||
services_list_parser = services_subparsers.add_parser("list", help="List services")
|
||||
services_list_parser.add_argument("--date", type=str, help="Filter by specific date (YYYY-MM-DD)")
|
||||
services_list_parser.add_argument("--upcoming", action="store_true", help="Show only upcoming services")
|
||||
services_list_parser.add_argument("--limit", type=int, help="Limit number of results")
|
||||
Reference in New Issue
Block a user