Attendence-V8.py

# V8
# PESU Academy Attendence Calculator
'''
New Features:
    - Cache system for same day data
    - Removed the need for requirements.txt
Future Updates:
    - Update the Documentation
    - Something related to graphs
'''
import os
import time
import json
try:
    from selenium import webdriver
    from selenium import *
    from selenium.webdriver.chrome.options import Options
except ModuleNotFoundError:
    print('Missing Modules')
    print('Installing Selenium....')
    os.system('pip install selenium')
    print('Installing Selenium....Done')
    from selenium import webdriver
    from selenium import *
    from selenium.webdriver.chrome.options import Options

try:
    import pandas as pd
except ModuleNotFoundError:
    print('Missing Modules')
    print('Installing Pandas....')
    os.system('pip install pandas')
    print('Installing Pandas....Done')
    import pandas as pd

try:
    import PySimpleGUI as sg
except ModuleNotFoundError:
    print('Missing Modules')
    print('Installing PySimpleGUI....')
    os.system('pip install PySimpleGUI')
    print('Installing PySimpleGUI....Done')
    import PySimpleGUI as sg




def buildGUI():
    '''
    ------------------------------------------------------
    Function to build Login GUI
    ------------------------------------------------------
    '''

    # read secrets.json file
    print('Reading Files....')

    if os.path.exists('secrets.json'):
        with open('secrets.json') as f:
            data = json.load(f)
            DefaultUsername = data['Username'].strip()
            DefaultPassword= data['Password'].strip()
            DefaultAttendence = data['Attendence'].strip()
    else:
        with open('secrets.json','w') as f:
            data = {'Username':'','Password':'','Attendence':'80','Data': {'Date':time.strftime("%d/%m/%Y"),'DataFrame':''}}
            json.dump(data,f)
            DefaultUsername = ''
            DefaultPassword = ''
            DefaultAttendence = '80'
            
    
    print('Reading Files....Done')


    # sg.theme('Black')   # Add a touch of color
    sg.theme('DarkBrown2')
    # make a window which takes username and password as input
    layout = [[sg.Text('Enter the Username:',font=("Cascadia Code",12) ),sg.Input(key='-USERNAME-',background_color='white',default_text = DefaultUsername)],
                [sg.Text('Enter the Password:',font=("Cascadia Code",12)),sg.Input(key='-PASSWORD-',password_char='*',background_color='white',default_text= DefaultPassword)],
                [sg.Text('Enter the Attendence Limit:',font=("Cascadia Code",12)),sg.Input(key='-Attendence-',default_text=DefaultAttendence,background_color='white',size=(37,1))],
                [sg.Button('Submit',font=("Cascadia Code",12),focus= True),sg.Button('Cancel',font=("Cascadia Code",12))],
                [sg.Checkbox('Remember Me',font=("Cascadia Code",12),default= True,key='-REMEMBER-')]]
    window = sg.Window('Attendence Calculator',layout = layout,grab_anywhere=True , icon = 'Media\icon.ico',font=("Cascadia Code",12),element_justification='center',resizable=True)
    print('Building GUI....Done')
    print('Waiting for the User to Enter the Credentials....')


    event, values = window.read()
    

    if event == 'Submit':
        print('Waiting for the User to Enter the Credentials....Done')
        if (values['-REMEMBER-'] and values['-USERNAME-'] != '' or values['-PASSWORD-'] != '' ):
            data['Username'] = values['-USERNAME-']
            data['Password'] = values['-PASSWORD-']
            data['Attendence'] = values['-Attendence-']
            with open('secrets.json','w') as f:
                json.dump(data,f)
        
        window.close()        
        main(username = values['-USERNAME-'],password = values['-PASSWORD-'],attnLimit = values['-Attendence-'],data=data)
    
    elif event == 'Cancel' or event == sg.WIN_CLOSED:
            print('Cancelled')
            window.close()
            browser.close()
            exit()

    window.close()


def OpenBrowser():
    '''
    ------------------------------------------------------
    Function to open the browser and Webpage
    ------------------------------------------------------

    Returns:
        browser: [webdriver]
    '''
    # ++++++++++++++++++++++++++++++++++++++++++++++++Scrapping Part++++++++++++++++++++++++++++++++++++++++++++++++
    global browser
    options = Options()

    options.add_argument("--start-minimized")
    browser = webdriver.Chrome(executable_path = r"chromedriver_win32\chromedriver.exe", options = options)
    browser.minimize_window()

    browser.get('https://www.pesuacademy.com/Academy/')

    return browser

def TryElement(browser,elementType,element,action = None, Timeout = 15):
    '''
    ------------------------------------------------------
    Function to click the element once it is present in the webpage    
    ------------------------------------------------------

    @param
    browser: [webdriver]
    elementType: [str]
    element: [str]
    action: [str] (click,text,None) [default = None]
    Timeout: [int] [default = 120 seconds]

    Returns:
        Element: [webdriver element]
    '''
    checking = True
    while checking:
        Timeout -= 1

        if Timeout == 0 and action != 'Input':
            # make a popup window for timeout error
            LoadingWindow.close()
            sg.popup('Timeout Error','Please Check your Internet Connection and Try Again',title='Error',font=("Cascadia Code",12))
            print('Timeout Error')
            exit()
        elif Timeout == 0 and action == 'Input':
            # make a popup window for timeout error
            LoadingWindow.close()
            sg.popup('Timeout Error','Wrong Username or Password',title='Error',font=("Cascadia Code",12))
            print('Wrong Username or Password')
            exit()
        try:
            if action == 'Input':
                Element = browser.find_element(elementType, element)
            elif action == 'click':
                Element = browser.find_element(elementType, element).click()
            elif action == 'text':
                Element = browser.find_element(elementType, element).text
        except:
            time.sleep(1)
            print('Waiting for the Page to Load')
        else:
            checking = False
    return Element
def Scraping(browser,username,password):
    '''
    ------------------------------------------------------
    Function to scrap the data from the webpage
    ------------------------------------------------------

    @param
    browser: [webdriver]
    username: [str]
    password: [str]

    Returns:
        df: [pandas dataframe]
    '''

    # Login
    
    # TODO: Fix the TryElement function
    print('Logging In....')
    user = TryElement(browser,"xpath", '//*[@id= "j_scriptusername" ]',action= "Input")
    pwd = TryElement(browser,"xpath", '/html/body/div[2]/div[1]/div/div[2]/form/fieldset/div[2]/input',action= "Input")
    
    user.send_keys(username)
    pwd.send_keys(password)

    # Submit Button
    TryElement(browser,"xpath", '//*[@id="postloginform#/Academy/j_spring_security_check"]',action= 'click')

    print('Logging In....Done')

    print('Getting the Data....')
    # Attendece Menu
    TryElement(browser,"xpath", '//*[@id="menuTab_660"]/a/span[2]',action= 'click')

    # Getting Info
    table = TryElement(browser,"xpath", '//*[@id="subjetInfo"]',action= 'text')
    item = table.split("\n")

    print()
    # making dataframe
    ClassesAndPercentrage = []
    row = []
    subjectName = []
    for i in item:
            split_list = i.split()
            subjectString = ''
            subjectName.append([subjectString + x for x in split_list[1:-2]])
            ClassesAndPercentrage.append(split_list[-2:])

    # getting subject name
    for i in range(len(subjectName)):
            subjectName[i] = " ".join(subjectName[i])

    df = pd.DataFrame(ClassesAndPercentrage,columns=["Classes","Percentage"],index = subjectName)
    df.index.name = "Subject"

    print('Getting the Data....Done')
    return df

# Attendence Calcuation
def Attendence(Subject ,Classes = '0/0', Attendence_limit = 80, Percents = 0):
    '''
    ------------------------------------------------------
    Function to calculate the Attendence
    ------------------------------------------------------

    @param
    Subject: Subject Name [str]
    Classes: Classes Attended and Total Classes [str]
    Attendence_limit: Attendence Limit [int]
    Percents: Current Percentage [int]

    Returns:
        stringItem: [str]   
    '''

    if type(Attendence_limit) != int or Attendence_limit < 0 or Attendence_limit > 100:
        print('Attendence Limit should be an integer between 0 and 100, Using Default Value (80)')
        Attendence_limit = 80
    Attendence_limit = int(Attendence_limit)
    count=0
    stringItem = ''

    if Classes == 'NA':
        stringItem+='----------------------------------------------------------------------------------------------------\n'
        stringItem += f"\t\t\t\t{Subject}\t\t\t\t\n"
        stringItem += f"No Classes Taken Yet\n"

    else:
        Classes_list = Classes.split('/')
        classesAttended = int(Classes_list[0])
        classesTaken = int(Classes_list[1])
    
        stringItem+='----------------------------------------------------------------------------------------------------\n'
        
        stringItem += f"\t\t\t\t{Subject}\t\t\t\t\n"

        stringItem += f"Current Percentage: {Percents}\n"

        stringItem += f"Current Classes: {Classes}\n"

        if classesTaken<classesAttended:
                classesTaken,classesAttended=classesAttended,classesTaken

        if (classesAttended/classesTaken * 100 < Attendence_limit):
                while (classesAttended/classesTaken * 100 < Attendence_limit):
                    classesAttended+=1
                    classesTaken+=1
                    count+=1

                stringItem += f"Need {count} Classes\n"
        else:                  
                while (classesAttended/classesTaken * 100 > Attendence_limit):
                    classesTaken+=1
                    count+=1
                

                stringItem += f"Can Miss {count - 1} Classes\n"

        # stringItem += f"Min No of Classes to be Attended {classesAttended}\nNo of Classes it will take Totally {classesTaken - 1}\n"
        
        stringItem += f"The New Percentage Will Be: {round(classesAttended/(classesTaken - 1)*100,2)}\n"
    print(stringItem)
    return stringItem

def ShowData(df,TotalData,LoadingWindow,browser,directRead = False):
    '''
    ------------------------------------------------------
    Function to show the data in a new window
    ------------------------------------------------------

    @param
    df: [pandas dataframe]
    TotalData: [str]
    LoadingWindow: [sg.Window]
    browser: [webdriver]

    Returns:
        None
    '''
    if not directRead:
        print('Closing the Browser....')
        browser.close()
        print('Closing the Browser....Done')

    print('Showing the Data....')
    print('Building the Window....')

    out = df.to_markdown(tablefmt='pipe', colalign=['center']*len(df.columns))
    
    col = [sg.Column([[sg.Text(out,font=('Courier'))],[sg.Text(TotalData,font=('Courier'))],[sg.Button('Ok',key="--OK--",auto_size_button=True,font= ('Courier'))]],scrollable=True, element_justification='center',vertical_scroll_only=True)]
    layout = [col]
    
    newWindow = sg.Window('Data',layout=layout,resizable=True , icon='Media/icon.ico')
    LoadingWindow.close()

    print('Closing the Loading Window....Done')
    
    print('Building the Window....Done')
    print('Showing the Data....Done')
    
    event, values = newWindow.read()
    


    if event == 'Ok' or event == sg.WIN_CLOSED or event == '--OK--':
        newWindow.close()
        print('Exiting....')
        print('Exiting....Done')
        print('Thank You for using this Software')
        exit()

def main(username,password,attnLimit = 80, data = None):
    '''
    ------------------------------------------------------
    Main Function
    ------------------------------------------------------

    @param
    username: [str]
    password: [str]
    attnLimit: [int]

    Returns:
        None
    '''

    global LoadingWindow
    print("Loading....")

    LoadingWindow = sg.Window('Loading',[[sg.Text('Loading....',font=("Casadia Code",15))]] , icon='Media/icon.ico', grab_anywhere=True)
    LoadingWindow.read(timeout=0)

    if data['Data']['Date'] == time.strftime("%d/%m/%Y") and data['Data']['DataFrame'] != '' :
        print('Opening the Data Window....')
        print('Building the Window....')

        print('Getting the Data Directly from Preivous Data Scrap....')

        df = data['Data']['DataFrame']
        # converting json to pandas dataframe
        df = pd.read_json(df, orient='index')
        print('Getting the Data Directly from Preivous Data Scrap....Done')


        TotalData = ""
        print('Calculating Attendence....')
        for ind in df.index:
            text = Attendence(Subject = ind,Classes = df["Classes"][ind], Attendence_limit = int(attnLimit), Percents = df["Percentage"][ind])
            TotalData += text
        print('Calculating Attendence....Done')
        print('Closing the Loading Window....')
        print('Building the Data Window....')
        ShowData(df,TotalData,LoadingWindow,None,directRead=True)
        return
    
    else:

        print("Opening Browser....")
        browser = OpenBrowser()
        print("Opening Browser....Done")

        df = Scraping(browser=browser,username=username,password=password)

        json_df = df.to_json(orient='index')

        with open('secrets.json','w') as f:
            json.dump({'Username':username,'Password':password,'Attendence':'80','Data':{'Date':time.strftime("%d/%m/%Y"),'DataFrame':json_df}},f)
        # Attendence(Classes = (df.loc[row[0],"Classes"]), Attendence_limit = 80)
        TotalData = ""
        print('Calculating Attendence....')
        for ind in df.index:
                text = Attendence(Subject = ind,Classes = df["Classes"][ind], Attendence_limit = int(attnLimit), Percents = df["Percentage"][ind])
                TotalData += text
        print('Calculating Attendence....Done')
        print('Closing the Loading Window....')
        print('Building the Data Window....')
        ShowData(df,TotalData,LoadingWindow,browser)

if __name__ == "__main__":
    print('Starting....')
    print('Building GUI...')
    buildGUI()