import json
import time
import csv
import argparse
import logging
from functools import partial
import config
import sqlite_utils as db_tools
import scraper
import utils 
import config
import convert_utils 

config.setup_logging()

books_dir = f"{config.global_host_data_dir}/aabook/data"

map_books_1 = {
    380     : "多情村妇",
    2547    : "乡村乱情|奇思妙想",
    2261    : "宋家湾那些事儿",
    3152    : "渔港春夜",
    3167    : "天狗",
    3681    : "东北大炕",
    5524    : "淫乱竹楼",
    5389    : "杨家洼情事",
    2783    : "风流岁月",
    4827    : "蛮荒小村的风流韵事",
    1114    : "熟透了的村妇",
    4347    : "潭河峪的那些事儿",
    3744    : "香火",
    4984    : "山里人家",
    
}
map_books_2 = {
    2426    : "我和我的母亲（寄印传奇）",
    2980    : "我和我的母亲（改写寄印传奇）",
    3720    : "小村春色【实体版】",
    4289    : "天若有情(一家之主)",
    2037    : "流年",
    5240    : "幸福的借种经历",
    5171    : "我的师娘和师妹",
}

map_books = map_books_2
# 使用示例
if __name__ == "__main__":
    #books = [2689, 3727, 4698, 5446]
    #books = [3167, 2985, 3098]
    #books = [2783]
    for book, name in map_books.items():
        data = db_tools.get_contents_by_book(book)
        if data:
            title = data['title']
            convert_utils.generate_epub(data, f"{books_dir}/{title}.epub")
            print(f"dump {title} suss!")