193 lines
6.3 KiB
Python
193 lines
6.3 KiB
Python
import click
|
|
import math
|
|
from collections import defaultdict
|
|
from datetime import datetime, timedelta
|
|
from typing import Any, DefaultDict, Dict, List, Sequence, Tuple, Optional
|
|
|
|
import base32_crockford as base32
|
|
import tabulate
|
|
|
|
import api.projects
|
|
import api.stories
|
|
from config import Config
|
|
from util import require_login, Color
|
|
|
|
from . import COLOR_HEADER, _format_state, _get_persons
|
|
from ._stories_info import stories_info
|
|
from .cli import cli
|
|
|
|
STATES = 'unstarted', 'planned', 'started', 'finished', 'delivered', 'accepted'
|
|
|
|
|
|
Persons = Dict[int, Dict[str, Any]]
|
|
Totals = DefaultDict[int, Dict[str, int]]
|
|
|
|
COLOR_PLANNED = Color(Color.BRIGHT_BLACK)
|
|
COLOR_STARTED = Color(226)
|
|
COLOR_FINISHED = Color(Color.BRIGHT_BLUE)
|
|
COLOR_DELIVERED = Color(208)
|
|
COLOR_ACCEPTED = Color(Color.BRIGHT_GREEN)
|
|
|
|
|
|
def __ceil(value: float) -> int:
|
|
return int(math.ceil(value))
|
|
|
|
|
|
def __get_row(item: Tuple[int, Dict[str, int]], persons: Persons,
|
|
show_accepted: bool) -> Sequence:
|
|
owner_id, points = item
|
|
name = persons[owner_id]['name']
|
|
|
|
estimates = [points[state] for state in STATES]
|
|
|
|
progress = '['
|
|
progress += COLOR_PLANNED.format('P' * __ceil(points['planned']))
|
|
progress += COLOR_STARTED.format('S' * __ceil(points['started']))
|
|
progress += COLOR_FINISHED.format('F' * __ceil(points['finished']))
|
|
progress += COLOR_DELIVERED.format('D' * __ceil(points['delivered']))
|
|
if show_accepted:
|
|
progress += COLOR_ACCEPTED.format('A' * __ceil(points['accepted']))
|
|
progress += ']'
|
|
|
|
return name, (*estimates), sum(estimates), progress
|
|
|
|
|
|
def __format_story(story: Dict[str, Any], persons: Persons, totals: Totals) \
|
|
-> Tuple[str, str, str, str, str, int]:
|
|
code = base32.encode(story['id'])
|
|
|
|
is_owner = False
|
|
initials = []
|
|
owner_ids = story.get('owner_ids', [])
|
|
for owner_id in owner_ids:
|
|
value = persons[owner_id]['initials']
|
|
if value == Config['user']['initials']:
|
|
is_owner = True
|
|
value = f'\033[96m{value}\033[97m'
|
|
initials.append(value)
|
|
|
|
type_ = story['story_type']
|
|
|
|
estimate = story.get('estimate')
|
|
if estimate:
|
|
for owner_id in owner_ids:
|
|
state = story['current_state']
|
|
totals[owner_id][state] += estimate / len(owner_ids)
|
|
|
|
if is_owner:
|
|
code = f'\033[1;97m{code}'
|
|
estimate = f'\033[97m{estimate}\033[0m' if estimate else None
|
|
|
|
owners = ', '.join(initials)
|
|
|
|
state = _format_state(story['current_state'])
|
|
return code, type_, story['name'], owners, state, estimate
|
|
|
|
|
|
def __print_stories(stories_: List[Dict[str, Any]], persons: Persons,
|
|
totals: Totals, show_accepted: bool) -> None:
|
|
table = (__format_story(story, persons, totals) for story in stories_
|
|
if show_accepted or story['current_state'] != 'accepted')
|
|
headers = 'Code', 'Type', 'Story name', 'Owners', 'State', 'Pts'
|
|
print(tabulate.tabulate(table, headers=headers), end='\n\n')
|
|
|
|
|
|
def __print_totals(totals: Totals, persons: Persons, show_accepted: bool) \
|
|
-> None:
|
|
COLOR_HEADER.print('Point totals:', end='\n\n')
|
|
|
|
state_headers = [_format_state(state) for state in STATES]
|
|
headers = ('Owner', *state_headers, 'Total', 'Progress')
|
|
|
|
data = sorted((__get_row(item, persons, show_accepted)
|
|
for item in totals.items()),
|
|
key=lambda row: row[-2], reverse=True)
|
|
print(tabulate.tabulate(data, headers), end='\n\n')
|
|
|
|
|
|
def __print_burndown(token: str, iteration: Dict[str, Any],
|
|
show_accepted: bool) -> None:
|
|
COLOR_HEADER.print('Burndown:', end='\n\n')
|
|
|
|
start = datetime.strptime(iteration['start'], '%Y-%m-%dT%H:%M:%SZ')
|
|
history = api.projects.get_history_days(
|
|
token, iteration['project_id'], start - timedelta(days=1))
|
|
|
|
accepted_points = history['data'][0][1]
|
|
|
|
last_counts: List[int] = []
|
|
for date, *counts in history['data'][1:]:
|
|
if len(counts) > 0:
|
|
# Update the last_counts variable if an update is available.
|
|
last_counts = counts
|
|
elif len(last_counts) > 0:
|
|
# If no counts are available, use those of the previous day.
|
|
counts = last_counts
|
|
else:
|
|
# If there are no last_counts either, just skip the day
|
|
continue
|
|
|
|
progress = ''
|
|
|
|
if show_accepted:
|
|
progress += '\033[92m' + 'A' * round(counts[0] - accepted_points)
|
|
|
|
progress += '\033[38;5;208m' + 'D' * round(counts[1])
|
|
progress += '\033[94m' + 'F' * round(counts[2])
|
|
progress += '\033[38;5;226m' + 'S' * round(counts[3])
|
|
progress += '\033[90m' + 'P' * round(counts[5])
|
|
progress += '\033[0m'
|
|
|
|
print(f'{date}: {progress}')
|
|
|
|
print()
|
|
|
|
|
|
def _stories_current(project: str, scope: str, show_accepted: bool) -> None:
|
|
try:
|
|
project_id = int(Config['project_aliases'][project])
|
|
except KeyError:
|
|
project_id = base32.decode(project)
|
|
|
|
token = Config['user']['api_token']
|
|
|
|
iterations = api.projects.get_iterations(
|
|
token, project_id, scope=scope)
|
|
if not iterations:
|
|
print('No current iteration.')
|
|
return
|
|
iteration = iterations[0]
|
|
|
|
persons = _get_persons(token, project_id=project_id)
|
|
totals: DefaultDict[int, Dict[str, int]] = \
|
|
defaultdict(lambda: dict((state, 0) for state in STATES))
|
|
|
|
__print_stories(iteration['stories'], persons, totals, show_accepted)
|
|
__print_totals(totals, persons, show_accepted)
|
|
__print_burndown(token, iteration, show_accepted)
|
|
|
|
|
|
def _set_story_state(story: str, state: str) -> None:
|
|
token = Config['user']['api_token']
|
|
story_id = base32.decode(story)
|
|
api.stories.put_story(token, story_id, current_state=state)
|
|
|
|
|
|
@cli.command('stories')
|
|
@click.argument('project')
|
|
@click.argument('story', required=False)
|
|
@click.option('--scope', default='current')
|
|
@click.option('--show-accepted/--hide-accepted', default=True)
|
|
@click.option('--set-state', type=click.Choice([
|
|
'started', 'finished', 'delivered', 'rejected', 'accepted']))
|
|
@require_login
|
|
def stories(project: str, story: Optional[str], scope: str,
|
|
show_accepted: bool, set_state: str) -> None:
|
|
if story is not None:
|
|
if set_state is not None:
|
|
_set_story_state(story, set_state)
|
|
else:
|
|
stories_info(story)
|
|
else:
|
|
_stories_current(project, scope, show_accepted)
|