Greasy Fork

Greasy Fork is available in English.

豆瓣读书+电影+音乐+游戏+舞台剧导出工具

将【读过/看过/听过/玩过】的 【读书/电影/音乐/游戏/舞台剧】条目分别导出为 csv 文件

您需要先安装一款用户脚本管理器扩展,例如 Tampermonkey 篡改猴Greasemonkey 油猴子Violentmonkey 暴力猴,才能安装此脚本。

您需要先安装一款用户脚本管理器扩展,例如 Tampermonkey 篡改猴,才能安装此脚本。

您需要先安装一款用户脚本管理器扩展,例如 Tampermonkey 篡改猴Violentmonkey 暴力猴,才能安装此脚本。

您需要先安装一款用户脚本管理器扩展,例如 Tampermonkey 篡改猴Userscripts ,才能安装此脚本。

您需要先安装一款用户脚本管理器扩展,例如 Tampermonkey 篡改猴,才能安装此脚本。

您需要先安装一款用户脚本管理器扩展后才能安装此脚本。

(我已经安装了用户脚本管理器,让我安装!)

您需要先安装一款用户样式管理器扩展,比如 Stylus,才能安装此样式。

您需要先安装一款用户样式管理器扩展,比如 Stylus,才能安装此样式。

您需要先安装一款用户样式管理器扩展,比如 Stylus,才能安装此样式。

您需要先安装一款用户样式管理器扩展后才能安装此样式。

您需要先安装一款用户样式管理器扩展后才能安装此样式。

您需要先安装一款用户样式管理器扩展后才能安装此样式。

(我已经安装了用户样式管理器,让我安装!)

// ==UserScript==
// @name             豆瓣读书+电影+音乐+游戏+舞台剧导出工具
// @namespace        https://ulyc.github.io/
// @version          0.1.2
// @description      将【读过/看过/听过/玩过】的 【读书/电影/音乐/游戏/舞台剧】条目分别导出为 csv 文件
// @author           ulyc
// @match            https://book.douban.com/people/*/collect*
// @match            https://movie.douban.com/people/*/collect*
// @match            https://music.douban.com/people/*/collect*
// @match            https://www.douban.com/location/people/*/drama/collect*
// @match            https://*.douban.com/people/*
// @require          https://unpkg.com/dexie@latest/dist/dexie.js
// @grant            none
// @original-script  http://greasyfork.icu/en/scripts/420999-%E8%B1%86%E7%93%A3%E8%AF%BB%E4%B9%A6-%E7%94%B5%E5%BD%B1-%E9%9F%B3%E4%B9%90-%E6%B8%B8%E6%88%8F-%E8%88%9E%E5%8F%B0%E5%89%A7%E5%AF%BC%E5%87%BA%E5%B7%A5%E5%85%B7
// @original-license MIT
// @license MIT
// ==/UserScript==


(function () {
    'use strict';
    let people;
    const MOVIE = 'movie', BOOK = 'book', MUSIC = 'music', GAME = 'game', DRAMA = 'drama';
    const commonItem = "++id, cover, title, rating, rating_date, comment,";

    /* global $, Dexie */

    function getExportLink(type, people) {
        return 'https://' + type + '.douban.com/people/' + people + '/collect?start=0&sort=time&mode=grid&rating=all&export=1';
    }

    function getGameExportLink(people) { // type=game
        return 'https://www.douban.com/people/' + people + '/games?action=collect&start=0&export=1';
    }

    function getDramaExportLink(people) { // type=drama
        return 'https://www.douban.com/location/people/' + people + '/drama/collect?start=0&sort=time&mode=grid&rating=all&export=1';
    }


    function addButton(type, people, isGeneral) {
        let typeElement;
        if (isGeneral) {
            typeElement = $('#' + type + ' h2 .pl a:last');
        } else {
            typeElement = $('#db-' + type + '-mine h2:contains(过)  .pl');
        }
        switch (type) {
            case BOOK:
                typeElement.after('&nbsp;·&nbsp;<a href="' + getExportLink(type, people) + '">导出读过的书</a>');
                break;
            case MOVIE:
                typeElement.after('&nbsp;·&nbsp;<a href="' + getExportLink(type, people) + '">导出看过的片</a>');
                break;
            case MUSIC:
                typeElement.after('&nbsp;·&nbsp;<a href="' + getExportLink(type, people) + '">导出听过的碟</a>');
                break;
            case GAME:
                typeElement.after('&nbsp;·&nbsp;<a href="' + getGameExportLink(people) + '">导出玩过的游戏</a>');
                break;
            case DRAMA:
                typeElement.after('&nbsp;·&nbsp;<a href="' + getDramaExportLink(people) + '">导出看过的舞台剧</a>');
                break;
            default:
                return;
        }
    }

    function getType() {
        let type = location.hostname.split(".")[0];

        if (type === "www") {
            if (location.pathname.indexOf('games') > -1) {
                type = GAME;
            }
            if (location.pathname.indexOf('drama') > -1) {
                type = DRAMA;
            }
        }
        return type;
    }

    // 根据hostname判断type
    // 加入导出按钮
    if (location.href.indexOf('douban.com') > -1 && location.href.indexOf('export=1') < 0) {
        let match;
        let type = getType();
        if (type === DRAMA) {
            match = location.pathname.match(/location\/people\/([^/]+)\//);
        } else {
            match = location.pathname.match(/people\/([^/]+)\//);
        }
        people = match ? match[1] : null;
        if (!people) {
            return
        }

        if (type === "www") {
            addButton(BOOK, people, true);
            addButton(MOVIE, people, true);
            addButton(MUSIC, people, true);
            addButton(GAME, people, true);
            addButton(DRAMA, people, true);
        } else {
            addButton(type, people, false);
        }
    }

    if (location.href.indexOf('export=1') > -1) {
        let type = getType();
        if ([DRAMA, GAME, BOOK, MUSIC, MOVIE].indexOf(type) < 0) {
            alert("类型错误,请检查路径")
            return;
        }

        init(type);
    }


    function escapeQuote(str) {
        return str.replaceAll('"', '""'); // " need to be replaced with two quotes to escape inside csv quoted string
    }

    // 获取当前页数据
    function getCurPageItems(type) {
        var items = [];
        var elems = $('.grid-view .item');

        if (type === GAME) {
            elems = $('.game-list .common-item');
        } else if (type === BOOK) {
            elems = $('.interest-list .subject-item');
        }

        elems.each(function (index) {
            let item = {
                title: "",
                rate: 5,
                comment: "",
                cover: escapeQuote($(this).find('.pic img').attr('src').trim()),
                'rating_date': $(this).find('.date').text().trim().replaceAll('-', '/'), // 2020-07-17 => 2020/07/17
            };
            transWebp2Jpg(item);
            getTitleAndLink($(this), item, type);
            //兼容被封图书项目,电影被封项目似乎不展示
            if (item.cover.indexOf("book-default-lpic") > -1 && item.title.indexOf("未知") > -1) {
                item.title = "被封禁的图书";
                item.rate = 5;
                item.comment = "无";
                item.author = "佚名";
                item.release_date = "1984/01/01";
                items[index] = item;
            } else if (item.cover.indexOf("lpic/music-default") > -1 && item.title.indexOf("未知") > -1) {
                item.title = "被封禁的音乐";
                item.rate = 5;
                item.comment = "无";
                item.musician = "佚名";
                item.release_date = "1984/01/01";
                items[index] = item;
            } else {
                getRate($(this), item, type);
                getComment($(this), item, type);
                getExtraInfo($(this), item, type, index);
            }

        });
        return items;

        function getTitleAndLink(elem, item, type) {
            if (type === BOOK) {
                item.title = elem.find('.info a').attr("title");
                item.link = elem.find('.info a').attr("href").trim();
            } else {
                item.title = elem.find('.title a').text();
                item.link = elem.find('.title a').attr('href').trim();
            }
            // 处理空格/换行符 , 优化格式
            item.title = escapeQuote(item.title.trim().replace(/\s*/g, "").replace(/\r\n/g, "").replace(/\n/g, ""))
        }

        function getRate(elem, item, type) {
            // 获取 评分
            if (type === GAME) {
                let rating = elem.find('.rating-info .rating-star').attr('class');
                rating = rating
                    ? (rating.slice(19, 20) === 'N' ? '' : Number(rating.slice(19, 20)))
                    : '';
                item.rating = rating;
            } else {
                let rating = elem.find('.date')[0].previousElementSibling;
                if (rating) {
                    rating = $(rating).attr('class').slice(6, 7);
                }
                item.rating = rating ? Number(rating) : '';
            }
        }

        function getComment(elem, item, type) {
            let co = elem.find('.comment');
            if (co.length) {
                co = co[0];
                item.comment = escapeQuote(co.textContent.trim());
            } else if (type === GAME) {
                co = elem.find('.user-operation');
                if (co.length) {
                    co = co[0];
                    item.comment = co.previousElementSibling.textContent.trim();
                    item.comment = escapeQuote(item.comment);
                }
            } else if (type === DRAMA || type === MUSIC) {
                co = elem.find('ul li:last');
                item.comment = escapeQuote(co.text().trim());
            }
        }

        function getExtraInfo(elem, item, type, index) {
            let extra;
            switch (type) {
                case GAME:
                    extra = elem.find('.desc')[0].firstChild.textContent.trim();
                    item.release_date = extra.split(' / ').slice(-1)[0];
                    items[index] = item;
                    return; // for type=game, here is over
                case DRAMA:
                    extra = elem.find('.intro')[0].textContent.trim();
                    item.mixed_info = extra;
                    items[index] = item;
                    return; // for type=drama, here is over
                case MOVIE:
                    extra = elem.find('.intro').text().split(' / ')[0];
                    let res = extra.match(/^(\d{4}-\d{2}-\d{2})\((.*)\)$/);
                    if (res) {
                        item.release_date = res[1].replaceAll('-', '/');
                        item.country = res[2];
                    }
                    break;
                case MUSIC:
                case BOOK:
                    let className = type === BOOK ? 'pub' : 'intro';
                    extra = elem.find('.' + className).text().split(' / ');
                    // 不一定有准确日期,可能是 2009-5 这样的, 也可能就只有年份 2000
                    let dateReg = /\d{4}(?:-\d{1,2})?(?:-\d{1,2})?/;
                    if (!dateReg.test(extra[0])) { // extra 首项非日期,则一般为作者或音乐家
                        let author = escapeQuote(extra[0].trim());
                        item.musician = item.author = author;
                    }
                    let d = extra.filter(function (txt) {
                        return dateReg.test(txt);
                    });
                    if (d.length) {
                        item.release_date = d[0].replaceAll('-', '/');
                    }
                    break;
            }

            items[index] = item;
        }

        function transWebp2Jpg(item) {
            // notion不识别webp格式图片
            // 将豆瓣webp格式图片 转换为 jpg

            if (!!item.cover && item.cover.endsWith("webp")) {
                item.cover = item.cover.replace(".webp", ".jpg")
            }
        }

    }

    function getDBItems(type) {
        let items;
        switch (type) {
            case MOVIE:
                items = commonItem + ' release_date, country, link';
                break;
            case MUSIC:
                items = commonItem + ' release_date, author, link';
                break;
            case BOOK:
                items = commonItem + ' release_date, author, link';
                break;
            case GAME:
                items = commonItem + ' release_date, link';
                break;
            case DRAMA:
                items = commonItem + ' mixed_info, link';
                break;
        }
        return items;
    }

    function init(type) {
        const db = new Dexie('db_export'); // init indexedDB
        let dbItems = getDBItems(type);
        db.version(1).stores({
            items: dbItems
        });

        const items = getCurPageItems(type);
        db.items.bulkAdd(items).then(function () {
            console.log('添加成功+', items.length);

            let nextPageLink = $('.paginator span.next a').attr('href');
            if (nextPageLink) {
                nextPageLink = nextPageLink + '&export=1';
                window.location.href = nextPageLink;
            } else {
                exportAll(type);
            }
        }).catch(function (error) {
            console.error("Ooops: " + error);
        });
    }

    function exportAll(type) {
        const db = new Dexie('db_export');
        let items = getDBItems(type);

        db.version(1).stores({
            items: items
        });
        db.items.orderBy('rating_date').reverse().toArray().then(function (all) {
            all = all.map(function (item) {
                delete item.id;
                return item;
            });

            let title = ['封面', '标题', '个人评分', '打分日期', '我的短评'];
            let key = ['cover', 'title', 'rating', 'rating_date', 'comment', 'release_date'];
            if (type === MOVIE) {
                title = title.concat(['上映日期', '制片国家', '条目链接']);
                key = key.concat(['country', 'link']);
            } else if (type === BOOK) {
                title = title.concat(['出版日期', '作者', '条目链接']);
                key = key.concat(['author', 'link']);
            } else if (type === MUSIC) {
                title = title.concat(['发行日期', '音乐家', '条目链接']);
                key = key.concat(['musician', 'link']);
            } else if (type === GAME) {
                title = title.concat(['发行日期', '条目链接']);
                key.push('link');
            } else if (type === DRAMA) {
                title = title.concat(['混合信息', '条目链接']);
                key.pop();
                key = key.concat(['mixed_info', 'link']);
            }

            JSonToCSV.setDataConver({
                data: all,
                fileName: 'db-' + type + '-' + new Date().toISOString().split('T')[0].replaceAll('-', ''),
                columns: {title, key},
            });
            db.delete();
        });
    }

    // 导出CSV函数
    // https://github.com/liqingzheng/pc/blob/master/JsonExportToCSV.js
    var JSonToCSV = {
        /*
         * obj是一个对象,其中包含有:
         * ## data 是导出的具体数据
         * ## fileName 是导出时保存的文件名称 是string格式
         * ## showLabel 表示是否显示表头 默认显示 是布尔格式
         * ## columns 是表头对象,且title和key必须一一对应,包含有
              title:[], // 表头展示的文字
              key:[], // 获取数据的Key
              formatter: function() // 自定义设置当前数据的 传入(key, value)
         */
        setDataConver: function (obj) {
            var bw = this.browser();
            if (bw['ie'] < 9) return; // IE9以下的
            var data = obj['data'],
                ShowLabel = typeof obj['showLabel'] === 'undefined' ? true : obj['showLabel'],
                fileName = (obj['fileName'] || 'UserExport') + '.csv',
                columns = obj['columns'] || {
                    title: [],
                    key: [],
                    formatter: undefined
                };
            ShowLabel = typeof ShowLabel === 'undefined' ? true : ShowLabel;
            var row = "",
                CSV = '',
                key;
            // 如果要现实表头文字
            if (ShowLabel) {
                // 如果有传入自定义的表头文字
                if (columns.title.length) {
                    columns.title.map(function (n) {
                        row += n + ',';
                    });
                } else {
                    // 如果没有,就直接取数据第一条的对象的属性
                    for (key in data[0]) row += key + ',';
                }
                row = row.slice(0, -1); // 删除最后一个,号,即a,b, => a,b
                CSV += row + '\r\n'; // 添加换行符号
            }
            // 具体的数据处理
            data.map(function (n) {
                row = '';
                // 如果存在自定义key值
                if (columns.key.length) {
                    columns.key.map(function (m) {
                        row += '"' + (typeof columns.formatter === 'function' ? columns.formatter(m, n[m]) || n[m] || '' : n[m] || '') + '",';
                    });
                } else {
                    for (key in n) {
                        row += '"' + (typeof columns.formatter === 'function' ? columns.formatter(key, n[key]) || n[key] || '' : n[key] || '') + '",';
                    }
                }
                row.slice(0, row.length - 1); // 删除最后一个,
                CSV += row + '\r\n'; // 添加换行符号
            });
            if (!CSV) return;
            this.SaveAs(fileName, CSV);
        },
        SaveAs: function (fileName, csvData) {
            var bw = this.browser();
            if (!bw['edge'] || !bw['ie']) {
                var alink = document.createElement("a");
                alink.id = "linkDwnldLink";
                alink.href = this.getDownloadUrl(csvData);
                document.body.appendChild(alink);
                var linkDom = document.getElementById('linkDwnldLink');
                linkDom.setAttribute('download', fileName);
                linkDom.click();
                document.body.removeChild(linkDom);
            } else if (bw['ie'] >= 10 || bw['edge'] == 'edge') {
                var _utf = "\uFEFF";
                var _csvData = new Blob([_utf + csvData], {
                    type: 'text/csv'
                });
                navigator.msSaveBlob(_csvData, fileName);
            } else {
                var oWin = window.top.open("about:blank", "_blank");
                oWin.document.write('sep=,\r\n' + csvData);
                oWin.document.close();
                oWin.document.execCommand('SaveAs', true, fileName);
                oWin.close();
            }
        },
        getDownloadUrl: function (csvData) {
            var _utf = "\uFEFF"; // 为了使Excel以utf-8的编码模式,同时也是解决中文乱码的问题
            if (window.Blob && window.URL && window.URL.createObjectURL) {
                csvData = new Blob([_utf + csvData], {
                    type: 'text/csv'
                });
                return URL.createObjectURL(csvData);
            }
            // return 'data:attachment/csv;charset=utf-8,' + _utf + encodeURIComponent(csvData);
        },
        browser: function () {
            var Sys = {};
            var ua = navigator.userAgent.toLowerCase();
            var s;
            (s = ua.indexOf('edge') !== -1 ? Sys.edge = 'edge' : ua.match(/rv:([\d.]+)\) like gecko/)) ? Sys.ie = s[1] :
                (s = ua.match(/msie ([\d.]+)/)) ? Sys.ie = s[1] :
                    (s = ua.match(/firefox\/([\d.]+)/)) ? Sys.firefox = s[1] :
                        (s = ua.match(/chrome\/([\d.]+)/)) ? Sys.chrome = s[1] :
                            (s = ua.match(/opera.([\d.]+)/)) ? Sys.opera = s[1] :
                                (s = ua.match(/version\/([\d.]+).*safari/)) ? Sys.safari = s[1] : 0;
            return Sys;
        }
    };

})();