147 lines
5.4 KiB
Python
147 lines
5.4 KiB
Python
"""
|
|
Data formatting utilities for TimeTrack application.
|
|
Handles conversion of time entries and analytics data to various display formats.
|
|
"""
|
|
|
|
from datetime import datetime
|
|
from collections import defaultdict
|
|
|
|
|
|
def format_duration(seconds):
|
|
"""Format duration in seconds to HH:MM:SS format."""
|
|
if seconds is None:
|
|
return '00:00:00'
|
|
hours = seconds // 3600
|
|
minutes = (seconds % 3600) // 60
|
|
seconds = seconds % 60
|
|
return f"{hours:d}:{minutes:02d}:{seconds:02d}"
|
|
|
|
|
|
def prepare_export_data(entries):
|
|
"""Prepare time entries data for export."""
|
|
data = []
|
|
for entry in entries:
|
|
row = {
|
|
'Date': entry.arrival_time.strftime('%Y-%m-%d'),
|
|
'Project Code': entry.project.code if entry.project else '',
|
|
'Project Name': entry.project.name if entry.project else '',
|
|
'Arrival Time': entry.arrival_time.strftime('%H:%M:%S'),
|
|
'Departure Time': entry.departure_time.strftime('%H:%M:%S') if entry.departure_time else 'Active',
|
|
'Work Duration (HH:MM:SS)': format_duration(entry.duration) if entry.duration is not None else 'In progress',
|
|
'Break Duration (HH:MM:SS)': format_duration(entry.total_break_duration),
|
|
'Work Duration (seconds)': entry.duration if entry.duration is not None else 0,
|
|
'Break Duration (seconds)': entry.total_break_duration if entry.total_break_duration is not None else 0,
|
|
'Notes': entry.notes if entry.notes else ''
|
|
}
|
|
data.append(row)
|
|
return data
|
|
|
|
|
|
def prepare_team_hours_export_data(team, team_data, date_range):
|
|
"""Prepare team hours data for export."""
|
|
export_data = []
|
|
|
|
for member_data in team_data:
|
|
user = member_data['user']
|
|
daily_hours = member_data['daily_hours']
|
|
|
|
# Create base row with member info
|
|
row = {
|
|
'Team': team['name'],
|
|
'Member': user['username'],
|
|
'Email': user['email'],
|
|
'Total Hours': member_data['total_hours']
|
|
}
|
|
|
|
# Add daily hours columns
|
|
for date_str in date_range:
|
|
formatted_date = datetime.strptime(date_str, '%Y-%m-%d').strftime('%m/%d/%Y')
|
|
row[formatted_date] = daily_hours.get(date_str, 0.0)
|
|
|
|
export_data.append(row)
|
|
|
|
return export_data
|
|
|
|
|
|
def format_table_data(entries):
|
|
"""Format data for table view in analytics."""
|
|
formatted_entries = []
|
|
for entry in entries:
|
|
formatted_entry = {
|
|
'id': entry.id,
|
|
'date': entry.arrival_time.strftime('%Y-%m-%d'),
|
|
'arrival_time': entry.arrival_time.strftime('%H:%M:%S'),
|
|
'departure_time': entry.departure_time.strftime('%H:%M:%S') if entry.departure_time else 'Active',
|
|
'duration': format_duration(entry.duration) if entry.duration else 'In progress',
|
|
'break_duration': format_duration(entry.total_break_duration),
|
|
'project_code': entry.project.code if entry.project else None,
|
|
'project_name': entry.project.name if entry.project else 'No Project',
|
|
'notes': entry.notes or '',
|
|
'user_name': entry.user.username
|
|
}
|
|
formatted_entries.append(formatted_entry)
|
|
|
|
return {'entries': formatted_entries}
|
|
|
|
|
|
def format_graph_data(entries, granularity='daily'):
|
|
"""Format data for graph visualization in analytics."""
|
|
# Group data by date
|
|
daily_data = defaultdict(lambda: {'total_hours': 0, 'projects': defaultdict(int)})
|
|
project_totals = defaultdict(int)
|
|
|
|
for entry in entries:
|
|
if entry.departure_time and entry.duration:
|
|
date_key = entry.arrival_time.strftime('%Y-%m-%d')
|
|
hours = entry.duration / 3600 # Convert seconds to hours
|
|
|
|
daily_data[date_key]['total_hours'] += hours
|
|
project_name = entry.project.name if entry.project else 'No Project'
|
|
daily_data[date_key]['projects'][project_name] += hours
|
|
project_totals[project_name] += hours
|
|
|
|
# Format time series data
|
|
time_series = []
|
|
for date, data in sorted(daily_data.items()):
|
|
time_series.append({
|
|
'date': date,
|
|
'hours': round(data['total_hours'], 2)
|
|
})
|
|
|
|
# Format project distribution
|
|
project_distribution = [
|
|
{'project': project, 'hours': round(hours, 2)}
|
|
for project, hours in project_totals.items()
|
|
]
|
|
|
|
return {
|
|
'timeSeries': time_series,
|
|
'projectDistribution': project_distribution,
|
|
'totalHours': sum(project_totals.values()),
|
|
'totalDays': len(daily_data)
|
|
}
|
|
|
|
|
|
def format_team_data(entries, granularity='daily'):
|
|
"""Format data for team view in analytics."""
|
|
# Group by user and date
|
|
user_data = defaultdict(lambda: {'daily_hours': defaultdict(float), 'total_hours': 0})
|
|
|
|
for entry in entries:
|
|
if entry.departure_time and entry.duration:
|
|
date_key = entry.arrival_time.strftime('%Y-%m-%d')
|
|
hours = entry.duration / 3600
|
|
|
|
user_data[entry.user.username]['daily_hours'][date_key] += hours
|
|
user_data[entry.user.username]['total_hours'] += hours
|
|
|
|
# Format for frontend
|
|
team_data = []
|
|
for username, data in user_data.items():
|
|
team_data.append({
|
|
'username': username,
|
|
'daily_hours': dict(data['daily_hours']),
|
|
'total_hours': round(data['total_hours'], 2)
|
|
})
|
|
|
|
return {'team_data': team_data} |