Python path.Path類代碼示例

    def create(self):
        self.image_container.set_save_path(Path.image_pool_path)
        self.image_container.start_download()
        title = "_".join([book.property.epub.title for book in self.book_list])
        title = title.strip()
        Path.chdir(Path.base_path + u"/知乎電子書臨時資源庫/")
        epub = Book(title, 27149527)
        html_tmp_path = Path.html_pool_path + "/"
        image_tmp_path = Path.image_pool_path + "/"
        for book in self.book_list:
            page = book.page_list[0]
            with open(html_tmp_path + page.filename, "w") as html:
                html.write(page.content)
            epub.createChapter(html_tmp_path + page.filename, ExtraTools.get_time(), page.title)

            for page in book.page_list[1:]:
                with open(html_tmp_path + page.filename, "w") as html:
                    html.write(page.content)
                epub.addHtml(html_tmp_path + page.filename, page.title)
        for image in self.book["image_list"]:
            epub.addImg(image_tmp_path + image["filename"])
        epub.addLanguage("zh-cn")
        epub.addCreator("ZhihuHelp1.7.0")
        epub.addDesc(u"該電子書由知乎助手生成，知乎助手是姚澤源為知友製作的僅供個人使用的簡易電子書製作工具，源代碼遵循WTFPL，希望大家能認真領會該協議的真諦，為飛麵事業做出自己的貢獻 XD")
        epub.addRight("CC")
        epub.addPublisher("ZhihuHelp")
        Debug.logger.debug(u"當前目錄為")
        Path.pwd()
        epub.addCss(Path.base_path + u"/epubResource/markdown.css")
        epub.addCss(Path.base_path + u"/epubResource/front.css")
        epub.buildingEpub()
        return

    def __init__(self, recipe_kind='Notset', read_list='ReadList.txt', url=None, debug=False):
        u"""
        配置文件使用$符區隔，同一行內的配置文件歸並至一本電子書內
        :param recipe_kind:
        :param read_list: default value: ReadList.txt
        :param url:
        :param debug:
        :return:
        """
        self.recipe_kind = recipe_kind
        self.read_list = read_list
        self.url = url
        log.warning_log(u"website type: " + str(self.recipe_kind) + '\n')
        import logging
        if debug is True:
            Debug.logger.setLevel(logging.DEBUG)
        else:
            Debug.logger.setLevel(logging.INFO)

        Debug.logger.debug(u"read_list: " + str(self.read_list))
        Debug.logger.debug(u"url: " + str(self.url))
        Debug.logger.debug(u"recipe type:" + str(recipe_kind))

        Path.init_base_path(recipe_kind)        # 設置路徑
        Path.init_work_directory()              # 創建路徑
        self.init_database()                    # 初始化數據庫
        Config._load()
        return

 def create_book(self, book_package):
     book_package.image_container.set_save_path(Path.image_pool_path)
     book_package.image_container.start_download()
     title = book_package.get_title()
     if not title:
         # 電子書題目為空時自動跳過
         # 否則會發生『rm -rf / 』的慘劇
         return
     Path.chdir(Path.base_path + u'/知乎電子書臨時資源庫/')
     epub = Epub(title)
     html_tmp_path = Path.html_pool_path + u'/'
     image_tmp_path = Path.image_pool_path + u'/'
     epub.set_creator(u'ZhihuHelp1.7.0')
     epub.set_book_id()
     epub.set_output_path(Path.result_path)
     epub.add_css(Path.base_path + u'/www/css/markdown.css')
     epub.add_css(Path.base_path + u'/www/css/customer.css')
     epub.add_css(Path.base_path + u'/www/css/normalize.css')
     for book in book_package.book_list:
         page = book.page_list[0]
         with open(html_tmp_path + page.filename, u'w') as html:
             html.write(page.content)
         epub.create_chapter(html_tmp_path + page.filename, page.title)
         for page in book.page_list[1:]:
             with open(html_tmp_path + page.filename, u'w') as html:
                 html.write(page.content)
             epub.add_html(html_tmp_path + page.filename, page.title)
         epub.finish_chapter()
     for image in book_package.image_list:
         epub.add_image(image_tmp_path + image['filename'])
     epub.create()
     Path.reset_path()
     return

 def filepath(answertmp = []):
     if answertmp:
         answerpath = Path.answer_path+'/Answer_qid_aid/'+'{0}'.format(answertmp['question_id'])
     if not Path.is_dir(answerpath):
         Path.mkdirs(answerpath)
     filename = '{0}'.format(answertmp['question_id'])+'_'+'{0}'.format(answertmp['answer_id'])+'.txt'
     return Path.join_dir(answerpath,filename)

    def create_book(self, command, counter):
        Path.reset_path()
        Debug.logger.info(u"開始製作第 {} 本電子書".format(counter))
        Debug.logger.info(u"對記錄 {} 進行分析".format(command))
        task_list = CommandParser.get_task_list(command)  # 分析命令

        if len(task_list) == 0:
            return

        for task in task_list:
            if Config.debug_for_create_book:
                pass
            else:
                Worker.distribute(task)
        Debug.logger.info(u"網頁信息抓取完畢")

        task_result_list = []
        for task in task_list:
            task_result = TaskResult(task)
            task_result.extract_data()
            task_result_list.append(task_result)
        Debug.logger.info(u"數據庫信息獲取完畢")

        #   下載圖片
        for task_result in task_result_list:
            task_result.download_img()
        Debug.logger.info(u"所有任務圖片獲取完畢")

        #   按體積自動分卷
        #   渲染html && 壓縮為電子書
        book = Book(task_result_list)
        book_list = book.auto_split(Config.max_book_size_mb * 1024)
        for chapter in book_list:
            chapter.create_book()
        return

 def add_index_html(self, src, title):
     Path.copy(src, EpubPath.html_path)
     filename = Path.get_filename(src)
     new_src = u'html/' + filename
     resource_id = self.opf.add_html(new_src)
     self.toc.add_item(resource_id, new_src, title)
     return

 def __init__(self):
     u"""
     配置文件使用$符區隔，同一行內的配置文件歸並至一本電子書內
     """
     init.init_database()
     Path.init_base_path()
     Config._load()
     return

 def create_chapter(self, src, title):
     Path.copy(src, EpubPath.html_path)
     filename = Path.get_filename(src)
     new_src = u'html/' + filename
     resource_id = self.opf.add_title_page_html(new_src)
     self.directory.create_chapter(new_src, title)
     self.toc.create_chapter(resource_id, new_src, title)
     return

 def __init__(self):
     u"""
     配置文件使用$符區隔，同一行內的配置文件歸並至一本電子書內
     """
     Path.init_base_path()       # 設置路徑
     Path.init_work_directory()  # 創建路徑
     self.init_database()        # 初始化數據庫
     Config._load()
     return

 def __init__(self):
     #   初始化目錄結構
     Path.init_base_path()
     Path.init_work_directory()
     #   初始化數據庫鏈接
     DB.init_database()
     #   初始化配置
     Config.init_config()
     return

    def create_book(self):
        #   確定文件信息
        title = Match.fix_filename(self.book_title)
        if self.is_split:
            title = self.book_title + u'_卷{}'.format(self.chapter_no)

        #   先切換到電子書臨時資源目錄下
        Path.chdir(Path.book_pool_path)
        epub = Epub(title)
        for task_result in self.task_result_list:
            chapter_src = ''
            # info_page
            if task_result.task.task_type == Type.question:
                chapter_src = self.generate_question_info_page(task_result.info_page)
            elif task_result.task.task_type == Type.answer:
                chapter_src = self.generate_question_info_page(task_result.info_page)
            elif task_result.task.task_type == Type.collection:
                chapter_src = self.generate_collection_info_page(task_result.info_page)
            elif task_result.task.task_type == Type.topic:
                chapter_src = self.generate_topic_info_page(task_result.info_page)
            elif task_result.task.task_type == Type.author:
                chapter_src = self.generate_author_info_page(task_result.info_page)
            elif task_result.task.task_type == Type.column:
                chapter_src = self.generate_column_info_page(task_result.info_page)
            elif task_result.task.task_type == Type.article:
                chapter_src = self.generate_article_info_page(task_result.info_page)
            epub.create_chapter(chapter_src, task_result.get_title())
            for question in task_result.question_list:
                #   添加圖片文件
                for filename in question.img_filename_list:
                    epub.add_image(Path.image_pool_path + '/' + filename)
                question_src = self.generate_question_page(question)
                epub.add_html(question_src, question.question_info.title)

            for column in task_result.column_list:
                #   添加圖片文件
                for filename in column.img_filename_list:
                    epub.add_image(Path.image_pool_path + '/' + filename)
                for article in column.article_list:
                    article_src = self.generate_article_page(article)
                    epub.add_html(article_src, article.title)
            epub.finish_chapter()

        epub.set_creator(u'ZhihuHelp1.8.0')
        epub.set_language(u'zh-cn')
        epub.set_book_id()
        epub.set_output_path(Path.result_path)
        epub.add_css(Path.base_path + u'/www/css/markdown.css')
        epub.add_css(Path.base_path + u'/www/css/customer.css')
        epub.add_css(Path.base_path + u'/www/css/normalize.css')
        epub.add_css(Path.base_path + u'/www/css/bootstrap.css')
        epub.create()

        Path.reset_path()
        return

    def create_book(command, counter):
        Path.reset_path()

        Debug.logger.info(u"開始製作第 {} 本電子書".format(counter))
        Debug.logger.info(u"對記錄 {} 進行分析".format(command))
        task_package = ReadListParser.get_task(command)  # 分析命令

        if not task_package.is_work_list_empty():
            worker_factory(task_package.work_list)  # 執行抓取程序
            Debug.logger.info(u"網頁信息抓取完畢")

        if not task_package.is_book_list_empty():
            Debug.logger.info(u"開始從數據庫中生成電子書")
            book = Book(task_package.book_list)
            book.create()
        return

    def download_img(self):
        from src.container.image_container import ImageContainer
        img_container = ImageContainer()
        img_src_dict = Match.match_img_with_src_dict(self.content)
        self.img_filename_list = []
        for img in img_src_dict:
            src = img_src_dict[img]
            filename = img_container.add(src)
            self.img_filename_list.append(filename)
            self.content = self.content.replace(img, Match.create_img_element_with_file_name(filename))

        #   下載文章封麵圖像
        filename = img_container.add(self.image_url)
        self.img_filename_list.append(filename)
        self.image_url = Match.create_local_img_src(filename)

        #   下載用戶頭像
        filename = img_container.add(self.author_avatar_url)
        self.img_filename_list.append(filename)
        self.author_avatar_url = Match.create_local_img_src(filename)

        img_container.start_download()

        #   下載完成後，更新圖片大小
        for filename in self.img_filename_list:
            self.total_img_size_kb += Path.get_img_size_by_filename_kb(filename)
        return

 def init_database():
     if Path.is_file(Path.db_path):
         DB.set_conn(sqlite3.connect(Path.db_path))
     else:
         DB.set_conn(sqlite3.connect(Path.db_path))
         with open(Path.sql_path) as sql_script:
             DB.cursor.executescript(sql_script.read())
         DB.commit()

 def getAnswerContentFromFile( answertmp = {}):
     if not answertmp:
         return
     filepath_name = Ans2File.filepath(answertmp)
     if Path.is_file(filepath_name):
         with open(filepath_name, 'r') as f:
             fileContent = f.read()
             answertmp['content'] = fileContent
     return