plan_sprint.py

import settings
from planner.export import CsvExport
from planner.visualise import print_calendar, print_issues
from planner.propose_schedule import Algorithm, propose_schedule
from services.google_calendar import GoogleCalendarEventsClient
from services.jira_client import JiraClient


def main():
    jira_client = JiraClient()
    issues = jira_client.get_issues_list()
    current_sprint = jira_client.get_current_sprint()

    google_event_client = GoogleCalendarEventsClient(
        start_date=current_sprint.startDate, end_date=current_sprint.endDate
    )
    calendar = google_event_client.get_calendar_list()

    print("Available calendar:\n")
    print_calendar(calendar)

    result = propose_schedule(
        calendar,
        issues,
        time_per_estimation_point=settings.TIME_PER_ESTIMATION_POINT,
        algorithm=Algorithm[settings.ALGORITHM],
    )

    result.print_stats()

    print("Output:\n")
    print_calendar(result.schedule)

    exporter = CsvExport("sprint_plan", result)
    exporter.export_all()


if __name__ == "__main__":
    main()