weiban-tool/Utils.py

import time
import requests
import json
import asyncio

class main:
    tenantCode = 0
    userId = ""
    x_token = ""
    userProjectId = ""
    headers = {'x-token': "",
               "User-agent": "Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.5060.134 Mobile Safari/537.36 Edg/103.0.1264.77"
               }

    def __init__(self, code, id, token,projectId):
        self.tenantCode = code
        self.userId = id
        self.x_token = token
        self.userProjectId = projectId

    def init(self):
        self.headers['x-token'] = self.x_token

    def getRealName(self):
        url = f"https://weiban.mycourse.cn/pharos/my/getInfo.do?timestamp={int(time.time())}"
        data = {
            'tenantCode': self.tenantCode,
            'userId': self.userId
        }
        response = requests.post(url, data=data, headers=self.headers)
        text = response.text
        data = json.loads(text)
        return data['data']['realName']

    def getTaskName(self):
        url = f"https://weiban.mycourse.cn/pharos/index/listStudyTask.do?timestamp={int(time.time())}"
        data = {
            'tenantCode': self.tenantCode,
            'userId': self.userId,
            'limit': 2
        }
        response = requests.post(url, data=data, headers=self.headers)
        text = response.text
        data = json.loads(text)
        for i in data['data']:
            if self.userProjectId in i['userProjectId']:taskName = i['projectName']
        return taskName

    def getProgress(self):
        url = "https://weiban.mycourse.cn/pharos/project/showProgress.do"
        data = {
            'userProjectId': self.userProjectId,
            'tenantCode': self.tenantCode,
            'userId': self.userId
        }
        response = requests.post(url, data=data, headers=self.headers)
        text = response.text
        data = json.loads(text)
        return data['data']['progressPet']

    def getCategory(self):
        url = "https://weiban.mycourse.cn/pharos/usercourse/listCategory.do"
        data = {
            'userProjectId': self.userProjectId,
            'tenantCode': self.tenantCode,
            'userId': self.userId,
            'chooseType': 3
        }
        response = requests.post(url, data=data, headers=self.headers)
        text = response.text
        data = json.loads(text)
        list = data['data']
        result = []
        for i in list:
            if i['totalNum'] > i['finishedNum']:
                result.append(i['categoryCode'])
        return result

    def getCourse(self):
        url = "https://weiban.mycourse.cn/pharos/usercourse/listCourse.do"
        result = []
        for i in self.getCategory():
            data = {
                'userProjectId': self.userProjectId,
                'tenantCode': self.tenantCode,
                'userId': self.userId,
                'chooseType': 3,
                'name': "",
                'categoryCode': i
            }
            response = requests.post(url, data=data, headers=self.headers)
            text = response.text
            data = json.loads(text)['data']
            for i in data:
                if i['finished'] == 2:
                    result.append(i['resourceId'])
        return result

    def getFinishIdList(self):
        url = "https://weiban.mycourse.cn/pharos/usercourse/listCourse.do"
        result = {}
        for i in self.getCategory():
            data = {
                'userProjectId': self.userProjectId,
                'tenantCode': self.tenantCode,
                'userId': self.userId,
                'chooseType': 3,
                'name': "",
                'categoryCode': i
            }
            response = requests.post(url, data=data, headers=self.headers)
            text = response.text
            data = json.loads(text)['data']
            for i in data:
                if i['finished'] == 2:
                    result[i['resourceId']] = i['userCourseId']
        return result


    async def start(self,courseId):
        data = {
            'userProjectId': self.userProjectId,
            'tenantCode': self.tenantCode,
            'userId': self.userId,
            'courseId': courseId
        }
        headers = {
            "x-token":self.x_token
        }
        res = requests.post("https://weiban.mycourse.cn/pharos/usercourse/study.do",data=data,headers=headers)
        while json.loads(res.text)['code'] == -1:
            await asyncio.sleep(5)
            res = requests.post("https://weiban.mycourse.cn/pharos/usercourse/study.do",data=data,headers=headers)
        print(f"start:{courseId}\r",end='')

    def finish(self,finishId):
        params = {
            "callback":"",
            "userCourseId":finishId,
            "tenantCode":self.tenantCode
        }
        url = "https://weiban.mycourse.cn/pharos/usercourse/finish.do"
        requests.get(url=url,params=params)
        print(f"finish:{finishId}\r",end='')