import re
from time import sleep

import requests
from bs4 import BeautifulSoup
from datetime import datetime, timedelta
import time

#根据周和星期获取日期的类
class WeekDateCalculator:
    def __init__(self, start_year, start_month, start_day):
        """
        初始化类，设置第一周的星期日的日期
        :param start_year: 年份
        :param start_month: 月份
        :param start_day: 日期
        """
        self.start_date = datetime(start_year, start_month, start_day)
        # 确保start_date是星期日
        while self.start_date.weekday() != 6:  # 星期日对应的weekday()值是6
            self.start_date += timedelta(days=1)

    def get_date_by_week_and_day(self, week_number, day_of_week):
        """
        根据周数和星期几获取日期
        :param week_number: 周数（从第一周开始）
        :param day_of_week: 星期几（0代表星期一，1代表星期二，...，6代表星期日）
        :return: 对应日期的datetime对象
        """
        # 计算从第一周星期日到目标星期几的天数偏移
        target_day_offset = (day_of_week - self.start_date.weekday() + 7) % 7
        # 计算目标日期与第一周星期日之间的总天数
        total_days = (week_number - 1) * 7 + target_day_offset
        # 计算目标日期
        target_date = self.start_date + timedelta(days=total_days)
        return target_date

#录课系统登录类
class dl :
    #登录页面
    #判断是否登录成功，成功则返回会话对象
    def cdym__ff(self):
        while 1:
            print("欢迎使用自动录课系统")
            cookie2 = input("请输入录课会话秘钥:")
            B =self.lk(cookie2)
            if"state"in B.text:
                print("登录成功")
                break
            else:
                print("登录失败,请检查秘钥")

        while 1:
            cookie =input("请输入登录会话秘钥:")
            A = dl.dl_ff(cookie)
            if "用户没有登录，请重新登录" in A.text:
                print("登录失败")
            elif"您登录后过长时间没有操作或您的用户名已经在别处登录！"in A.text:
                print("您登录后过长时间没有操作或您的用户名已经在别处登录,请重新获取cookie")
            elif "教室" in A.text:
                print("登录成功")
                break

        return cookie,cookie2

        # 录课模块，用于被录课方法引用
        # 录课数据
    def lk(self,cookie2):
        data = {
            'room_id': 1,  # 教室id     1需要
            'room_name': "1-301",  # 教室名称         1需要
            'year': '2024',  # 年
            'term': '1',  # 学期
            'term_start': '2024-09-02',  # 学期开头日期
            'week_no': 20,  # 周                  需要
            'weekday': 1,  # 星期                  需要
            'section': 1,  # 从哪小节进入录课界面      1需要
            'start_section': 1,  # 录课的开始节       1需要
            'end_section': 2,  # 录课的结束节          2需要
            'people_id': '0001130071',  # 老师id
            'audience': '借用',  # 两个填的地方
            'reason': '借用'
        }
        url = "录课平台URL，防止泄露，故不展示，505电脑中有此记载"
        A = requests.post(url, data=data, headers={"cookie": f"{cookie2}"})#获取数据
        return A


    # 返回登录对象
    def dl_ff(self,cookie):
        data2 = {
            "gnq_mh": "",
            "jsmc_mh": "",
            "typewhere": "jwyx",
            "xnxqh": "2024-2025-1",  # 学期
            "xqbh": "2",  # 校区，白云
            "jxqbh": "CBD7DE758E094061AE21E96B1B32CC56",  # 教学区：多媒体
            "jslx": "",  # 教室类型
            "jxlbh": "",  # 教学楼
            "jsbh": "",  # 教室
            "bjfh": "=",  # 人数
            "rnrs": "",  # 人数空白
            "jszt": "2",  # 借用，1是正常上课，2是借用，3是锁定，4是考试，5是空闲....
            "yx": "",  # 借用院系
            "zc": 1,  # 周次1         1需要
            "zc2": 1,  # 周次2        1需要
            "xq": 1,  # 星期1          1需要
            "xq2": 1,  # 星期2         1需要
            "jc": "01",  # 节次1
            "jc2": "11",  # 节次2
            "kbjcmsid": "29637FA5710B46F3B01242A09D7E9CCD",  # 时间模式，白云
            "qsrq": "2024-09-01",  # 日期        1需要
            "jsrq": "2024-09-01",  # 日期        1需要
            "ssdw": ""  # 教室所属单位
        }
        url = ("教务网对应URL，防止泄露，故不展示，505电脑中有此记载")
        A = requests.post(url, data=data2, headers={"cookie": f"{cookie}"})
        return A

    #获取 周次 星期 查询节次1 查询节次2 日期
    def sj_hq(self):
        zc_ =input("周次:")
        xq_=input("星期:")
        jc1_=1#input("查询节次1:")
        jc2_=11#input("查询节次2:")
        #设置开学日期
        calculator = WeekDateCalculator(2024, 9, 1)  # 根据实际情况设置年份、月份和日期
        #设置周次和星期
        week_number = int(zc_)
        day_of_week = int(xq_)-1
        #获取日期
        result_date = calculator.get_date_by_week_and_day(week_number, day_of_week)
        rq_ =result_date.strftime('%Y-%m-%d')

        #返回 周次 星期 查询节次1 查询节次2 日期
        return zc_,xq_,jc1_,jc2_,rq_

#自动录课系统
class luke:
    # 定义各种参数
    zc = 0  # 周次
    xq = 0  # 星期
    jc1 = 1  # 查询节次1
    jc2 = 2  # 查询节次2
    rq = "2024-12-09"  # 日期
    # #教室id
    id = []
    # 教室名称
    js_name = ""
    cookie =""
    cookie2 = ""

    # 初始化查询的
    def __init__(self, zc_, xq_, jc1_, jc2_, rq_,cookie_1,cookie_2):
        self.zc = zc_  # 周次
        self.xq = xq_  # 星期
        self.jc1 = jc1_  # 查询节次1
        self.jc2 = jc2_ # 查询节次2
        self.rq = rq_  # 日期
        self.cookie =cookie_1 #登录秘钥
        self.cookie2 = cookie_2  # 录课秘钥

    #查询数据,获取html，返回html
    def cx(self):
        data2 ={
        "gnq_mh":"" ,
        "jsmc_mh":"",
        "typewhere": "jwyx",
        "xnxqh": "2024-2025-1",#学期
        "xqbh": "2",#校区，白云
        "jxqbh": "CBD7DE758E094061AE21E96B1B32CC56",#教学区：多媒体
        "jslx":"",#教室类型
        "jxlbh":"",#教学楼
        "jsbh":"",#教室
        "bjfh":"=",#人数
        "rnrs":"",#人数空白
        "jszt":"2",#借用，1是正常上课，2是借用，3是锁定，4是考试，5是空闲....
        "yx":"",#借用院系
        "zc": self.zc,#周次1         1需要
        "zc2": self.zc,#周次2        1需要
        "xq": self.xq,#星期1          1需要
        "xq2": self.xq,#星期2         1需要
        "jc": self.jc1,#节次1
        "jc2": self.jc2,#节次2
        "kbjcmsid": "29637FA5710B46F3B01242A09D7E9CCD",#时间模式，白云
        "qsrq": self.rq,#日期        1需要
        "jsrq": self.rq,#日期        1需要
        "ssdw":""#教室所属单位
        }
        url =("教务网对应URL，防止泄露，故不展示，505电脑中有此记载")
        A =requests.post(url,data= data2, headers={"cookie":f"{self.cookie}"})
        return A

#获取借用清单
    def qd(self,A):
        B = BeautifulSoup(A.text,"html.parser")
        C=B.find("table",attrs={"id":"kbtable"})#获取总表格
        D =C.find("tbody")#获取没有标题的表格
        C =D.find_all("tr")#获取每间教室

        # #总借用清单
        jyqd=[]
        #获取总借用清单，格式[[['1-302'], 0, 1, 0, 0, 1, 1], [['1-406'], 0, 0, 0, 1, 1, 0]]
        #0表示没有借用，1表示借用
        for i in C:
            #大节
            sj =0
            #获取查询大节次数
            xj =int(self.jc2)-int(self.jc2)

            # 教室借用清单
            jyqd2 = []
            for a in i.find_all_next("td"):#一个教室的每个时间段
                if sj==0:#获取教室名称
                    A=re.findall("\d-\d{3}",a.text)
                    #print(sj,A)
                    jyqd2.append(A)
                    sj=sj+1
                elif sj<7:#获取1-6大节借用情况
                    A = re.findall("Ｊ", a.text)#查看
                    #print(sj, A)
                    if A ==["Ｊ"]:
                        jyqd2.append(1)
                    else:
                        jyqd2.append(0)
                    sj = sj + 1

            jyqd.append(jyqd2)
            # 循环输出借用每个教室清单
            #print(jyqd2)
        #输出借用总清单
        #print(jyqd)
        #教室数量
        #print(len(jyqd))
        #获取教室名称
        T=[]
        for i in range(len(jyqd)):
            T.append(jyqd[i][0])
        return jyqd,len(T),T

#录用开始结束小节换算
    def hs (self,a):
        if a==1:
            return 1,2
        elif a==2:
            return  3,4
        elif a==3:
            return  5,6
        elif a==4:
            return  7,8
        elif a==5:
            return  9,10
        elif a==6:
            return  11,11
#录课方法
    def luk(self,QD,SL,MC):#借用总清单，教室数量，教室名称
        #教室名称对应的id字典
        js_id ={'1-301': '1', '1-302': '2', '1-303': '3', '1-305': '4', '1-311': '5', '1-312': '6', '1-313': '7', '1-314': '8', '1-315': '9', '1-405': '10', '1-406': '11', '1-407': '12', '1-408': '13', '1-409': '14', '1-410': '15', '1-411': '16', '1-412': '17', '1-413': '18', '1-414': '19', '1-415': '20', '1-416': '21', '1-417': '22', '1-418': '23', '1-419': '24', '1-420': '25', '1-506': '26', '1-507': '27', '1-508': '28', '1-509': '29', '1-510': '30', '1-511': '31', '1-512': '32', '1-513': '33', '1-514': '34', '1-515': '35', '1-516': '36', '1-517': '37', '1-518': '38', '1-519': '39', '1-520': '40', '1-602': '41', '1-603': '42', '1-604': '43', '1-605': '44', '1-606': '45', '1-607': '46', '1-608': '47', '1-609': '48', '1-610': '49', '1-611': '50', '1-612': '51', '1-613': '52', '1-614': '53', '1-615': '54', '1-616': '55', '1-617': '56', '1-618': '57', '1-619': '58', '1-703': '59', '1-704': '60', '1-705': '61', '1-706': '62', '1-707': '63', '1-708': '64', '1-709': '65', '1-710': '66', '1-711': '67', '1-713': '68', '1-714': '69', '1-715': '70', '1-716': '71', '1-717': '72', '1-718': '73', '1-719': '74', '1-720': '75', '1-721': '76', '1-722': '77', '2-101': '78', '2-102': '79', '2-103': '80', '2-104': '81', '2-201': '82', '2-202': '83', '2-203': '84', '2-204': '85','2-301': '86', '2-302': '87', '2-303': '88', '2-304': '89', '2-305': '90', '2-306': '91', '2-401': '92', '2-402': '93', '2-403': '94', '2-404': '95', '2-501': '96', '2-502': '97', '2-503': '98', '2-504': '99', '2-601': '100', '2-602': '101', '2-603': '102', '2-604': '103', '2-701': '104', '2-702': '105', '2-703': '106', '2-704': '107'}
        #获取单间教室名称以及id，并录课
        id_=0
        for i in range(SL):
            A =MC[i]
            C=str(A).strip("['")
            C=C.strip("']")#教室名称
            self.id.append(js_id[C])#教室名称对应id

        for i in QD:  # 获取每间教室借用信息
            for a in range(1, 7):  # 检查教室哪节有借用，有则录入
                if i[a] == 0:
                    # print(f"第{a}大节没有借用")
                    pass
                elif i[a] == 1:
                    ks, js = self.hs(a)  # 获取开始结束小节
                    #print(f"{i[0]}第{ks,js}小节有借用", self.id[id_])
                    print(self.id[id_],i[0],self.zc,self.xq,ks,js)
                    self.lk(self.id[id_],i[0],self.zc,self.xq,ks,js)

            id_ = id_ + 1
            sleep(0.1)

#录课模块，用于被录课方法引用
        # 录课数据
    def lk(self,id,name,zc,xq,ks,js):
        data ={
        'room_id': id,#教室id     1需要
        'room_name':name,#教室名称         1需要
        'year': '2024',#年
        'term': '1',#学期
        'term_start': '2024-09-02',#学期开头日期
        'week_no': zc,#周                  需要
        'weekday': xq,#星期                  需要
        'section': ks,#从哪小节进入录课界面      1需要
        'start_section': ks,#录课的开始节       1需要
        'end_section': js,#录课的结束节          2需要
        'people_id': '0001130071',#老师id
        'audience': '借用',#两个填的地方
        'reason': '借用'
        }
        url = "录课平台对应URL，防止泄露，故不展示，505电脑中有此记载"
        A =requests.post(url,data=data,headers={"cookie":f"{self.cookie2}"})
        if "state" in A.text:
            print("入录成功")



if __name__=="__main__":
    #实例登录对象
    dl =dl()
    #获取登录成功的cookie
    cookie,cookie2 =dl.cdym__ff()
    QD=[]
    SL =0
    MC =[]
    #菜单
    while True:
        print(" 欢迎使用自动录课系统")
        print("*******************")
        print("*———————功能———————*")
        print("*1.查询借用清单      *")
        print("*2.录入查询的清单    *")
        print("*3.退出            *")
        print("*******************")
        a =input()
        if a == "1":
            # 获取想查询的数据
            zc, xq, jc1, jc2, rq = dl.sj_hq()
            print(zc, xq, jc1, jc2, rq)
            # 实例一个自动录课系统类的对象
            luke = luke(zc, xq, jc1, jc2, rq, cookie, cookie2)
            A = luke.cx()#获取查询到的html
            QD, SL, MC = luke.qd(A)  # 返回借用清单，借用数量，借用教室名称
            print(f"借用教室数量：{SL}")
            for a in QD:
                print(a)
            #print(f"有借用的教室名称{MC}")

        elif a=="2":
            luke.luk(QD, SL, MC)
        elif a == "3":
            break
        else:
            print("请输入对应功能的数字，如：1")

import re
from time import sleep
import requests
from bs4 import BeautifulSoup
from datetime import datetime, timedelta

#根据周和星期获取日期的类
class WeekDateCalculator:
    def __init__(self, start_year, start_month, start_day):
        """
        初始化类，设置第一周的星期日的日期
        :param start_year: 年份
        :param start_month: 月份
        :param start_day: 日期
        """
        self.start_date = datetime(start_year, start_month, start_day)
        # 确保start_date是星期日
        while self.start_date.weekday() != 6:  # 星期日对应的weekday()值是6
            self.start_date += timedelta(days=1)

    def get_date_by_week_and_day(self, week_number, day_of_week):
        """
        根据周数和星期几获取日期
        :param week_number: 周数（从第一周开始）
        :param day_of_week: 星期几（0代表星期一，1代表星期二，...，6代表星期日）
        :return: 对应日期的datetime对象
        """
        # 计算从第一周星期日到目标星期几的天数偏移
        target_day_offset = (day_of_week - self.start_date.weekday() + 7) % 7
        # 计算目标日期与第一周星期日之间的总天数
        total_days = (week_number - 1) * 7 + target_day_offset
        # 计算目标日期
        target_date = self.start_date + timedelta(days=total_days)
        return target_date

class dl :
    cookie2 =None
    cookie = None

def cdym__ff(self,jw_cookie,luke_cookie):
    while 1:
        self.cookie2 = luke_cookie
        B =self.lk(self.cookie2)
        if"state"in B.text:
            print("登录成功1")
            break
        else:
            print("登录失败,请检查秘钥")
            return 1,0

    while 1:
        self.cookie =jw_cookie
        A = self.dl_ff(self.cookie)
        if "用户没有登录，请重新登录" in A.text:
            print("登录失败")
            return 2,0
        elif"您登录后过长时间没有操作或您的用户名已经在别处登录！"in A.text:
            print("您登录后过长时间没有操作或您的用户名已经在别处登录,请重新获取cookie")
            return 3,0
        elif "教室" in A.text:
            print("登录成功2")
            break

    return self.cookie,self.cookie2

def lk(self,cookie2):
    data = {
        'room_id': 1,  # 教室id     1需要
        'room_name': "1-301",  # 教室名称         1需要
        'year': '2024',  # 年
        'term': '1',  # 学期
        'term_start': '2024-09-02',  # 学期开头日期
        'week_no': 20,  # 周                  需要
        'weekday': 1,  # 星期                  需要
        'section': 1,  # 从哪小节进入录课界面      1需要
        'start_section': 1,  # 录课的开始节       1需要
        'end_section': 2,  # 录课的结束节          2需要
        'people_id': '0001130071',  # 老师id
        'audience': '借用',  # 两个填的地方
        'reason': '借用'
    }
    url = "录课平台URL，防止泄露，故不展示，505电脑中有此记载"
    A = requests.post(url, data=data, headers={"cookie": f"{cookie2}"})
    return A

def dl_ff(self,cookie):
    data2 = {
        "gnq_mh": "",
        "jsmc_mh": "",
        "typewhere": "jwyx",
        "xnxqh": "2024-2025-1",  # 学期
        "xqbh": "2",  # 校区，白云
        "jxqbh": "CBD7DE758E094061AE21E96B1B32CC56",  # 教学区：多媒体
        "jslx": "",  # 教室类型
        "jxlbh": "",  # 教学楼
        "jsbh": "",  # 教室
        "bjfh": "=",  # 人数
        "rnrs": "",  # 人数空白
        "jszt": "2",  # 借用，1是正常上课，2是借用，3是锁定，4是考试，5是空闲....
        "yx": "",  # 借用院系
        "zc": 1,  # 周次1         
        "zc2": 1,  # 周次2        
        "xq": 1,  # 星期1          
        "xq2": 1,  # 星期2         
        "jc": "01",  # 节次1
        "jc2": "11",  # 节次2
        "kbjcmsid": "29637FA5710B46F3B01242A09D7E9CCD",  # 时间模式，白云
        "qsrq": "2024-09-01",  # 日期        
        "jsrq": "2024-09-01",  # 日期        
        "ssdw": ""  # 教室所属单位
    }
    url = ("教务网URL，防止泄露，故不展示，505电脑中有此记载")
    A = requests.post(url, data=data2, headers={"cookie": f"{cookie}"})
    return A

def sj_hq(self,zc,xq):
    zc_ =zc
    xq_=xq
    jc1_=1#input("查询节次1:")
    jc2_=11#input("查询节次2:")
    #设置开学日期
    calculator = WeekDateCalculator(2024, 9, 1)  # 根据实际情况设置年份、月份和日期
    #设置周次和星期
    week_number = int(zc_)
    day_of_week = int(xq_)-1
    #获取日期
    result_date = calculator.get_date_by_week_and_day(week_number, day_of_week)
    rq_ =result_date.strftime('%Y-%m-%d')

    #返回 周次 星期 查询节次1 查询节次2 日期
    return zc_,xq_,jc1_,jc2_,rq_

class luke:
    # 定义各种参数
    zc = 0  # 周次
    xq = 0  # 星期
    jc1 = 1  # 查询节次1
    jc2 = 2  # 查询节次2
    rq = "2024-12-09"  # 日期
    #借用教室的id列表
    id = []
    # 教室名称
    js_name = ""
    cookie =""
    cookie2 = ""

# 初始化查询的
def __init__(self, zc_, xq_, jc1_, jc2_, rq_,cookie_1,cookie_2):
    self.zc = zc_  # 周次
    self.xq = xq_  # 星期
    self.jc1 = jc1_  # 查询节次1
    self.jc2 = jc2_ # 查询节次2
    self.rq = rq_  # 日期
    self.cookie =cookie_1 #登录秘钥
    self.cookie2 = cookie_2  # 录课秘钥