一鍵匯出微信讀書的書籍和筆記

簡介

全民閱讀的時代已經來臨，目前使用讀書軟體的用戶數2.1億，榷訓躍用戶超過500萬，其中19-35歲年輕用戶占比超過60%，本科及以上學歷用戶占比高達80%，北上廣深及其他省會城市/直轄市用戶占比超過80%，本人習慣使用微信讀書，為了方便整理書籍和匯出筆記，便開發了這個小工具，

部分截圖

代碼思路

1. 目錄結構

首先，我們先看一下整體目錄結構

Code
├─ excel_func.py                   讀寫excel檔案
├─ pyqt_gui.py                     PyQt GUI界面
└─ wereader.py                     微信讀書相關api

• excel_func.py
  使用xlrd和xlwt庫對excel檔案進行讀寫操作

• pyqt_gui.py
  使用PyQt繪制GUI界面

• wereader.py
  通過抓包決議獲得相關api

2. excel_func.py

def write_excel_xls(path, sheet_name_list, value):
    # 新建一個作業簿
    workbook = xlwt.Workbook()

    # 獲取需要寫入資料的行數
    index = len(value)

    for sheet_name in sheet_name_list:

        # 在作業簿中新建一個表格
        sheet = workbook.add_sheet(sheet_name)

        # 往這個作業簿的表格中寫入資料
        for i in range(0, index):
            for j in range(0, len(value[i])):
                sheet.write(i, j, value[i][j])

    # 保存作業簿
    workbook.save(path)

該函式的代碼流程為:

1. 創建excel檔案
2. 創建表格
3. 往表格寫入資料

3. pyqt_gui.py

class MainWindow(QMainWindow):

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.DomainCookies = {}

        self.setWindowTitle('微信讀書助手') # 設定視窗標題
        self.resize(900, 600) # 設定視窗大小
        self.setWindowFlags(Qt.WindowMinimizeButtonHint) # 禁止最大化按鈕
        self.setFixedSize(self.width(), self.height()) # 禁止調整視窗大小

        url = 'https://weread.qq.com/#login' # 目標地址
        self.browser = QWebEngineView() # 實體化瀏覽器物件

        QWebEngineProfile.defaultProfile().cookieStore().deleteAllCookies() # 初次運行軟體時洗掉所有cookies

        QWebEngineProfile.defaultProfile().cookieStore().cookieAdded.connect(self.onCookieAdd) # cookies增加時觸發self.onCookieAdd()函式
        self.browser.loadFinished.connect(self.onLoadFinished) # 網頁加載完畢時觸發self.onLoadFinished()函式

        self.browser.load(QUrl(url)) # 加載網頁
        self.setCentralWidget(self.browser) # 設定中心視窗

該函式的代碼流程為:

1. 新建QT視窗
2. 實體化QWebEngineView物件
3. 系結self.onCookieAdd事件
4. 系結self.onLoadFinished事件
5. 加載網頁

    # 網頁加載完畢事件
    def onl oadFinished(self):

        global USER_VID
        global HEADERS

        # 獲取cookies
        cookies = ['{}={};'.format(key, value) for key,value in self.DomainCookies.items()]
        cookies = ' '.join(cookies)
        # 添加Cookie到header
        HEADERS.update(Cookie=cookies)

        # 判斷是否成功登錄微信讀書
        if login_success(HEADERS):
            print('登錄微信讀書成功!')

            # 獲取用戶user_vid
            if 'wr_vid' in self.DomainCookies.keys():
                USER_VID = self.DomainCookies['wr_vid']
                print('用戶id:{}'.format(USER_VID))

                # 關閉整個qt視窗
                self.close()

        else:
            print('請掃描二維碼登錄微信讀書...')

該函式的代碼流程為:

1. 當網頁加載完畢時，檢測是否成功登錄微信讀書
2. 如果成功登錄微信讀書，則關閉QT視窗，開始進行資料匯出
3. 如果失敗登錄微信讀書，則繼續等待用戶掃描二維碼

    # 添加cookies事件
    def onCookieAdd(self, cookie):
        if 'weread.qq.com' in cookie.domain():
            name = cookie.name().data().decode('utf-8')
            value = https://www.cnblogs.com/cloudbird/p/cookie.value().data().decode('utf-8')
            if name not in self.DomainCookies:
                self.DomainCookies.update({name: value})

該函式的代碼流程為:

1. 保存微信讀書網址的cookies，以便后續操作

    books = get_bookshelf(USER_VID, HEADERS) # 獲取書架上的書籍
    books_finish_read = books['finishReadBooks']
    books_recent_read = books['recentBooks']
    books_all = books['allBooks']
    write_excel_xls_append(data_dir + '我的書架.xls', '已讀完的書籍', books_finish_read) # 追加寫入excel檔案
    write_excel_xls_append(data_dir + '我的書架.xls', '最近閱讀的書籍', books_recent_read)  # 追加寫入excel檔案
    write_excel_xls_append(data_dir + '我的書架.xls', '所有的書籍', books_all)  # 追加寫入excel檔案

    # 獲取書架上的每本書籍的筆記
    for index, book in enumerate(books_finish_read):
        book_id = book[0]
        book_name = book[1]
        notes = get_bookmarklist(book[0], HEADERS)

        with open(note_dir + book_name + '.txt', 'w') as f:
            f.write(notes)
        print('匯出筆記 {} ({}/{})'.format(note_dir + book_name + '.txt', index+1, len(books_finish_read)))

該函式的代碼流程為:

1. 呼叫write_excel_xls_append函式，保存書籍，并且匯出筆記

4. wereader.py

def get_bookshelf(userVid, headers):
    """獲取書架上所有書"""
    url = "https://i.weread.qq.com/shelf/friendCommon"
    params = dict(userVid=userVid)
    r = requests.get(url, params=params, headers=headers, verify=False)
    if r.ok:
        data = https://www.cnblogs.com/cloudbird/p/r.json()
    else:
        raise Exception(r.text)

    books_finish_read = set() # 已讀完的書籍
    books_recent_read = set() # 最近閱讀的書籍
    books_all = set() # 書架上的所有書籍


    for book in data['recentBooks']:
        if not book['bookId'].isdigit(): # 過濾公眾號
            continue
        b = Book(book['bookId'], book['title'], book['author'], book['cover'], book['intro'], book['category'])
        books_recent_read.add(b)

    books_all = books_finish_read + books_recent_read

    return dict(finishReadBooks=books_finish_read, recentBooks=books_recent_read, allBooks=books_all)

該函式的代碼流程為:

1. 獲取最近閱讀的書籍、已經讀完的書籍、所有書籍
2. 過濾公眾號部分
3. 將書籍資料保存為字典格式

def get_bookmarklist(bookId, headers):
    """獲取某本書的筆記回傳md文本"""
    url = "https://i.weread.qq.com/book/bookmarklist"
    params = dict(bookId=bookId)
    r = requests.get(url, params=params, headers=headers, verify=False)

    if r.ok:
        data = https://www.cnblogs.com/cloudbird/p/r.json()
        # clipboard.copy(json.dumps(data, indent=4, sort_keys=True))
    else:
        raise Exception(r.text)
    chapters = {c['chapterUid']: c['title'] for c in data['chapters']}
    contents = defaultdict(list)

    for item in sorted(data['updated'], key=lambda x: x['chapterUid']):
        # for item in data['updated']:
        chapter = item['chapterUid']
        text = item['markText']
        create_time = item["createTime"]
        start = int(item['range'].split('-')[0])
        contents[chapter].append((start, text))

    chapters_map = {title: level for level, title in get_chapters(int(bookId), headers)}
    res = ''
    for c in sorted(chapters.keys()):
        title = chapters[c]
        res += '#' * chapters_map[title] + ' ' + title + '\n'
        for start, text in sorted(contents[c], key=lambda e: e[0]):
            res += '> ' + text.strip() + '\n\n'
        res += '\n'

    return res

該函式的代碼流程為:

1. 獲取某一本書籍的筆記
2. 將回傳的字串改寫成markdown格式并輸出

如何運行

# 跳轉到當前目錄
cd 目錄名
# 先卸載依賴庫
pip uninstall -y -r requirement.txt
# 再重新安裝依賴庫
pip install -r requirement.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
# 開始運行
python pyqt_gui.py

補充

完整版源代碼存放在github上，有需要的請點擊這里下載

專案持續更新，歡迎您star本專案

License

The MIT License (MIT)

標籤：Python

上一篇：農夫過河問題

下一篇：使用python生成隨機圖片驗證碼