From ace509b0c6d58aa9a24364a7198cb75fe120fc6c Mon Sep 17 00:00:00 2001 From: Guest Liang <47164120+Guest-Liang@users.noreply.github.com> Date: Thu, 5 Sep 2024 13:46:58 +0800 Subject: [PATCH] =?UTF-8?q?=E7=9B=B4=E6=8E=A5ip=EF=BC=8C=E7=BB=95=E8=BF=87?= =?UTF-8?q?=E8=A7=A3=E6=9E=90=E9=98=B2=E6=AD=A2=E5=8D=A1=E4=BD=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Constant.py | 100 +++++------ Main.py | 485 +++++++++++++++++++++++++++------------------------- 2 files changed, 298 insertions(+), 287 deletions(-) diff --git a/Constant.py b/Constant.py index a6a5df7..3fadb3b 100644 --- a/Constant.py +++ b/Constant.py @@ -1,51 +1,51 @@ -#个人信息Personal Information -SchoolID='' #引号之间填入学号 -Password_jwgl='' #引号之间填入教务系统密码 -StartDate='2023-02-20' #引号之间填入开始日期，格式为'2023-02-20' - -#教务系统加密算法，由[https://jwgl.bupt.edu.cn/jsxsd/framework/xsMain.jsp]得到的conwork.js -CONWORK_JS = ''' -eval(function (p, a, c, k, e, d) -{ - e = function (c){return (c < a ? "": e(parseInt(c / a))) + ((c = c % a) > 35 ? String.fromCharCode(c + 29) : c.toString(36))}; - if (!''.replace(/^/, String)) - { - while (c--) d[e(c)] = k[c] || e(c); - k = [function (e){return d[e]}]; - e = function() {return '\\w+'}; - c = 1; - }; - while (c--) if (k[c]) p = p.replace(new RegExp('\\b' + e(c) + '\\b', 'g'), k[c]); - return p; -} ('b 9="o+/=";p q(a){b e="";b 8,5,7="";b f,g,c,1="";b i=0;m{8=a.h(i++);5=a.h(i++);7=a.h(i++);f=8>>2;g=((8&3)<<4)|(5>>4);c=((5&s)<<2)|(7>>6);1=7&t;k(j(5)){c=1=l}v k(j(7)){1=l}e=e+9.d(f)+9.d(g)+9.d(c)+9.d(1);8=5=7="";f=g=c=1=""}u(i> 2; - enc2 = ((chr1 & 3) << 4) | (chr2 >> 4); - enc3 = ((chr2 & 15) << 2) | (chr3 >> 6); - enc4 = chr3 & 63; - if (isNaN(chr2)) { - enc3 = enc4 = 64 - } else if (isNaN(chr3)) { - enc4 = 64 - } - output = output + keyStr.charAt(enc1) + keyStr.charAt(enc2) + keyStr.charAt(enc3) + keyStr.charAt(enc4); - chr1 = chr2 = chr3 = ""; - enc1 = enc2 = enc3 = enc4 = "" - } while (i < input.length); - return output -} +#个人信息Personal Information +SchoolID='' #引号之间填入学号 +Password_jwgl='' #引号之间填入教务系统密码 +StartDate='' #引号之间填入开始日期，格式为'2023-02-20' + +#教务系统加密算法，由[https://jwgl.bupt.edu.cn/jsxsd/framework/xsMain.jsp]得到的conwork.js +CONWORK_JS = ''' +eval(function (p, a, c, k, e, d) +{ + e = function (c){return (c < a ? "": e(parseInt(c / a))) + ((c = c % a) > 35 ? String.fromCharCode(c + 29) : c.toString(36))}; + if (!''.replace(/^/, String)) + { + while (c--) d[e(c)] = k[c] || e(c); + k = [function (e){return d[e]}]; + e = function() {return '\\w+'}; + c = 1; + }; + while (c--) if (k[c]) p = p.replace(new RegExp('\\b' + e(c) + '\\b', 'g'), k[c]); + return p; +} ('b 9="o+/=";p q(a){b e="";b 8,5,7="";b f,g,c,1="";b i=0;m{8=a.h(i++);5=a.h(i++);7=a.h(i++);f=8>>2;g=((8&3)<<4)|(5>>4);c=((5&s)<<2)|(7>>6);1=7&t;k(j(5)){c=1=l}v k(j(7)){1=l}e=e+9.d(f)+9.d(g)+9.d(c)+9.d(1);8=5=7="";f=g=c=1=""}u(i> 2; + enc2 = ((chr1 & 3) << 4) | (chr2 >> 4); + enc3 = ((chr2 & 15) << 2) | (chr3 >> 6); + enc4 = chr3 & 63; + if (isNaN(chr2)) { + enc3 = enc4 = 64 + } else if (isNaN(chr3)) { + enc4 = 64 + } + output = output + keyStr.charAt(enc1) + keyStr.charAt(enc2) + keyStr.charAt(enc3) + keyStr.charAt(enc4); + chr1 = chr2 = chr3 = ""; + enc1 = enc2 = enc3 = enc4 = "" + } while (i < input.length); + return output +} ''' \ No newline at end of file diff --git a/Main.py b/Main.py index 4fc757d..535798d 100644 --- a/Main.py +++ b/Main.py @@ -1,238 +1,249 @@ -import requests -import icalendar -import re -import datetime -import win32com.client -import openpyxl -import os -from Constant import * - -#是否写入本地html -IsLocalHtml = False - -#用于登录的URL以及UA -USER_AGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36 Edg/114.0.0.0' -LOGIN_URL = 'https://jwgl.bupt.edu.cn/jsxsd/' -GET_API = 'https://jwgl.bupt.edu.cn/jsxsd/xskb/xskb_list.do' -POST_URL = 'https://jwgl.bupt.edu.cn/jsxsd/xk/LoginToXk' -KBJCMSID = { #由[https://jwgl.bupt.edu.cn/jsxsd/framework/xsMain.jsp]的html代码得到的'课表节次模式id' - '0': '9475847A3F3033D1E05377B5030AA94D', #默认节次模式 - '1': '857D7D63586B4AF4B4C59B1DFEEB56E4' #海南校区课表时间 -} -KBLB = { #由[https://jwgl.bupt.edu.cn/jsxsd/framework/xsMain.jsp]的html代码得到的'课表类别' - '0': 'kckb', #课程课表 - '1': 'tzdkb' #通知单课表 -} - -def GetXNXQ(): #获取学年学期，形如'2022-2023-2'，2-7月为第二学期，8-次年1月为第一学期 - CurrentYear = datetime.datetime.now().year - CurrentMonth = datetime.datetime.now().month - if CurrentMonth >= 2 and CurrentMonth <= 7: - XNXQ = str(CurrentYear - 1) + '-' + str(CurrentYear) + '-2' - else: - XNXQ = str(CurrentYear) + '-' + str(CurrentYear + 1) + '-1' - return XNXQ - -def ProcessCK(CK): #处理cookies，将cookies转换为字符串 - Cookie = '' - for Name, Value in CK: - Cookie += f'{Name}={Value};' - return Cookie - -#以下加密算法参考[https://github.com/LAWTED/BUPT-Auto-Syllabu/blob/main/process.py]得到 -KEYSTRING = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=" -def EncodeInp(Input): - Output = '' - Chr1, Chr2, Chr3 = '', '', '' - Enc1, Enc2, Enc3 = '', '', '' - i = 0 - while True: - Chr1 = ord(Input[i]) - i += 1 - Chr2 = ord(Input[i]) if i < len(Input) else 0 - i += 1 - Chr3 = ord(Input[i]) if i < len(Input) else 0 - i += 1 - Enc1 = Chr1 >> 2 - Enc2 = ((Chr1 & 3) << 4) | (Chr2 >> 4) - Enc3 = ((Chr2 & 15) << 2) | (Chr3 >> 6) - Enc4 = Chr3 & 63 - if Chr2 == 0: - Enc3 = Enc4 = 64 - elif Chr3 == 0: - Enc4 = 64 - Output = Output + KEYSTRING[Enc1] + KEYSTRING[Enc2] + KEYSTRING[Enc3] + KEYSTRING[Enc4] - Chr1 = Chr2 = Chr3 = '' - Enc1 = Enc2 = Enc3 = Enc4 = '' - if i >= len(Input): break - return Output - -Encoded = EncodeInp(SchoolID) + "%%%" + EncodeInp(Password_jwgl) #加密后的学号和密码 -Xueqi = GetXNXQ() #获取学年学期，形如'2022-2023-2' -UserData = { #用户数据，保存在Constant.py中，根据需要修改 - 'userAccount': SchoolID, - 'userPassword': Password_jwgl, - 'encoded': Encoded -} - -#登录教务系统 -print('登录北邮教务管理系统') -Session = requests.Session() -Login = Session.get(url=LOGIN_URL, headers={'User-Agent': USER_AGENT}) -Cookies1 = Login.cookies.items() #处理cookies -Cookie = ProcessCK(Cookies1) - -#设置headers，先post用户数据 -Headers_UserData = { - 'Host': 'jwgl.bupt.edu.cn', - 'Referer': 'https://jwgl.bupt.edu.cn/jsxsd/xk/LoginToXk?method=exit&tktime=1631723647000', - 'User-Agent': USER_AGENT, - 'cookie': Cookie -} -Session.post(url=POST_URL, data=UserData, headers=Headers_UserData) #由LOGIN_URL的html决定是post方法，action='/jsxsd/xk/LoginToXk' -Cookies2 = Login.cookies.items() #登录后cookie可能变了，重新处理一下 -Cookie_Logged_In = ProcessCK(Cookies2) - -#设置headers -Headers = {'User-Agent': USER_AGENT, 'cookie': Cookie_Logged_In} -#周次（默认全部），学年学期（默认当前学期），课表节次id，课表类别 -Info = { - 'zc': '', - 'xnxq01id': Xueqi, - 'kbjcmsid': KBJCMSID['0'], - 'kblb': KBLB['0'] -} -Url = 'https://jwgl.bupt.edu.cn/jsxsd/xskb/xskb_list.do' -Data = requests.post(url=Url, headers=Headers, data=Info) #获取课表html，检查是否成功 -if "请先登录系统" in Data.text: - print("登录失败，请检查学号和密码以及网络连接，并重新运行程序") - exit() -else: - print("登录成功，正在处理") - print("已获取到课表信息") -xlsUrl = f'https://jwgl.bupt.edu.cn/jsxsd/xskb/xskb_print.do?xnxq01id={Xueqi}&zc=&kbjcmsid=9475847A3F3033D1E05377B5030AA94D' #获取课表xls文件的url -Excel = requests.post(url=xlsUrl, headers=Headers) #获取课表xls文件 -f = open(f'学生个人课表_{SchoolID}.xls', 'wb') -f.write(Excel.content) #写入本地xls文件 -f.close() - -#可选，将.do写入本地html文件 -if IsLocalHtml: - print("正在写入本地html") - try: - with open(f'./{datetime.datetime.now().strftime("%Y-%m-%d-%H-%M")}.html', 'w', encoding='utf-8') as file: - file.write(Data.text) - print(f'[success]') - except Exception: - print("生成文件失败") - -#将课表xls文件转换为xlsx文件 -Fname = f'{os.getcwd()}/学生个人课表_{SchoolID}.xls' -excel = win32com.client.gencache.EnsureDispatch('Excel.Application') -wb = excel.Workbooks.Open(Fname) -wb.SaveAs(Fname + 'x', FileFormat=51) #FileFormat = 51 is for .xlsx extension -wb.Close() #FileFormat = 56 is for .xls extension -excel.Application.Quit() - -#----------------------------------------------------------------------------------- -#----------------------------------------------------------------------------------- -#----------------------------------------------------------------------------------- -#以下Excel处理代码来自https://github.com/Guest-Liang/BUPT-iOSCalendar-Excel -#定义课程开始时间 -StartTime = [ - datetime.time(8, 0, 0), datetime.time(8, 50, 0), datetime.time(9, 50, 0), - datetime.time(10, 40, 0), datetime.time(11, 30, 0), datetime.time(13, 00, 0), - datetime.time(13, 50, 0), datetime.time(14, 45, 0), datetime.time(15, 40, 0), - datetime.time(16, 35, 0), datetime.time(17, 25, 0), datetime.time(18, 30, 0), - datetime.time(19, 20, 0), datetime.time(20, 10, 0) -] -#定义课程结束时间 -EndTime = [ - datetime.time(8, 45, 0), datetime.time(9, 35, 0), datetime.time(10, 35, 0), - datetime.time(11, 25, 0), datetime.time(12, 15, 0), datetime.time(13, 45, 0), - datetime.time(14, 35, 0), datetime.time(15, 30, 0), datetime.time(16, 25, 0), - datetime.time(17, 20, 0), datetime.time(18, 10, 0), datetime.time(19, 15, 0), - datetime.time(20, 5, 0), datetime.time(20, 55, 0) -] - -#找到字符串中某个关键值的所有索引，存放在list(int)中 -def GetElementIndex(char, string): - return [idx.start() for idx in re.finditer(char, string)] - -#将上课周数转为list(int) -def ChangeIntoList_int(s): - ranges = re.findall(r'(\d+)-(\d+)', s) - for start, end in ranges: - s = s.replace(f'{start}-{end}', ','.join(map(str, range(int(start), int(end) + 1)))) - return list(map(int, s.split(','))) - -#获取学号，打开xlsx文件 -userid = SchoolID -WorkBook = openpyxl.load_workbook(filename=f"./学生个人课表_{userid}.xlsx") -Sheet = WorkBook.active - -print("-------------------------") -print("您的信息为：") -print("课程表所属人：", end="") -StudentName = Sheet['A1'].value.replace("北京邮电大学 ", "").replace(" 学生个人课表", "") -print(StudentName) -print("学年：", end="") -SchoolYear = Sheet['A2'].value[5:16] -print(SchoolYear) - -#输入学期的第一周的周一日期 -#StartDate=datetime.date(2023, 2, 20) -# StartDate = datetime.datetime.strptime(input("输入学期的第一周的周一的日期，以YYYY-MM-DD格式\n"), '%Y-%m-%d').date() -# while StartDate.isoweekday() != 1: -# StartDate = datetime.datetime.strptime(input("日期并非周一！请以YYYY-MM-DD格式输入\n"), '%Y-%m-%d').date() -StartDate=datetime.datetime.strptime(f'{StartDate}', '%Y-%m-%d').date() -print("正在处理") - -#制作部分 -MyCalendar = icalendar.Calendar() -MyCalendar.add('PRODID', '-//MY_CALENDAR_PRODUCT//GL//') -MyCalendar.add('VERSION', '2.0') #固定属性，版本2.0 -MyCalendar.add('CALSCALE', 'GREGORIAN') #公历 -MyCalendar.add('METHOD', 'PUBLISH') -MyCalendar.add('X-WR-CALNAME', f'{SchoolYear}') #通用属性，日历名称，默认为学年 -MyCalendar.add('X-WR-TIMEZONE', 'Asia/Shanghai') #通用属性，指定时区 -MyCalendar.add('X-APPLE-CALENDAR-COLOR', '#E1FFFF') #私有属性，指定Apple日历颜色，可自己更改，填入十六进制代码 -for Column in range(2, 9): - for Row in range(4, 18): - CellBR = GetElementIndex("\n", Sheet.cell(row=Row, column=Column).value) #根据换行符位置进行拆分 - for i in range(int(len(CellBR) / 5)): #拆分 课程、教师名字、上课周数、上课教室、上课节次 - Course = Sheet.cell(Row, Column).value[CellBR[5 * i] + 1:CellBR[5 * i + 1]] - TeacherName = Sheet.cell(Row, Column).value[CellBR[5 * i + 1] + 1:CellBR[5 * i + 2]] - ClassWeeks = Sheet.cell(Row, Column).value[CellBR[5 * i + 2] + 1:CellBR[5 * i + 3]] - Classroom = Sheet.cell(Row, Column).value[CellBR[5 * i + 3] + 1:CellBR[5 * i + 4]] - if i == int(len(CellBR) / 5) - 1: - LessonNum = Sheet.cell(Row, Column).value[CellBR[5 * i + 4] + 1:] - else: - LessonNum = Sheet.cell(Row, Column).value[CellBR[5 * i + 4] + 1:CellBR[5 * i + 5]] - ListLessonNum = LessonNum.replace("[", "").replace("]", "").replace("节", "").split("-") - ListLessonNum = list(map(int, ListLessonNum)) - if (Row - 3 == ListLessonNum[0]): #是第一节课才添加，下一节跳过（意思是默认合并课程，避免导致相邻、连着的课程重复提醒） - ListClassWeeks = ChangeIntoList_int(ClassWeeks.replace("[周]", "")) - for j in range(len(ListClassWeeks)): - MyEvent = icalendar.Event() - MyEvent.add('SUMMARY', Course + ' ' + Classroom) #事件名称：课程名加教室 - MyEvent.add('DTSTAMP', datetime.datetime.today()) - MyEvent.add('DTSTART', datetime.datetime.combine(StartDate + datetime.timedelta(weeks=ListClassWeeks[j] - 1, days=Column - 2), StartTime[ListLessonNum[0] - 1])) - MyEvent.add('DTEND', datetime.datetime.combine(StartDate + datetime.timedelta(weeks=ListClassWeeks[j] - 1, days=Column - 2), EndTime[ListLessonNum[-1] - 1])) - MyEvent.add('DESCRIPTION', TeacherName) #教师姓名写在备注里 - MyAlarm = icalendar.Alarm() #添加提醒作为事件的附加属性 - MyAlarm.add('TRIGGER', datetime.timedelta(minutes=-10)) #提前10分钟提醒 - MyAlarm.add('ACTION', "DISPLAY") #通知提醒 - MyAlarm.add('DESCRIPTION', Course) #提醒内容：课程名称 - MyEvent.add_component(MyAlarm) - MyCalendar.add_component(MyEvent) - del MyAlarm, MyEvent - del TeacherName, ClassWeeks, Classroom, LessonNum, Course, ListClassWeeks - del CellBR -try: - with open(f'TimeTable_{StudentName}.ics', 'wb') as file: - file.write(MyCalendar.to_ical()) - print('[Success]') - del MyCalendar -except Exception: +import requests +import icalendar +import re +import datetime +import win32com.client +import openpyxl +import os +from Constant import * + +#是否写入本地html +IsLocalHtml = False + +#----------------------------------------------------------------------------------- +JWIP = ["http://10.3.58.10:18080/jsxsd", "http://10.3.58.11:18081/jsxsd", + "http://10.3.58.12:18082/jsxsd", "http://10.3.58.13:18083/jsxsd", + "http://10.3.58.14:18084/jsxsd", "http://10.3.58.15:18085/jsxsd", + "http://10.3.58.16:18086/jsxsd", "http://10.3.58.17:18087/jsxsd"] #仅10可用，11-17不可用 + +#用于登录的URL以及UA +USER_AGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36 Edg/114.0.0.0' +LOGIN_URL = JWIP[0] +GET_API = f'{LOGIN_URL}/xskb/xskb_list.do' +POST_URL = f'{LOGIN_URL}/xk/LoginToXk' +KBJCMSID = { #由[https://jwgl.bupt.edu.cn/jsxsd/framework/xsMain.jsp]的html代码得到的'课表节次模式id' + '0': '9475847A3F3033D1E05377B5030AA94D', #默认节次模式 + '1': '857D7D63586B4AF4B4C59B1DFEEB56E4' #海南校区课表时间 +} +KBLB = { #由[https://jwgl.bupt.edu.cn/jsxsd/framework/xsMain.jsp]的html代码得到的'课表类别' + '0': 'kckb', #课程课表 + '1': 'tzdkb' #通知单课表 +} + +def GetXNXQ(): #获取学年学期，形如'2022-2023-2'，2-7月为第二学期，8-次年1月为第一学期 + CurrentYear = datetime.datetime.now().year + CurrentMonth = datetime.datetime.now().month + if CurrentMonth >= 2 and CurrentMonth <= 7: + XNXQ = str(CurrentYear - 1) + '-' + str(CurrentYear) + '-2' + else: + XNXQ = str(CurrentYear) + '-' + str(CurrentYear + 1) + '-1' + return XNXQ + +def ProcessCK(CK): #处理cookies，将cookies转换为字符串 + Cookie = '' + for Name, Value in CK: + Cookie += f'{Name}={Value};' + return Cookie + +#以下加密算法参考[https://github.com/LAWTED/BUPT-Auto-Syllabu/blob/main/process.py]得到 +KEYSTRING = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=" +def EncodeInp(Input): + Output = '' + Chr1, Chr2, Chr3 = '', '', '' + Enc1, Enc2, Enc3 = '', '', '' + i = 0 + while True: + Chr1 = ord(Input[i]) + i += 1 + Chr2 = ord(Input[i]) if i < len(Input) else 0 + i += 1 + Chr3 = ord(Input[i]) if i < len(Input) else 0 + i += 1 + Enc1 = Chr1 >> 2 + Enc2 = ((Chr1 & 3) << 4) | (Chr2 >> 4) + Enc3 = ((Chr2 & 15) << 2) | (Chr3 >> 6) + Enc4 = Chr3 & 63 + if Chr2 == 0: + Enc3 = Enc4 = 64 + elif Chr3 == 0: + Enc4 = 64 + Output = Output + KEYSTRING[Enc1] + KEYSTRING[Enc2] + KEYSTRING[Enc3] + KEYSTRING[Enc4] + Chr1 = Chr2 = Chr3 = '' + Enc1 = Enc2 = Enc3 = Enc4 = '' + if i >= len(Input): break + return Output + +if SchoolID == '' or Password_jwgl == '': #检查学号和密码是否为空 + SchoolID = input("请输入学号\n") + Password_jwgl = input('请输入教务系统的密码\n') +Encoded = EncodeInp(SchoolID) + "%%%" + EncodeInp(Password_jwgl) #加密后的学号和密码 +Xueqi = GetXNXQ() #获取学年学期，形如'2022-2023-2' +UserData = { #用户数据，保存在Constant.py中，根据需要修改 + 'userAccount': SchoolID, + 'userPassword': Password_jwgl, + 'encoded': Encoded +} + +#登录教务系统 +print('登录北邮教务管理系统') +Session = requests.Session() +Login = Session.get(url=LOGIN_URL, headers={'User-Agent': USER_AGENT},verify=False) +Cookies1 = Login.cookies.items() #处理cookies +Cookie = ProcessCK(Cookies1) + +#设置headers，先post用户数据 +Headers_UserData = { + # 'Host': 'jwgl.bupt.edu.cn', + 'Host': '10.3.58.10:18080', + 'Referer': f'{LOGIN_URL}/xk/LoginToXk?method=exit&tktime=1631723647000', + 'User-Agent': USER_AGENT, + 'cookie': Cookie +} +Session.post(url=POST_URL, data=UserData, headers=Headers_UserData,verify=False) #由LOGIN_URL的html决定是post方法，action='/jsxsd/xk/LoginToXk' +Cookies2 = Login.cookies.items() #登录后cookie可能变了，重新处理一下 +Cookie_Logged_In = ProcessCK(Cookies2) + +#设置headers +Headers = {'User-Agent': USER_AGENT, 'cookie': Cookie_Logged_In} +#周次（默认全部），学年学期（默认当前学期），课表节次id，课表类别 +Info = { + 'zc': '', + 'xnxq01id': Xueqi, + 'kbjcmsid': KBJCMSID['0'], + 'kblb': KBLB['0'] +} +Url = f'{LOGIN_URL}/xskb/xskb_list.do' +Data = requests.post(url=Url, headers=Headers, data=Info) #获取课表html，检查是否成功 +if "请先登录系统" in Data.text: + print("登录失败，请检查学号和密码以及网络连接，并重新运行程序") + exit() +else: + print("登录成功，正在处理") + print("已获取到课表信息") +xlsUrl = f'{LOGIN_URL}/xskb/xskb_print.do?xnxq01id={Xueqi}&zc=&kbjcmsid=9475847A3F3033D1E05377B5030AA94D' #获取课表xls文件的url +Excel = requests.post(url=xlsUrl, headers=Headers) #获取课表xls文件 +f = open(f'学生个人课表_{SchoolID}.xls', 'wb') +f.write(Excel.content) #写入本地xls文件 +f.close() + +#可选，将.do写入本地html文件 +if IsLocalHtml: + print("正在写入本地html") + try: + with open(f'./{datetime.datetime.now().strftime("%Y-%m-%d-%H-%M")}.html', 'w', encoding='utf-8') as file: + file.write(Data.text) + print(f'[success]') + except Exception: + print("生成文件失败") + +#将课表xls文件转换为xlsx文件 +Fname = f'{os.getcwd()}/学生个人课表_{SchoolID}.xls' +excel = win32com.client.Dispatch('Excel.Application') +wb = excel.Workbooks.Open(Fname) +wb.SaveAs(Fname + 'x', FileFormat=51) #FileFormat = 51 is for .xlsx extension +wb.Close() #FileFormat = 56 is for .xls extension +excel.Application.Quit() + +#----------------------------------------------------------------------------------- +#----------------------------------------------------------------------------------- +#----------------------------------------------------------------------------------- +#以下Excel处理代码来自https://github.com/Guest-Liang/BUPT-iOSCalendar-Excel +#定义课程开始时间 +StartTime = [ + datetime.time(8, 0, 0), datetime.time(8, 50, 0), datetime.time(9, 50, 0), + datetime.time(10, 40, 0), datetime.time(11, 30, 0), datetime.time(13, 00, 0), + datetime.time(13, 50, 0), datetime.time(14, 45, 0), datetime.time(15, 40, 0), + datetime.time(16, 35, 0), datetime.time(17, 25, 0), datetime.time(18, 30, 0), + datetime.time(19, 20, 0), datetime.time(20, 10, 0) +] +#定义课程结束时间 +EndTime = [ + datetime.time(8, 45, 0), datetime.time(9, 35, 0), datetime.time(10, 35, 0), + datetime.time(11, 25, 0), datetime.time(12, 15, 0), datetime.time(13, 45, 0), + datetime.time(14, 35, 0), datetime.time(15, 30, 0), datetime.time(16, 25, 0), + datetime.time(17, 20, 0), datetime.time(18, 10, 0), datetime.time(19, 15, 0), + datetime.time(20, 5, 0), datetime.time(20, 55, 0) +] + +#找到字符串中某个关键值的所有索引，存放在list(int)中 +def GetElementIndex(char, string): + return [idx.start() for idx in re.finditer(char, string)] + +#将上课周数转为list(int) +def ChangeIntoList_int(s): + ranges = re.findall(r'(\d+)-(\d+)', s) + for start, end in ranges: + s = s.replace(f'{start}-{end}', ','.join(map(str, range(int(start), int(end) + 1)))) + return list(map(int, s.split(','))) + +#获取学号，打开xlsx文件 +userid = SchoolID +WorkBook = openpyxl.load_workbook(filename=f"./学生个人课表_{userid}.xlsx") +Sheet = WorkBook.active + +print("-------------------------") +print("您的信息为：") +print("课程表所属人：", end="") +StudentName = Sheet['A1'].value.replace("北京邮电大学 ", "").replace(" 学生个人课表", "") +print(StudentName) +print("学年：", end="") +SchoolYear = Sheet['A2'].value[5:16] +print(SchoolYear) + +#输入学期的第一周的周一日期 +if (StartDate==''): + StartDate = datetime.datetime.strptime(input("输入学期的第一周的周一的日期，以YYYY-MM-DD格式\n"), '%Y-%m-%d').date() + while StartDate.isoweekday() != 1: + StartDate = datetime.datetime.strptime(input("日期并非周一！请以YYYY-MM-DD格式输入\n"), '%Y-%m-%d').date() +else: + StartDate=datetime.datetime.strptime(f'{StartDate}', '%Y-%m-%d').date() +print("正在处理") + +#制作部分 +MyCalendar = icalendar.Calendar() +MyCalendar.add('PRODID', '-//MY_CALENDAR_PRODUCT//GL//') +MyCalendar.add('VERSION', '2.0') #固定属性，版本2.0 +MyCalendar.add('CALSCALE', 'GREGORIAN') #公历 +MyCalendar.add('METHOD', 'PUBLISH') +MyCalendar.add('X-WR-CALNAME', f'{SchoolYear}') #通用属性，日历名称，默认为学年 +MyCalendar.add('X-WR-TIMEZONE', 'Asia/Shanghai') #通用属性，指定时区 +MyCalendar.add('X-APPLE-CALENDAR-COLOR', '#E1FFFF') #私有属性，指定Apple日历颜色，可自己更改，填入十六进制代码 +for Column in range(2, 9): + for Row in range(4, 18): + CellBR = GetElementIndex("\n", Sheet.cell(row=Row, column=Column).value) #根据换行符位置进行拆分 + for i in range(int(len(CellBR) / 5)): #拆分 课程、教师名字、上课周数、上课教室、上课节次 + Course = Sheet.cell(Row, Column).value[CellBR[5 * i] + 1:CellBR[5 * i + 1]] + TeacherName = Sheet.cell(Row, Column).value[CellBR[5 * i + 1] + 1:CellBR[5 * i + 2]] + ClassWeeks = Sheet.cell(Row, Column).value[CellBR[5 * i + 2] + 1:CellBR[5 * i + 3]] + Classroom = Sheet.cell(Row, Column).value[CellBR[5 * i + 3] + 1:CellBR[5 * i + 4]] + if i == int(len(CellBR) / 5) - 1: + LessonNum = Sheet.cell(Row, Column).value[CellBR[5 * i + 4] + 1:] + else: + LessonNum = Sheet.cell(Row, Column).value[CellBR[5 * i + 4] + 1:CellBR[5 * i + 5]] + ListLessonNum = LessonNum.replace("[", "").replace("]", "").replace("节", "").split("-") + ListLessonNum = list(map(int, ListLessonNum)) + if (Row - 3 == ListLessonNum[0]): #是第一节课才添加，下一节跳过（意思是默认合并课程，避免导致相邻、连着的课程重复提醒） + ListClassWeeks = ChangeIntoList_int(ClassWeeks.replace("[周]", "")) + for j in range(len(ListClassWeeks)): + MyEvent = icalendar.Event() + MyEvent.add('SUMMARY', Course + ' ' + Classroom) #事件名称：课程名加教室 + MyEvent.add('DTSTAMP', datetime.datetime.today()) + MyEvent.add('DTSTART', datetime.datetime.combine(StartDate + datetime.timedelta(weeks=ListClassWeeks[j] - 1, days=Column - 2), StartTime[ListLessonNum[0] - 1])) + MyEvent.add('DTEND', datetime.datetime.combine(StartDate + datetime.timedelta(weeks=ListClassWeeks[j] - 1, days=Column - 2), EndTime[ListLessonNum[-1] - 1])) + MyEvent.add('DESCRIPTION', TeacherName) #教师姓名写在备注里 + MyAlarm = icalendar.Alarm() #添加提醒作为事件的附加属性 + MyAlarm.add('TRIGGER', datetime.timedelta(minutes=-10)) #提前10分钟提醒 + MyAlarm.add('ACTION', "DISPLAY") #通知提醒 + MyAlarm.add('DESCRIPTION', Course) #提醒内容：课程名称 + MyEvent.add_component(MyAlarm) + MyCalendar.add_component(MyEvent) + del MyAlarm, MyEvent + del TeacherName, ClassWeeks, Classroom, LessonNum, Course, ListClassWeeks + del CellBR +try: + with open(f'TimeTable_{StudentName}_{Xueqi}.ics', 'wb') as file: + file.write(MyCalendar.to_ical()) + print('[Success]') + del MyCalendar +except Exception: print("生成文件失败，请重试") \ No newline at end of file