import json
import requests
import uuid # 用于獲取驗證碼的，暫時可以不看
import yagmail # 用于發(fā)送郵件
import random
import datetime #用于獲取
from fake_useragent import UserAgent # 用于獲取隨機(jī)的UserAgent

username = '********' # 賬號
passwd = '*********' # 密碼
now_time = datetime.datetime.now ().strftime('%Y-%m-%d') # 當(dāng)天日期
contents = '' # 后面用于存儲課表內(nèi)容

# 隨機(jī)生成uuid
uuid = uuid.uuid4()


# 隨機(jī)UA
def get_random_ua():
    ua = UserAgent()
    return ua.random

# url匯總
code_url = f'http://sso.cqcet.edu.cn/validata/code/{uuid}'  # 驗證碼獲取url
url = f'http://sso.cqcet.edu.cn/verificationCode?userCode={username}'  # 判斷是否需要輸入驗證碼url
login_url = 'http://sso.cqcet.edu.cn/uaa/login_process'  # 登陸url
info_url = 'http://ossc.cqcet.edu.cn/getUserInfo'  # 獲取個人信息url
schedule_url = 'http://ossc.cqcet.edu.cn/api/schedule/query'  # 課表url

# 模擬登陸
headers = {
    'User-Agent': get_random_ua()
}

session = requests.session()


# 驗證碼圖片下載
def getImgCode():
    img_name = str(uuid) + '.jpg'
    img_data = requests.get(url=code_url, headers=headers).content
    with open(img_name, 'wb') as fp:
        fp.write(img_data)
        print('驗證碼獲取成功，請手動識別！')

# 驗證碼圖片下載
def getImgCode():
    img_name = str(uuid) + '.jpg'
    img_data = requests.get(url=code_url, headers=headers).content
    with open(img_name, 'wb') as fp:
        fp.write(img_data)
        print('驗證碼獲取成功，請手動識別！')


# 判斷是否需要輸入驗證碼
isCode = session.get(url=url, headers=headers).text

if isCode == 'true':
    getImgCode()
    # 獲取img_code
    img_code = input('請輸入驗證碼：')
else:
    img_code = ''

data = {
    'type': '1',
    'deviceId': str(uuid),
    'username': username,
    'password': passwd,
    'img_code': img_code
}
response = session.post(url=login_url, data=data)
login_text = response.text
# print(response.status_code)  # 返回200表示模擬登陸成功了！

# 個人信息獲取
info = session.get(url=info_url).text
info_data = json.loads(info)  # 將json字符串轉(zhuǎn)為字典
# print(info_data)
# pprint.pprint(info_data)

contents += '親愛的' + info_data['user']['name'] + '同學(xué):\n'

# 課表獲取
schedule_json = {
    'endDate': now_time,
    'limit': '10',
    'page': '1',
    'startDate': now_time,
    'type': '0'
}

schedule = session.post(url=schedule_url, json=schedule_json).text
schedule_data = json.loads(schedule)  # 將json字符串轉(zhuǎn)為字典
# pprint.pprint(len(schedule_data['data'])) # 獲取數(shù)組長度
# 判斷是否有課
if len(schedule_data['data']) == 0:
    re_list = ['本來無一物,何處惹塵埃~', '今天沒課，放松一下吧！', '小主，今日無課~\n可想好游玩之地？', '今日無課，無需早起~']
    contents += '        ' + re_list[random.randint(0, 3)]
else:
    contents += '&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;今日課表請查收~\n<hr/><table border="1" cellpadding="2" ' \
                'cellspacing="0"><tr><th>上課時間</th><th>課程名稱</th><th>授課教師</th><th>上課地點</th></tr>'
    for course_table in schedule_data['data']:
        # print(course_table)
        time = '<tr><td>' + course_table['date'] + ' ' + course_table['time'] + '</td>'
        title = '<td>' + course_table['activity']['title'] + '</td>'
        teacher = '<td>' + course_table['activity']['remarks'] + '</td>'
        classroom = '<td>' + course_table['activity']['address'] + '</td></tr>'
        contents += time + title + teacher + classroom
contents += '</table>'

# 發(fā)送郵件函數(shù)
def to_mail(content):
    yag = yagmail.SMTP(
        user='sm*****e@qq.com',
        host='smtp.qq.com',
        password="pdxj**********", # 注意此處的密碼不是你的郵箱密碼，具體是什么?？梢匀グ俣纫幌聐agmail的使用。
        smtp_ssl=True)
    yag.send(to=['3000004806@qq.com','s*****oe@qq.com'],
             subject='小主，你的課表來啦~',
             contents=contents)
             
to_mail(contents)  # 運(yùn)行郵件發(fā)送函數(shù)