Gantt Charts Version 2 - Individual Workstreams

# SQL output is imported as a pandas dataframe variable called "df"
# primary reference: https://sukhbinder.wordpress.com/2016/05/10/quick-gantt-chart-with-matplotlib/
# reference http://www.clowersresearch.com/main/gantt-charts-in-matplotlib/

#######################################################################
### LIBRARIES ###
#######################################################################

import pandas as pd
import datetime as dt
import matplotlib.dates as dates
import matplotlib.pyplot as plt
from matplotlib.dates import MONTHLY, WEEKLY, DateFormatter, rrulewrapper, RRuleLocator
from collections import OrderedDict
import numpy as np
import datetime as dt

#######################################################################
### HELPER FUNCTIONS ###
#######################################################################

# Function: Return distinct projects/phases
# Input: dataframe with a column indicating project names
# Output: dataframe of all distinct phases

def all_phases(df):
  phases=df['project'].drop_duplicates().values.tolist()
  return phases

# Function: prepare dataframe for gantt visualization
# Input: dataframe with a column for project, team, start_date, and end_date
# Output: dataframe with all columns required for main function

def df_gantt(df):

    df['project']=df['project'].astype('category')
    df['person']=df['person'].astype('category')
    df['end_date2']=dates.date2num(df['end_date'])
    df['start_date2']=dates.date2num(df['start_date'])
    df['duration']=df['end_date2']-df['start_date2']

    #assign colors for each series
    colors=['paleturquoise','lightskyblue','plum','mediumorchid','salmon','lightpink','sandybrown','gold']
    allprojects=df['project'].drop_duplicates().sort_values().reset_index(drop=True)
# #     allgrps=df['person'].drop_duplicates().sort_values().reset_index(drop=True)
    se=pd.DataFrame(colors[0:len(allprojects)])
    se.columns=['col']
    colormatrix=pd.concat([allprojects,se],axis=1)
    df = df.merge(colormatrix,on='project',how='inner').sort_values('person').reset_index(drop=True)
    return df

##########################################################################
### MAIN PLOT
##########################################################################

# Function: Create Gantt chart
# Input: dataframe with a column for project, team, start_date, and end_date, optional parameter showtoday (boolean) that shows a line marking the current date
# Output: matplotlib object of Gantt chart

def gantt_graph(df, showtoday):

  df=df_gantt(df)
  fig, ax = plt.subplots()

  #initialize variables
  j=''
  change=0
  ylocs=[]

  #plot chart
  for i in range(len(df['start_date2'])):
    if (j!=df['person'][i]):
      change=change+1
      j=df['person'][i]
      ylocs.append(0.5*change+0.5)
    ax.barh(0.5*change+0.5, df['duration'][i], left=df['start_date2'][i],height=0.4, align='center', color=df['col'][i], alpha = 1.0, label=df['person'][i])
    ax.annotate(df['project'][i],xy=(df['start_date2'][i]+0.5*df['duration'][i],0.5*change+0.5),horizontalalignment='center')

  #format x axis
  rule = rrulewrapper(WEEKLY, interval=1)
  ax.xaxis.set_major_locator(RRuleLocator(rule))
  ax.xaxis.set_major_formatter(DateFormatter("%b %d '%y"))
  plt.tick_params(
    axis='x',          # changes apply to the x-axis
    which='both',      # both major and minor ticks are affected
    top='off',      # ticks along the bottom edge are off
    bottom='on',
    rotation=0)         # ticks along the top edge are off

    #format y axis
  ax.invert_yaxis()
  ax.set_yticks(ylocs)
  ppl=df['person'].drop_duplicates().values.tolist()
  ax.set_yticklabels(ppl)
  plt.tick_params(
    axis='y',          # changes apply to the x-axis
    which='both',      # both major and minor ticks are affected
    left='off',      # ticks along the bottom edge are off
    right='off')        # ticks along the top edge are off


  #format border
  ax.spines['bottom'].set_linewidth(0.2)
  ax.spines['left'].set_linewidth(0.2)
  ax.spines['top'].set_linewidth(0.2)
  ax.spines['right'].set_linewidth(0.2)

  #plot a line showing current date
  if showtoday == True:
    plt.axvline(x=dt.date.today(),color='gray')

  return fig

# Function: provide error message handling for double-booked team memebers
# Input: 1. dataframe with a column for project, team, start_date, and end_date; 2. optional boolean parameter to show a line for the current date
# Output: matplotlib object of gantt chart (if no double booking cases) or error message (if any cases of double booking exist)

def workflow_errors(df, show_today=True):
  df_updated=df_gantt(df)
  errors=0
  for person in df_updated['person'].drop_duplicates().values.tolist():
    df2=df_updated[df_updated['person'] == person]
    df3=df2
    for i in range(len(df2['start_date2'])):
      for j in range(len(df3['start_date2'])):
        if (df2['end_date2'].iloc[i]>df3['start_date2'].iloc[j] and df2['end_date2'].iloc[i]<df3['end_date2'].iloc[j]):
          stringtime=df2['end_date'].iloc[i].strftime('%B %d, %Y')
          txt = person + ' has multiple workflows on/around ' + stringtime
          errors+=1
  if (errors==0):
    fig = gantt_graph(df, showtoday=show_today)
  else:
    fig = plt.figure()
    plt.axis('off')
    plt.text(.5, .5, txt, fontsize=15, color='black', ha='center')
  return fig

# # Use Sisense for Cloud Data Teams to visualize a dataframe or an image by passing data to periscope.output()
periscope.image(workflow_errors(df, show_today=True))