#!/usr/bin/python3
import requests
from bs4 import BeautifulSoup
from prettytable import PrettyTable
import sys
from urllib.parse import urlencode
import getpass


class Score:
    def __init__(self, stuid: str, pwd: str):
        self.data = {
            'stuid': stuid,
            'pwd': pwd
        }
        self.cookie = ''
        self.bs = None
        self.rank_bs = None
        self.table = None
        self.courses = None
        self.out = PrettyTable()
        self.out.field_names = ['课程名', '考试类型', '学分', '实验成绩', '平时成绩', '期中成绩', '期末成绩', '总成绩',
                                '选课人数', '最高分', '最低分', '排名']
        self.get_cookie()
        self.all_courses = None
        self.gpa = 0
        self.cur_gpa = 0

    def get_cookie(self, retry_times=5):
        if retry_times == 0:
            print('无法获取 Cookies')
            exit(1)
        try:
            req2 = requests.post('http://222.194.15.1:7777/pls/wwwbks/bks_login2.login', self.data,
                                 allow_redirects=False)
            self.cookie = req2.headers.get('Set-Cookie')
        except Exception as e:
            print(e)
            self.get_cookie(retry_times - 1)

    def __get_courses_bs(self):
        req = requests.get('http://222.194.15.1:7777/pls/wwwbks/bkscjcx.curscopre',
                           headers={'Cookie': self.cookie})
        req.encoding = 'GBK'
        if len(req.text) == 0:
            raise Exception('无法获取成绩')
        self.bs = BeautifulSoup(req.text, 'lxml')
        self.table = self.bs.find_all('table')[4]
        self.courses = self.table.find_all('tr')[1:]

    def __get_gpa_bs(self):
        req = requests.get('http://222.194.15.1:7777/pls/wwwbks/bkscjcx.yxkc',
                           headers={'Cookie': self.cookie})
        req.encoding = "GBK"
        if len(req.text) == 0:
            raise Exception('无法获取成绩')
        self.bs = BeautifulSoup(req.text, 'lxml')
        tmp = self.bs.find_all('table')[4]
        self.all_courses = tmp.find_all('tr')[1:]

    def __get_score(self):
        ret = []
        for course in self.courses:
            td = course.find_all('td')
            ret.append([td[2].find('p').text, td[13].find('p').text,
                        td[4].find('p').text, td[6].find('p').text,
                        td[7].find('p').text, td[8].find('p').text,
                        td[9].find('p').text, td[10].find('p').text, ])
        return ret

    def __get_rank(self):
        rank_data = ''
        for course in self.courses:
            td = course.find_all('td')
            if td[0].input is None:
                continue
            rank_data += '&' + urlencode({'p_pm': td[0].input.attrs['value'].encode('GBK')})
        rank_data = rank_data[1:]
        req2 = requests.post('http://222.194.15.1:7777/pls/wwwbks/bkscjcx.cursco',
                             rank_data,
                             headers={'Cookie': self.cookie})
        req2.encoding = 'GBK'
        if len(req2.text) == 0:
            raise Exception('无法获取排名')
        ret = {}
        self.rank_bs = BeautifulSoup(req2.text, 'lxml').find_all('table')[4].find_all('tr')[1:]
        for course_rank in self.rank_bs:
            td = course_rank.find_all('td')
            ret[td[1].text] = [td[3].text, td[4].text, td[5].text, td[6].text]
        return ret

    def __get_gpa(self):
        all_tmp = []
        cur_tmp = []
        for course in self.all_courses:
            td = course.find_all('td')
            all_tmp.append([td[3].find('p').text, td[9].find('p').text])
        for course in self.courses:
            td = course.find_all('td')
            cur_tmp.append([td[2].find('p').text, td[6].find('p').text])
        credit = 0.0
        total_grade = 0.0
        for course in all_tmp:
            if(float(course[1]) >= 60.0):
                credit += float(course[0])
                total_grade += float(course[0]) * float(course[1])
        self.gpa = total_grade / credit / 100 * 4
        cur_credit = 0.0
        cur_total_grade = 0.0
        for course in cur_tmp:
            cur_credit += float(course[0])
            cur_total_grade += float(course[0]) * float(course[1])
        if(cur_credit == 0):
            return
        else:
            self.cur_gpa = cur_total_grade / cur_credit / 100 * 4

    @staticmethod
    def __append_rank(scores, ranks):
        ret = []
        for i in scores:
            ret.append(i + ranks.get(i[0], ['', '', '', '']))
        return ret

    def run(self, retry_times=5):
        if retry_times == 0:
            print('退出')
            exit(1)
        try:
            self.__get_courses_bs()
            self.__get_gpa_bs()
            result = self.__append_rank(self.__get_score(), self.__get_rank())
            for i in result:
                self.out.add_row(i)
            print(self.out)
            self.__get_gpa()
            if (self.cur_gpa == 0):
                print("No avilible course this semester for now")
            else:
                print("GPA of this semester : ", end="")
                print('%.2f' % (self.cur_gpa))
            print("GPA of all courses   : ", end="")
            print('%.2f' % (self.gpa))
        except Exception as e:
            print(e)
            self.run(retry_times - 1)


def print_usage():
    print("""Usage:
    {}
    {} 学号
    {} 学号 密码""".format(sys.argv[0], sys.argv[0], sys.argv[0]))
    exit(1)


if __name__ == '__main__':
    stu_id = None
    pwd = None
    if len(sys.argv) == 1:
        stu_id = input('学号：')
        pwd = getpass.getpass('密码：')
    elif len(sys.argv) == 2 and len(sys.argv[1]) == 9:
        stu_id = sys.argv[1]
        pwd = getpass.getpass('密码：')
    elif len(sys.argv) == 3 and len(sys.argv[1]) == 9:
        stu_id = sys.argv[1]
        pwd = sys.argv[2]
    else:
        print_usage()
        exit(1)
    Score(stu_id, pwd).run()