pivotalcli/api/projects.py

from datetime import datetime
from typing import Any, Dict, List, Optional

import requests

from . import _BASE_URL, _headers, _with_token


@_with_token
def get(token: str) -> List[Dict[str, Any]]:
    r = requests.get(f'{_BASE_URL}/projects', headers=_headers(token))
    return r.json()


@_with_token
def get_project(project_id: int, token: str) -> Dict[str, Any]:
    r = requests.get(f'{_BASE_URL}/projects/{project_id}',
                     headers=_headers(token))
    return r.json()


@_with_token
def get_stories(project_id: int, token: str) -> List[Dict[str, Any]]:
    r = requests.get(f'{_BASE_URL}/projects/{project_id}/stories',
                     headers=_headers(token))
    return r.json()


@_with_token
def get_memberships(project_id: int, token: str) -> List[Dict[str, Any]]:
    r = requests.get(f'{_BASE_URL}/projects/{project_id}/memberships',
                     headers=_headers(token))
    return r.json()


@_with_token
def get_iterations(project_id: int, scope: str, token: str) \
        -> List[Dict[str, Any]]:
    r = requests.get(f'{_BASE_URL}/projects/{project_id}/iterations',
                     headers=_headers(token), params={'scope': scope})
    return r.json()


@_with_token
def get_story_transitions(str, project_id: int, token: str,
                          after: Optional[datetime] = None,
                          before: Optional[datetime] = None) \
        -> List[Dict[str, Any]]:
    parameters = {}
    if after:
        parameters['occurred_after'] = after.isoformat()
    if before:
        parameters['occurred_before'] = before.isoformat()

    r = requests.get(f'{_BASE_URL}/projects/{project_id}/story_transitions',
                     headers=_headers(token), params=parameters)
    return r.json()


@_with_token
def get_history_days(project_id: int, start: datetime, token: str) \
        -> Dict[str, Any]:
    r = requests.get(f'{_BASE_URL}/projects/{project_id}/history/days',
                     headers=_headers(token),
                     params={'start_date': start.isoformat()})
    return r.json()