time_histogram.py

#!/usr/bin/env python3
"""
This function will get the time stamps for a users commits and create a
histogram to see what time of day they are typically working.
"""
import json
import matplotlib.pyplot as plt
from datetime import datetime
import pytz
from colors import *
from collections import Counter
from tzlocal import get_localzone


def what_times(file, username):
    """
    Args:
        file (str): file name to extract information from
    """
    commit_dates, commit_lists, commit_hours = [], [], []
    with open(file) as f:
        data = json.load(f)

    commit_data_per_repo = data.get("commit_data_per_repo", {})

    # Get all of the commit messages
    for _, repo_data in commit_data_per_repo.items():
        commit_messages = repo_data.get("commit_messages", 0)
        commit_lists.append(commit_messages)

    # Create a list of all of the timestamps
    for commit_message in commit_lists:
        for commit in commit_message:
            commit_dates.append(commit.get('date'))

    # Convert those timestamps to something usable for the plot
    local_tz = get_localzone()
    for date in commit_dates:
        commit_date = datetime.strptime(date, "%Y-%m-%dT%H:%M:%SZ")
        utc_timezone = pytz.timezone('UTC')
        central_timezone = pytz.timezone(str(local_tz))
        central_date = utc_timezone.localize(
            commit_date).astimezone(central_timezone)
        commit_hours.append(central_date.hour)

    # Finding the hour window with the most commits
    most_hours = [sorted(Counter(commit_hours).items(),
                         key=lambda x: x[1],
                         reverse=True)[i][0] for i in range(0, 1)]

    # If most hours is 0, do Midnight to 1:00AM
    if most_hours[0] == 0:
        print("You were most active between Midnight and " +
              str(most_hours[0] + 1) + ":00 AM")
    # If most hours is 1 thru 10
    if most_hours[0] >= 1 and most_hours[0] <= 10:
        print("You were most active between " + str(most_hours[0]) +
              ":00 and " + str(most_hours[0] + 1) + ":00 AM")

    # If most hours is 11, do 11:00AM to Noon
    if most_hours[0] == 11:
        print("You were most active between " + str(most_hours[0]) +
              " and Noon")

    # If most hours is 12, do Noon to 1:00PM
    if most_hours[0] == 12:
        print("You were most active between Noon and 1:00 PM")

    # If most hours is 13 thru 22
    if most_hours[0] >= 13 and most_hours[0] <= 22:
        print_time = most_hours[0] - 12
        print("You were most active between " + str(print_time) +
              ":00 and " + str(print_time + 1) + ":00 PM")

    # If most hours is 23, do 11:00PM to Midnight
    if most_hours[0] == 23:
        print("You were most active between 11:00 PM and Midnight")

    # Logic for what to return in the animation
    if most_hours[0] < 4:
        print("You are quite the Night Owl")
    elif most_hours[0] >= 4 and most_hours[0] < 9:
        print("You are an Early Bird")
    elif most_hours[0] >= 9 and most_hours[0] < 18:
        print("You are a Typical Daytimer")
    elif most_hours[0] >= 19 and most_hours[0] < 22:
        print("Looks like you're a Hobbyist")
    else:
        print("You are quite the Night Owl")

    # Create a histogram
    bins = [x for x in range(25)]
    labels = ['12AM', '1AM', '2AM', '3AM', '4AM', '5AM', '6AM', '7AM', '8AM',
              '9AM', '10AM', '11AM', '12PM', '1PM', '2PM', '3PM', '4PM', '5PM',
              '6PM', '7PM', '8PM', '9PM', '10PM', '11PM', '12AM']

    plt.figure(figsize=(8, 8), facecolor=BLACK)
    plt.hist(commit_hours, bins=bins, edgecolor=BLACK, alpha=0.7)
    plt.xlabel('Commit Hour', color=WHITE, fontweight='bold',
               fontsize='x-large')
    plt.ylabel('Number of Commits', color=WHITE, fontweight='bold',
               fontsize='x-large')
    plt.title(username + ' Commit Time Distribution (in Hours)',
              color=WHITE, fontweight='bold', fontsize='xx-large')
    plt.xticks(bins, labels, rotation=45, ha='right', color=WHITE,
               fontweight='bold')
    plt.yticks(color=WHITE, fontweight='bold')
    plt.grid(axis='y', linestyle='--', alpha=0.7)
    plt.tight_layout()
    ax = plt.gca()
    ax.set_facecolor(BLACK)
    fname = './png_files/' + username + 'histogram.png'
    plt.savefig(fname=fname, format='png')