From 445ff9489ae1fe72a3d36d70ea08d235cc39fb9e Mon Sep 17 00:00:00 2001 From: sleptworld Date: Fri, 22 Sep 2023 01:54:49 +0800 Subject: [PATCH] test --- cookies.txt | 4 +- oa.py | 204 +++++++++++++++++++++++++++++++++++++++++++++---- test.html | 216 ---------------------------------------------------- tst.html | 0 4 files changed, 192 insertions(+), 232 deletions(-) delete mode 100644 test.html create mode 100644 tst.html diff --git a/cookies.txt b/cookies.txt index 155b8b8..f1978de 100644 --- a/cookies.txt +++ b/cookies.txt @@ -2,6 +2,6 @@ # http://curl.haxx.se/rfc/cookie_spec.html # This is a generated file! Do not edit. -222.240.44.2 FALSE / FALSE ASP.NET_SessionId r3esuett3tcxqtmlqedcxleb +222.240.44.2 FALSE / FALSE ASP.NET_SessionId gx5ocm3lp1x1scmyb1zs11rb 222.240.44.2 FALSE / FALSE 253402300799 Login PassWord -222.240.44.2 FALSE / FALSE 253402300799 SessionID120082 r3esuett3tcxqtmlqedcxleb +222.240.44.2 FALSE / FALSE 253402300799 SessionID120082 gx5ocm3lp1x1scmyb1zs11rb diff --git a/oa.py b/oa.py index 3950228..c7995a2 100644 --- a/oa.py +++ b/oa.py @@ -1,14 +1,20 @@ import requests +import sqlite3 import re import os import http.cookiejar import base64 from urllib.parse import urljoin from bs4 import BeautifulSoup +from datetime import datetime import lxml BASE_URL = "http://222.240.44.2:8899/C6/" +INTERVALS = [ + (datetime.strptime('12:00:00', '%H:%M:%S'), datetime.strptime('13:00:00', '%H:%M:%S')), + (datetime.strptime('18:30:00', '%H:%M:%S'), datetime.strptime('19:30:00', '%H:%M:%S')) +] class OA: def __init__(self, base_url: str, username: str, pwd: str): @@ -16,6 +22,8 @@ class OA: self.pwd = pwd self.session = requests.session() self.base_url = base_url + self.work_over = {} + self.leave={} def login(self): if os.path.exists("cookies.txt"): @@ -52,6 +60,7 @@ class OA: self.session.cookies.save(ignore_expires=True, ignore_discard=True) else: print("Login Failed, error code: {}".format(response.status_code)) + print(response.text) def works(self): u = urljoin(self.base_url, "JHSoft.Web.WorkFlat/FlatWorksDone.aspx") @@ -61,32 +70,199 @@ class OA: self.parse(res) def parse(self, response): - - info_sql = "c2VsZWN0ICogZnJvbSBDdXN0b21Nb2R1bGVfMjAyMTA4MDIwNjAxIHdoZXJlIE1haW5JRD0nSkhDMDAwMzk4MTgn1-1MainID;lsh;sqrq;sqr;yggh;bm;zw;qjlx;qjsy;qjksrq;shi1;ksf;qjjsrq;shi2;jsf;gong;tspd" - 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: + self.__single_work_over_parse(jb) + + for leave in filter(lambda x: "QJSQ" in x.text, datatable): + self.__single_leave_parse(leave) + + def __single_leave_parse(self,leave_node): + info_sql = "c2VsZWN0ICogZnJvbSBDdXN0b21Nb2R1bGVfMjAyMTA4MDIwNjAxIHdoZXJlIE1haW5JRD0nSkhDMDAwMzk4MTgn1-1MainID;lsh;sqrq;sqr;yggh;bm;zw;qjlx;qjsy;qjksrq;shi1;ksf;qjjsrq;shi2;jsf;gong;tspd" + + current_id = re.search(re.compile(r"QJSQ\d{8}-\d{4}"), leave_node.text).group(0) + cut_path = re.match(re.compile(r"javascript:ClickTitle\('\.\./(.+)'\)"),leave_node.contents[1].a['onclick']).group(1) + djframe_path = urljoin(self.base_url,"JHSoft.Web.CustomQuery/GeneralXmlhttpPage.aspx") + + r = self.session.get(djframe_path,params={ + "type" : "base64DecodeByService", + "params": cut_path[45:-2] + }) + + if r.ok: + djsn_id = re.match(re.compile(r"djsn=(.+)&djtype"), r.text).group(1) + res = self.session.post(urljoin(self.base_url,"JHSoft.Web.Module/eformaspx/WebBill.aspx?dataset_fields1"),data=info_sql.format(djsn_id)) + if res.ok: + qj_datas = re.findall(re.compile(r""),res.text) + + if qj_datas[7] == "[调休]": + return + + start_date = qj_datas[9][1:-1] + start_hour = qj_datas[10][1:-1] + start_min = qj_datas[11][1:-1] + end_date = qj_datas[12][1:-1] + end_hour = qj_datas[13][1:-1] + end_min = qj_datas[14][1:-1] + + starttime = datetime.strptime(("{} {}:{}").format(start_date, start_hour, start_min), "%Y-%m-%d %H:%M") + endtime = datetime.strptime(("{} {}:{}").format(end_date, end_hour, end_min), "%Y-%m-%d %H:%M") + + self.leave[current_id] = { + "starttime":starttime, + "endtime":endtime, + "spend": self.__total_time_excluding_intervals(starttime, endtime, INTERVALS) + } + + rely = self.session.get(urljoin(self.base_url,"JHSoft.Web.Module/ToolBar/toolbarwf.aspx?{}".format(cut_path[45:])) + ,allow_redirects=True + ,headers={ + "Referer": urljoin(self.base_url,"JHSoft.Web.Module/fceform/common/djframe.htm?{}".format(cut_path[45:])), + "Upgrade-Insecure-Requests" : "1" + }) + if rely.ok: + rely_soup = BeautifulSoup(rely.text,features="lxml") + rely_app_id = rely_soup.find(id="hidRelevanceApproveID")['value'] + + if rely_app_id == "" or rely_app_id is None: + raise Exception("rely error") + + relyinfo = self.session.get(urljoin(self.base_url,"JHSoft.Web.WorkFlow/WorkFlow/MyWorkFlowWatchRelevanceAjax.aspx"),params={ + "delFlag" : 0, + "strAppID" : rely_app_id + }) + + if relyinfo.ok: + relyinfo_soup = BeautifulSoup(relyinfo.text,features="lxml") + trs = relyinfo_soup.find_all(id="ReleAppID{}".format(rely_app_id)) + + used_jb = [] + + for tr in trs: + rely_work_over_id = re.search(re.compile(r"JBTJ\d{8}-\d{4}"), tr.contents[0].text).group(0) + used_jb.append(rely_work_over_id) + + self.leave[current_id]["used_work"] = used_jb + + else: + raise Exception("rely error") + + else: + raise Exception("rely error") + + + + def __single_work_over_parse(self,work_node): + info_sql = "c2VsZWN0ICogZnJvbSBDdXN0b21Nb2R1bGVfMjAyMTA4MDIwNzAyIGEgaW5uZXIgam9pbiBDdXN0b21Nb2R1bGVfU3ViRGV0YWlsXzIwMjEwODAyMDcwMiBiIG9uIChhLk1haW5JRD1iLk1haW5JRCkgIHdoZXJlIGEuTWFpbklEPSdKSEMwMDAzOTI5NSc=1-1MainID;lb;kssj;kssj_shi;kssj_fen;jssj;jssj_shi;jssj_fen;bz" + cut_path = re.match(re.compile(r"javascript:ClickTitle\('\.\./(.+)'\)"),work_node.contents[1].a['onclick']).group(1) + djframe_path = urljoin(self.base_url,"JHSoft.Web.CustomQuery/GeneralXmlhttpPage.aspx") + + current_id = re.search(re.compile(r"JBTJ\d{8}-\d{4}"), work_node.text).group(0) + print(current_id) + + res = self.session.get(djframe_path,params={ + "type" : "base64DecodeByService", + "params": cut_path[45:-2] + }) - # 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: + djsn_id = re.match(re.compile(r"djsn=(.+)&djtype"), res.text).group(1) + res = self.session.post(urljoin(self.base_url,"JHSoft.Web.Module/eformaspx/WebBill.aspx?dataset_fields1"),data=info_sql.format(djsn_id)) + + if res.ok: + qj_datas = re.findall(re.compile(r""),res.text) + + rs = [] + + for q in range(0, len(qj_datas), 9): + start_date = qj_datas[q + 2] + start_hour = qj_datas[q + 3] + start_min = qj_datas[q + 4] + + end_date = qj_datas[q + 5] + end_hour = qj_datas[q + 6] + end_min = qj_datas[q + 7] + + starttime = datetime.strptime(("{} {}:{}").format(start_date[1:-1], start_hour[1:-1], start_min[1:-1]), "%Y-%m-%d %H:%M") + endtime = datetime.strptime(("{} {}:{}").format(end_date[1:-1], end_hour[1:-1], end_min[1:-1]), "%Y-%m-%d %H:%M") + + spend = self.__total_time_excluding_intervals(starttime, endtime, INTERVALS) + + rs.append({ + "start": starttime, + "end": endtime, + "spend": spend + }) + + self.work_over[current_id] = rs + + else: print(res.text) + raise Exception("djframe error") + + + def __intersection_time(self,start1, end1, start2, end2): + # 返回两个时间段的交集 + return max(start1, start2), min(end1, end2) + + def __total_time_excluding_intervals(self,start, end, intervals): + total_duration = end - start + for interval in intervals: + t1 = interval[0].replace(year=start.year, month=start.month, day=start.day) + t2 = interval[1].replace(year=start.year, month=start.month, day=start.day) + inter_start, inter_end = self.__intersection_time(start, end, t1,t2) + if inter_start < inter_end: # 有交集 + total_duration -= (inter_end - inter_start) + return total_duration + def close(self): self.session.close() -test = OA(BASE_URL, "120082", "7948799") +if __name__ == '__main__': -test.login() + test = OA(BASE_URL, "120082", "7948799") + test.login() + test.works() + test.close() -test.works() + leaves = test.leave + works = test.work_over + + result_works = {} + + for i,work in works.items(): + spend_time = sum(work["spend"],datetime.timedelta(0)) + result_works[i] = { + "spend" : spend_time, + "used_leave" : [], + "remaining" : spend_time + } + + for i,leave in leaves.items(): + used = leave['used_work'] + spend = leave['spend'] + + for u in used: + if spend > result_works[u]['remaining']: + result_works[u]['used_leave'].append({ + "id" : i, + "spend" : work[u]['remaining'] + }) + result_works[u]["remaining"] = datetime.timedelta(0) + spend -= result_works[u]['remaining'] + + else: + result_works[u]['used_leave'].append({ + "id" : i, + "spend" : spend + }) + result_works[u]["remaining"] -= spend + + + print(result_works) -test.close() diff --git a/test.html b/test.html deleted file mode 100644 index 6c3fada..0000000 --- a/test.html +++ /dev/null @@ -1,216 +0,0 @@ - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
  - -
- -------------- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
-
请假申请单
-
流水号
 -
 
 -
申请日期
 -
 
-
申请人
 -
 
 -
员工工号
 -
 
-
部门
 -
 -
请假类型
 -
 
-
请假时间
 -
从
 -
 
 -
时
 -
 
 -
分
 -
 
 -
共
 -
 
 -
天
-
至
 -
 
 -
时
 -
 
 -
分
 -
-
请假事由
 -
-
审批意见
-
-
 
 
-
附件
 -
-
- - - -
增加附件
 
-
所有的假单证明材料（如：结婚证复印件、病例证明、生育证明等）请扫描后添加附件。
diff --git a/tst.html b/tst.html new file mode 100644 index 0000000..e69de29