import json,os import logging from xml.dom.minidom import Document from utils.FileUtils import dbUtils as db from common.Comic import Comic from common.Constant import ComicPath from utils.Logger import logger class ComicInfoEntity: @classmethod def getNodes(cls): return [Comic.dict_chapter,Comic.dict_comic_name,Comic.dict_number,Comic.dict_comic_names, Comic.dict_dep,Comic.dict_year,Comic.dict_month,Comic.dict_day,Comic.dict_author, Comic.dict_cbs,Comic.dict_genre,Comic.dict_tags,Comic.dict_page_count, Comic.dict_language,Comic.dict_agerating,Comic.dict_pages] @classmethod def getJsonNodes(cls): return [Comic.dict_chapter,Comic.dict_comic_name,Comic.dict_icon,Comic.dict_number, Comic.dict_comic_names, Comic.dict_dep,Comic.dict_year,Comic.dict_month,Comic.dict_day,Comic.dict_author, Comic.dict_cbs,Comic.dict_genre,Comic.dict_tags,Comic.dict_page_count, Comic.dict_language,Comic.dict_agerating,Comic.dict_pages, Comic.dict_list_chapter,Comic.dict_chapter_imgs] class ComicInfo: IS_NEW_ICON = False document = Document() path_comic_info = None @classmethod def parseExec(cls,data,exec,start_add=None,item=True): if data !=None and exec != None: dots = str(exec).split(".") if not isinstance(data,dict): data = json.loads(data) for dot in dots: data = data.get(dot) if start_add != None and data != None: data = start_add+data return data @classmethod def setNodeAndValue(cls,node,value): if value != None: if isinstance(value,str): c_node = cls.document.createElement(node) child_node = cls.document.createTextNode(value) c_node.appendChild(child_node) return c_node else: return value return None #页数 @classmethod def setPages(cls,values=None): if values == None: values = Comic.getChapterFilesName() if values != None and isinstance(values,list): suffix = "."+str(values[0]).split(".")[-1] join_list=",".join(values).replace(suffix,"") values = join_list.split(",") Comic.setPageCount(len(values)+1 if cls.IS_NEW_ICON else len(values)) root_node = cls.document.createElement("Pages") if cls.IS_NEW_ICON: #添加封面 icon_node = cls.document.createElement("Page") icon_node.setAttribute("Image",ComicPath.COMIC_ICON_NAME) icon_node.setAttribute("Type","FrontCover") root_node.appendChild(icon_node) for page in values: c_node = cls.document.createElement("Page") page = page.split("_")[-1] c_node.setAttribute("Image",page) root_node.appendChild(c_node) Comic.dict_pages = Comic.setField(Comic.dict_pages,root_node,convert=False) @classmethod def getBaseUrl(cls,url=None): if url == None: url = Comic.getHomePage() (num,index) = [3,0] for x in range(0, num): index = str(url).find("/",index)+1 return url[0:index-1] #XML根文档 @classmethod def root_node(cls,root_value): return cls.document.createElement(root_value) @classmethod def add_nodes(cls,root,list_value): if len(list_value) == 0: return list_value for value in list_value: #Comic.chapter if value[0] == None and value[4]: #数据为空 value[0]， 但不允许为空value[4] = False msg = f"#数据为空 key={value[3]} value[0]={value[0]}， 但不允许为空value[4]={value[4]}" logger.error(msg) exit() if value[0] != None: root.appendChild(cls.setNodeAndValue(value[2],value[0])) @classmethod def initComicInfoXML(cls): cls.setPages() @classmethod def writeComicInfoXML(cls,overlay=False): save_path = ComicPath.getPathComicInfoXML() if os.path.exists(save_path): if overlay: os.remove(save_path) logging.info(f"已存在文件，已删除... {save_path}") else: logging.info(f"已存在文件，跳过中... {save_path}") return True cls.initComicInfoXML() root = cls.root_node("ComicInfo") new_document = Document() new_document.appendChild(root) cls.add_nodes(root,ComicInfoEntity.getNodes()) with open(save_path, "w", encoding="utf-8") as fo: new_document.writexml(fo, indent='', addindent='\t', newl='\n', encoding="utf-8") fo.close() logger.info(f"已生成文件... {save_path}") @classmethod def setComicInfo(cls,comicname=None,homepage=None,alias=None,author=None,icon=None,tags=None, dep=None,genre=None,lang=None,age_rating=None,chapters=None,current_chapter_img=None): author = ",".join(set(str(str(author).replace("&",",").replace(" ",",")).split(","))) Comic.setHomePage(homepage) Comic.setIcon(icon) Comic.setListChapter(chapters) #Comic.setUpdateAt(update_at) Comic.setComicName(str(comicname)) #if alias != None: comicInfo.setComicNames(alias) Comic.setAuthor(author) Comic.setTags(tags) Comic.setDep(dep) #comicInfo.setCBS("韩漫") if genre != None: Comic.setGenre(genre) Comic.setLanguage(lang) Comic.setAgeRating(age_rating) Comic.setCurrentChapterImg(current_chapter_img) @classmethod def writeJson(cls): dict_data = {} nodes = ComicInfoEntity.getJsonNodes() for node in nodes: key = Comic.getFieldNode(node) value = Comic.getFieldOrigin(node) if isinstance(value,list): value = ",".join(value) if key != None and isinstance(value,str): child_dict = { key : value} dict_data.update(child_dict) s = json.dumps(dict_data,ensure_ascii=True) logging.debug(f"json={s}") with open(ComicPath.getPathConfComicChapterJson(mkdir=True),"w") as fs: fs.write(s) class ComicInfoUtils: PROGRESS_DOWN = "down" PROGRESS_CBZ = "cbz" PROGRESS_DONE = "done" PROGRESS_FAIL = "fail" IS_NEW_ICON = False list_skip = [] @classmethod def setComicNameSkips(cls,value): return cls.list_skip.append(value) @classmethod def getIsComicNameSkips(cls,value): return value in ",".join(cls.list_skip) @classmethod def setProgress(cls,progress): db.set(Comic.getOriginChapterName(),progress,Comic.getOriginComicName()) @classmethod def isProgress(cls,progress): return db.query(Comic.getOriginChapterName(),progress,Comic.getOriginComicName()) #判断是否完成 @classmethod def isProgressDone(cls): return cls.isProgress(cls.PROGRESS_DONE) #设置为完成 @classmethod def setProgressDone(cls): return cls.setProgress(cls.PROGRESS_DONE) #设置未完成 @classmethod def setProgressFail(cls): return cls.setProgress(cls.PROGRESS_FAIL) @classmethod def isProgressDown(cls): return cls.isProgress(cls.PROGRESS_DOWN) @classmethod def setProgressDown(cls): return cls.setProgress(cls.PROGRESS_DOWN) @classmethod def isProgressCBZ(cls): return cls.isProgress(cls.PROGRESS_CBZ) @classmethod def setProgressCBZ(cls): return cls.setProgress(cls.PROGRESS_CBZ) @classmethod def iconDB(cls): db.set(Comic.getOriginComicName(),Comic.getIcon(),"icons") @classmethod def equIcon(cls): return db.query(Comic.getOriginComicName(),Comic.getIcon(),"icons") @classmethod def updateLastDate(cls,date=None): update_at = Comic.getUpdateAt() if date != None: update_at = date db.set(Comic.getOriginComicName(), update_at, "update") @classmethod def isUpdateComic(cls,update_at=None): if update_at == None: update_at = Comic.getUpdateAt() return db.query(Comic.getOriginComicName(), update_at,"update") @classmethod def comicChapterDownload(cls,imgs,names): Comic.setChapterImgs(imgs) ComicInfo.setPages(names) Comic.setChapterFilesName(names)