chooseafile.py

import pandas
from matplotlib import pyplot as plt
import os
import Reader as read

# The following code takes one day of data of user choice and analyzes it.

def chooseafile():
    # If more HMOS are recorded from, add their name below.
    HMOlist = ["25_McIntyre", "2_Himbleton", "37_Woodstock", "50_Bleinheim", "8_Bozward"]
    # If more data is recorded in the future, add the year, month number below.
    yearlist = ["2017", "2018", "2019"]
    monthlist = ["01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12"]

    hmocheck = 1

    while hmocheck == 1:
        print("HMO Names: 25_McIntyre, 2_Himbleton, 37_Woodstock, 50_Bleinheim, 8_Bozward")
        print("")
        HMO = input("Type name of the HMO you wish to analyze: ")
        if HMO in HMOlist:
            print("{} is a correct HMO name".format(HMO))
            hmocheck = 0
        else:
            print("The entered HMO name does not exist")
            hmocheck = 1

    yearcheck = 1

    while yearcheck == 1:
        year = input("Type the year you wish to analyze in Ex: 2018 format: ")
        if year in yearlist:
            yearcheck = 0
        else:
            print("The entered year name is formatted wrong or not a valid year")
            yearcheck = 1

    monthcheck = 1

    while monthcheck == 1:
        month = input("Enter the month of the data in number format, EX: 11 = November, 01 = January")
        if month in monthlist:
            monthcheck = 0
        else:
            print("The month entered is not a month number or not entered correctly")
            monthcheck = 1

    daycheck = 1

    while daycheck == 1:
        day = input("Enter the day of the data in number format")
        intday = int(day)
        if month == "01" or month == "03" or month == "05" or month == "07" or month == "08" or month == "10" or month == "12":
            if intday in range(1, 31):
                daycheck = 0
            else:
                print("Enter a day in the month chosen")
                daycheck = 1

        if month == "04" or month == "06" or month == "09" or month == "11":
            if intday in range(1, 30):
                daycheck = 0
            else:
                print("Enter a day in the month chosen")
                daycheck = 1

        if month == "04" or month == "06" or month == "09" or month == "11":
            if intday in range(1, 30):
                daycheck = 0
            else:
                print("Enter a day in the month chosen")
                daycheck = 1

        # doubtful that data will be taken in 2020 for this application / project so no need to check for leap year

        if month == "02":
            if intday in range(1, 28):
                daycheck = 0
            else:
                print("Enter a day in the month chosen")
                daycheck = 1

    if HMO == "25_McIntyre":
        stringstart = "720200236_Data_"
    elif HMO == "2_Himbleton":
        stringstart = "720200260_Data_"
    elif HMO == "37_Woodstock":
        stringstart = "720200288_Data_"
    elif HMO == "50_Bleinheim":
        stringstart = "720200295_Data_"
    elif HMO == "8_Bozward":
        stringstart = "720200262_Data_"
    else:
        print("error in code where likely the start of the string part does not have a check for the new HMO added")

    finalstring = "./RawWBData/{}/{}{} {} {}.csv".format(HMO, stringstart, year, month, day)
    # print(finalstring)

    return finalstring

def doesitexist(finalstring, filelist):
    print(type(finalstring), type(filelist[0]), finalstring in filelist)
    for file in filelist:
        # print(file, finalstring)
        if file == finalstring:
            print("Attempting to plot desired day")
            return "y"

    print("No file found")
    return "n"


def makelist():
    rootDir = './RawWBData/'
    filenamelist = []  # directories is a 3d array containing all of the days in the collected data
    for dirName, subdirList, fileList in os.walk(rootDir):
        for fname in sorted(fileList):

            # print("{}/{}".format(dirName, fname))

            filenamelist.append("{}/{}".format(dirName, fname))

    return filenamelist




def onedayplot(day, path):

    changeday = day[:8300]
    print(len(changeday))
    times = []
    for time in pandas.date_range('00:00', None, periods=8300, freq='10S'):
        times.append(str(time).split(' ')[-1])


    if len(day) >= 8300:
        plotchosen = "wrong"
        while plotchosen == "wrong":
            print("actPow | hwTSet | primT | chActive | primTSet | hWActive | hWTOutlet")
            plotchoose = input("Which variable do you want to plot? For multiple variables please separate by commas")
            plotchoose = plotchoose
            plotchosen = appender(plotchoose, changeday)

        fig = plt.figure()

        ax1 = fig.add_subplot(111)

        ax1.plot(times, plotchosen)

        plt.setp(ax1.get_xticklabels(), visible=False)
        plt.setp(ax1.get_xticklabels()[::700], visible=True)
        plt.xticks(fontsize=10, rotation=90)
        for tic in ax1.xaxis.get_major_ticks():
            if (ax1.xaxis.get_major_ticks().index(tic) % 700 == 0):
                continue
            else:
                tic.tick1On = tic.tick2On = False
                tic.label1On = tic.label2On = False

        plt.gcf().subplots_adjust(bottom=0.23)

        plt.xlabel("Time of Day", labelpad=10)


        plt.title(plotchoose + " " + path[39:49])
        plt.show()

    else:
        print("This day exists, but does not have enough data points to be considered for plotting")

            #put more plots of interest here!

    return


def chooserun():
    allfiles = makelist()

    requestedpath = chooseafile()
    answer = doesitexist(requestedpath, allfiles)
    if answer == "y":
        onedayplot(read.openFile(requestedpath), requestedpath)
    else:
        print("file does not exist!")

    return

def appender(plotchoose, changeday):

    actPow = []
    hwTSet = []
    primT = []
    chActive = []
    primTSet = []
    hWActive = []
    hWTOutlet = []

    for item in changeday:
        if plotchoose == "actPow":
            actPow.append((item[1]))
        elif plotchoose == "hwTSet":
            hwTSet.append((item[2]))
        elif plotchoose == "primT":
            primT.append((item[3]))
        elif plotchoose == "chActive":
            chActive.append((item[4]))
        elif plotchoose == "primTSet":
            primTSet.append((item[5]))
        elif plotchoose == "hWActive":
            hWActive.append((item[6]))
        elif plotchoose == "hWTOutlet":
            hWTOutlet.append((item[7]))
        else:
            print("that is not a variable!")
            plotchosen = "wrong"
    if plotchoose == "actPow":
        plotchosen = actPow
    elif plotchoose == "hwTSet":
        plotchosen = hwTSet
    elif plotchoose == "primT":
        plotchosen = primT
    elif plotchoose == "chActive":
        plotchosen = chActive
    elif plotchoose == "primTSet":
        plotchosen = primTSet
    elif plotchoose == "hWActive":
        plotchosen = hWActive
    elif plotchoose == "hWTOutlet":
        plotchosen = hWTOutlet

    return plotchosen


if __name__ == '__main__':
    chooserun()