objects.py

from bs4 import BeautifulSoup as bs
from typing import List,TypeVar,Tuple
from copy import deepcopy
from tabulate import tabulate
import sys
import os
import extensions

def make_root(path,encode='cp950'):
    """create the root bs4 object of a html """
    data=open(path,mode='r',encoding=encode,errors='ignore')
    root=bs(data.read(),"html.parser")
    return root

def get_ACIXSTORE():
    print("""
          Go to :

          https://www.ccxp.nthu.edu.tw/ccxp/INQUIRE/JH/6/6.2/6.2.9/JH629001.php

          get the url of a course's syllabus for better user experience , or enter "N" to ignore.
          """)
    
    SYLLABUS=input("url: ")
    if SYLLABUS != "N" :
        ACIXSTORE=SYLLABUS[(SYLLABUS.find("ACIXSTORE=")+len("ACIXSTORE=")):(SYLLABUS.find("&c_key="))]
    else : 
        ACIXSTORE=None

    return  ACIXSTORE

class selector:

    with open("extensions.py","r") as f:
            extension_file = f.read()
            f.close()
    
    exec(extension_file)
    
    
    def __init__(self,LCES_lis:List[bs],CASD:bs,ACIXSTORE) -> None:
        
        
        
        self.ACIXSTORE=ACIXSTORE
        self.courses=[]
        infos=dict()

        body_CASD=CASD.find("tbody")
        courses_CASD=body_CASD.find_all(lambda tag: tag.name == 'tr' and not tag.attrs)

        for cours in courses_CASD:

            content=[cours.contents[i] for i in range(1,len(cours.contents),2)] 
            num,name,teacher,people,gpa,deviation,score,score_diviation=[c.text.strip() for c in content]
            if gpa == '' and deviation == '':
                score,score_diviation=float(score),float(score_diviation)
                gpa,deviation = -1 , -1 
                tp =False
              
            else :
                gpa,deviation=float(gpa),float(deviation)
                score = -1
                score_diviation = -1
                tp= True
              

            infos[name+":"+teacher]=(gpa,deviation,score,score_diviation,tp)

        for LCES in LCES_lis:
            body_LCES=LCES
            courses_LCES=body_LCES.find_all("tr",attrs={'class':"word"})
            
            for cours in courses_LCES:
                
                content=[cours.contents[i] for i in range(1,len(cours.contents),2)]
                
                if len([c.text for c in content]) == 9:
                    num,name,teacher,dimension,time,none1,none2,x1,x2=[c.text for c in content]
                else:
                    num,name,teacher,time,none1,none2,x1,x2=[c.text for c in content]

                if x1=="無限制":
                    possibility=1
                else:
                    possibility=float(x1)/(float(x2)+1)

                if not infos.get(name+":"+teacher):
                    continue
                else:
                    gpa,deviation,score,score_diviation,tp=infos[name+":"+teacher]

                if time == ' ' :
                    time = 'X'

                self.courses.append(course(num,name,teacher,time,possibility,gpa,deviation,score,score_diviation,tp))
                
                


        self.tables={"BASE":deepcopy(self.courses)}
        self.cur_table="BASE"
        self.show_list={
            "num":True,
            "name":True,
            "teacher":True,
            "time":True,
            "possibility":True,
            "gpa":True,
            "deviation":True,
            "score":True,
            "score_deviation":True
        }
        
        self.show_mode="html"

        self.show(self.show_mode)
        
        
        

        
    

    def run(self,cmds):
        cmds=cmds.split()
        
        try:
            
            if cmds[0] == "list":
                current=[]
                print("start listing....")

                courses= deepcopy(self.courses)
    
                for cs  in courses :
                    loc={}
                    for member in self.show_list.keys():   
                        exec(member+"= cs." + member,locals(),loc)
                        
                    for key in loc.keys():
                        locals()[key]=loc[key]
                    
                        
                    
                    exec("current.append(cs) if ("+" ".join(cmds[1:])+") else cs\n")
                
                self.tables[self.cur_table]=current

                print("listed")
                
            
                
            elif cmds[0] == "filter":
                current=[]
                
                for cs  in self.tables[self.cur_table] :
                    
                    loc={}
                    for member in self.show_list.keys():   
                        exec(member+"= cs." + member,locals(),loc)
                        
                    for key in loc.keys():
                        locals()[key]=loc[key]
                    
                
                    exec("current.append(cs) if ("+" ".join(cmds[1:])+") else cs\n")

                self.tables[self.cur_table]=current

                print("filtered!")

                
            
            elif cmds[0] == "change":
                
                if not cmds[1] in self.tables:
                    print("no such table")
                else :
                    self.cur_table=cmds[1]
                    print("table changed !")


            elif cmds[0] == "sort":
                #thing
                def func(course,string):

                    for  member in self.show_list:
                        exec(member+"= course." + member)
                    
                    loc = {}
                    exec("answer="+string, locals(), loc)
                    return loc["answer"]

                if cmds[1]=="a":
                    reverse="False"
                elif cmds[1] == "d":
                    reverse="True"

                

                exec("self.tables[self.cur_table].sort(key=lambda course : func(course,'"+" ".join(cmds[2:])+"'),reverse="+reverse+")",locals())

                print("sorted !")
            
            elif cmds[0] == "function":
                
                exec("self.tables[self.cur_table]=self."+ " ".join(cmds[1:]))
                print("function executed !! ")

            elif cmds[0] == "remove":
                for i in cmds[1:]:
                    number = int(i)
                    del self.tables[self.cur_table][number]
                
                print("Removed !")

            elif cmds[0] == "add_table":
                name=cmds[1]
                if name in self.tables:
                    print("table existed !! ")
                else:
                    if(len(cmds)>2):
                        self.add_table(name,deepcopy(self.tables[cmds[2]]))
                    else: 
                        self.add_table(name)
                    
                    print("table added  !")

            elif cmds[0] == "delete_table":
                name=cmds[1]
                if len(self.tables.keys())==1:

                    print("need at least one table !!")
                    
                else:

                    self.delete_table(name)

                    if name==self.cur_table:
                        self.cur_table=list(self.tables.keys())[0]

                    print("table deleted !")

            elif cmds[0] == "reset":
                self.tables[self.cur_table]=deepcopy(self.courses)

                print("reset completed !")

            elif cmds[0] == "show":
                for i in cmds[1:]:
                    self.show_list[i]=True

            elif cmds[0] == "unshow":
                for i in cmds[1:]:
                    self.show_list[i]=False

            elif cmds[0] == "show_mode":
                    
                    if cmds[1]=="html":
                        self.show_mode="html"

                    elif cmds[1]=="text":
                        self.show_mode="fancy_grid"

            elif cmds[0] == "renew_url":
                    self.ACIXSTORE=get_ACIXSTORE()
            
            elif cmds[0] == "convert":
                for i in range(len(self.tables[self.cur_table])):
                    if self.tables[self.cur_table][i].tp == False:
                        self.tables[self.cur_table][i].gpa,self.tables[self.cur_table][i].deviation = self.SCORE_TO_GPA(self.tables[self.cur_table][i].score,self.tables[self.cur_table][i].score_deviation)
                    else:
                        self.tables[self.cur_table][i].score,self.tables[self.cur_table][i].score_deviation = self.GPA_TO_SCORE(self.tables[self.cur_table][i].gpa,self.tables[self.cur_table][i].deviation)
                print("converted!")

            else:
                print("invalid command !! ")
    
            self.show(self.show_mode)

        except Exception as e:
            print("Error,",e)
            



       

        return 
                    
            
        
    def add_table(self,name,courses=[]):
        self.tables[name]=courses
        self.cur_table=name
        return 

    def delete_table(self,name):
        if name in  self.tables:
            del  self.tables[name]
   
    def show(self,mode):
        table=[]
        if mode=="html":
            sys.stdout=open(os.path.dirname(__file__)+"/result.html","w",encoding='utf-8',errors='ignore')
        else:
            sys.stdout=open(os.path.dirname(__file__)+"/result.txt","w",encoding='utf-8',errors='ignore')

        to_show=[]
        for j in self.show_list.keys():
            if self.show_list[j]:
                to_show.append(j)
        
        show_format={
            "num":"COURSE NUMBER",
            "name":"COURSE NAME",
            "teacher":"TEACHER",
            "time":"TIME",
            "possibility":"ENROLL POSSIBILITY",
            "gpa":"AVERAGE GPA",
            "deviation":"DEVIATION",
            "score":"AVERAGE SCORE",
            "score_deviation":"SCORE DEVIATION"}
        
        headers=[show_format[i] for i in to_show]
        
        table.append(headers)

        for i in range(len(self.tables[self.cur_table])):
            

            table.append(self.tables[self.cur_table][i].show(to_show))
            
        
        tmp = tabulate(table,headers='firstrow',showindex="always",tablefmt=mode,numalign="center")

        

        if mode == "html" :
            tmp = "<body bgcolor='#EBDEF0'>" + tmp + "</body>"

            output=bs(tmp,"html.parser")
            
            if self.ACIXSTORE:
                # add link of courses 
                if not "name" in to_show:
                    pass
                else:
                    ind = to_show.index("name")
                    courses=output.find("tbody").find_all("tr")
                    for i in range(len(courses)):
                        
                        tmp=(str(courses[i].contents[1].string).split(" "))
                        year=tmp[0]
                        num = tmp[-1]

                        news=output.new_tag("a", href="https://www.ccxp.nthu.edu.tw/ccxp/INQUIRE/JH/common/Syllabus/1.php?ACIXSTORE="+self.ACIXSTORE+"&c_key="+year+"%20%20"+num)
                        news.string=courses[i].contents[ind+1].string
                        courses[i].contents[ind+1].clear()
                        courses[i].contents[ind+1].append(news)

            courses=output.find("tbody").find_all("tr")

            for i in range(len(courses)):
                tmp=self.tables[self.cur_table][i]
                if (not tmp.tp) and tmp.gpa != -1 :
                    
                    if not "gpa" in to_show:
                        pass
                    else:
                        ind = to_show.index("gpa")
                        courses[i].contents[ind+1].attrs['style']+="color:red;"

                    if not "deviation" in to_show:
                        pass
                    else:
                        ind = to_show.index("deviation")
                        courses[i].contents[ind+1].attrs['style']+="color:red;"
                    
                elif (tmp.tp) and tmp.score != -1 :
                    if not "score" in to_show:
                        pass
                    else:
                        ind = to_show.index("score")
                        courses[i].contents[ind+1].attrs['style']+="color:red;"
                    if not "score_deviation" in to_show:
                        pass
                    else:
                        ind = to_show.index("score_deviation")
                        courses[i].contents[ind+1].attrs['style']+="color:red;"
            
                

            news=output.new_tag("h1",align='center' ,style="background-color:powderblue;border-radius: 12px;")
            news.string="CURRENT TABLE : "+ self.cur_table
            output.table.insert(0,news)


            output.table.attrs["bgcolor"]="#ECF0F1"
            output.table.attrs["style"]="border-radius: 12px;"
            output.table.attrs["align"]="center"

            """
            titles = output.find("tr").contents

           
            for title in titles:
                tmp = title.string
                title.clear()
                news = output.new_tag("b")
                news.string = tmp
                title.append(news)

            """
            print(output.prettify())
        elif mode == "fancy_grid" :
            print(tmp)

        sys.stdout=sys.__stdout__

    
    


class course:
    def __init__(self,num,name,teacher,time,possibility,gpa,deviation,score,score_deviation,tp) -> None:
        self.num=num
        self.name=name
        self.teacher=teacher
        self.time=time
        self.possibility=possibility
        self.gpa=gpa
        self.deviation=deviation
        self.score=score
        self.score_deviation=score_deviation
        self.tp=tp
       

    def show(self,to_show):

        tmp = []
        for m in to_show:
                exec("tmp.append(self."+m+")")
    
        return tmp
        
        #print(self.num," ",self.name," ",self.teacher," ",self.time," ",self.possibility," ",self.gpa," ",self.deviation)