import requests
import re
import os
import http.cookiejar
import base64
from urllib.parse import urljoin
from bs4 import BeautifulSoup
import lxml

BASE_URL = "http://222.240.44.2:8899/C6/"


class OA:
    def __init__(self, base_url: str, username: str, pwd: str):
        self.username = username
        self.pwd = pwd
        self.session = requests.session()
        self.base_url = base_url

    def login(self):
        if os.path.exists("cookies.txt"):

            load_cookiejar = http.cookiejar.MozillaCookieJar()
            load_cookiejar.load("cookies.txt", ignore_discard=True, ignore_expires=True)
            load_cookies = requests.utils.dict_from_cookiejar(load_cookiejar)
            cookies = requests.utils.cookiejar_from_dict(load_cookies)
            self.session.cookies = cookies

            if not self.session.get(urljoin(self.base_url, "JHSoft.Web.WorkFlat/FlatWorksDone.aspx")).ok:
                self.__login()

        else:
            self.__login()

    def __login(self):
        print("re-login")
        self.session.cookies = http.cookiejar.MozillaCookieJar(filename="cookies.txt")
        login_path = urljoin(self.base_url, "JHSoft.Web.Login/AjaxForLogin.aspx")
        _u = base64.b64encode(self.username.encode('utf-8'))
        _p = base64.b64encode(self.pwd.encode('utf-8'))

        response = self.session.post(login_path, data={
            "type": "login",
            "loginCode": _u,
            "pwd": _p
        }, headers={
            "X-Requested-With": "XMLHttpRequest"
        })

        if response.ok:
            print("Login Success: {}".format(self.username))
            self.session.cookies.save(ignore_expires=True, ignore_discard=True)
        else:
            print("Login Failed, error code: {}".format(response.status_code))

    def works(self):
        u = urljoin(self.base_url, "JHSoft.Web.WorkFlat/FlatWorksDone.aspx")
        res = self.session.get(u)

        if res.ok:
            self.parse(res)

    def parse(self, response):

        info_sql = "<root><No djsn='{}'>c2VsZWN0ICogZnJvbSBDdXN0b21Nb2R1bGVfMjAyMTA4MDIwNjAxIHdoZXJlIE1haW5JRD0nSkhDMDAwMzk4MTgn</No><No1>1</No1><No2>-1</No2><No3>MainID;lsh;sqrq;sqr;yggh;bm;zw;qjlx;qjsy;qjksrq;shi1;ksf;qjjsrq;shi2;jsf;gong;tspd</No3></root>"

        soup = BeautifulSoup(response.text, features="lxml")
        datatable = soup.find(id="DGWorksDone_TBody").find_all("tr")
        jbtj = filter(lambda x: "JBTJ" in x.text, datatable)
        qjsq = filter(lambda x: "QJSQ" in x.text, datatable)

        work_overtime = map(lambda x: re.search(re.compile(r"JBTJ\d{8}-\d{4}"), x.text).group(0), jbtj)
        leave = map(lambda x: re.search(re.compile(r"QJSQ\d{8}-\d{4}"), x.text).group(0), qjsq)

        # for jb in jbtj:
        cut_path = re.match(re.compile(r"javascript:ClickTitle\('\.\./(.+)'\)"),jbtj.__next__().contents[1].a['onclick']).group(1)
        djframe_path = urljoin(self.base_url,cut_path)
        res = self.session.get(djframe_path,allow_redirects=True)
        if res.ok:
            print(res.text)

    def close(self):
        self.session.close()


test = OA(BASE_URL, "120082", "7948799")

test.login()

test.works()

test.close()