📝 新增完整项目文档、测试套件与构建配置
This commit is contained in:
@@ -24,101 +24,109 @@ class convertPPT(object):
|
||||
"PDF" : 32,
|
||||
"XPS" : 33
|
||||
}
|
||||
__all__ = ["savetype", "trans", "close", "open"]
|
||||
|
||||
def __init__(self, file:str|Path, trans:str = "JPG") -> None:
|
||||
def __init__(self, file: str | Path, trans: str = "JPG") -> None:
|
||||
if sys.platform != "win32":
|
||||
raise SystemError("Only support Windows system.")
|
||||
if not Path(file).exists():
|
||||
raise FileNotFoundError("File not found! Please check the file path.")
|
||||
if trans.upper() not in convertPPT.TTYPES.keys():
|
||||
raise ValueError("Save type is not supported.")
|
||||
self.__file = Path(file)
|
||||
self.__saveType = (convertPPT.TTYPES[trans.upper()], trans.upper())
|
||||
if not self.__file.exists():
|
||||
raise FileNotFoundError("File not found! Please check the file path.")
|
||||
trans_upper = trans.upper()
|
||||
if trans_upper not in convertPPT.TTYPES:
|
||||
raise ValueError("Save type is not supported.")
|
||||
self.__saveType = (convertPPT.TTYPES[trans_upper], trans_upper)
|
||||
self.__inUsing = wcc.Dispatch('PowerPoint.Application')
|
||||
|
||||
def __enter__(self) -> Self:
|
||||
return self
|
||||
|
||||
def __exit__(self, exc_type, exc_value, traceback) -> None:
|
||||
self.close()
|
||||
|
||||
@property
|
||||
def savetype(self) -> str :
|
||||
def savetype(self) -> str:
|
||||
return self.__saveType[1]
|
||||
|
||||
@classmethod
|
||||
def open(cls, file:str|Path, trans:str = "JPG") -> Self :
|
||||
return cls(file, trans = trans.upper())
|
||||
def saveAs(self, saveType:str|None = None) -> None :
|
||||
if saveType not in convertPPT.TTYPES.keys():
|
||||
raise ValueError("Save type is not supported.")
|
||||
if saveType is not None:
|
||||
self.__saveType = (convertPPT.TTYPES[saveType.upper()], saveType.upper())
|
||||
else:
|
||||
def open(cls, file: str | Path, trans: str = "JPG") -> Self:
|
||||
return cls(file, trans=trans.upper())
|
||||
|
||||
def saveAs(self, saveType: str | None = None) -> None:
|
||||
if saveType is None:
|
||||
self.__saveType = (convertPPT.TTYPES["JPG"], "JPG")
|
||||
def trans(self, saveto:str|Path = ".",
|
||||
width:int|None = None) -> None :
|
||||
else:
|
||||
saveType_upper = saveType.upper()
|
||||
if saveType_upper not in convertPPT.TTYPES:
|
||||
raise ValueError("Save type is not supported.")
|
||||
self.__saveType = (convertPPT.TTYPES[saveType_upper], saveType_upper)
|
||||
|
||||
def trans(self, saveto: str | Path = ".", width: int | None = None) -> None:
|
||||
"""
|
||||
saveto : 保存路径，默认为当前路径。
|
||||
"""
|
||||
ppt = self.__inUsing.Presentations.Open(self.__file.absolute())
|
||||
if Path(saveto) == Path('.') :
|
||||
output = self.__file.absolute()
|
||||
else :
|
||||
output = Path(saveto).absolute()
|
||||
if width is not None :
|
||||
output = Path(saveto).absolute()
|
||||
if width is not None:
|
||||
ppt.Export(output, self.__saveType[1], width)
|
||||
else:
|
||||
ppt.SaveAs(output, self.__saveType[0])
|
||||
def close(self, to_console:bool = False) -> None :
|
||||
|
||||
def close(self, to_console: bool = False) -> None:
|
||||
self.__inUsing.Quit()
|
||||
if to_console :
|
||||
if to_console:
|
||||
print("File converted successfully.")
|
||||
|
||||
|
||||
def PPT_longPic(pptFile:str|Path, saveName:str|None = None, width:int|str|None = None,
|
||||
saveto:str|Path = ".") -> None :
|
||||
def PPT_longPic(pptFile: str | Path, saveName: str | None = None, width: int | str | None = None,
|
||||
saveto: str | Path = ".") -> None:
|
||||
"""
|
||||
width : 画幅宽度，可以直接指定宽度像素，也可以使用字符串数据输入百分比。
|
||||
（700； "22.1%"）
|
||||
指定为None的时候，不进行图像缩放。
|
||||
"""
|
||||
if saveName :
|
||||
sType = saveName.split(".")[-1] if "." in saveName else "JPG"
|
||||
if sType.upper() not in ["JPG", "PNG"] :
|
||||
pptFile = Path(pptFile)
|
||||
if saveName:
|
||||
sType = Path(saveName).suffix[1:].upper() if Path(saveName).suffix else "JPG"
|
||||
if sType not in ["JPG", "PNG"]:
|
||||
raise ValueError("Unable to save this type `{}` of image.".format(sType))
|
||||
else:
|
||||
sType = "JPG"
|
||||
|
||||
with TemporaryDirectory() as tmpdirname:
|
||||
with convertPPT(pptFile, trans=sType) as ppt:
|
||||
ppt.trans(saveto=tmpdirname)
|
||||
picList = list(Path(tmpdirname).glob("*.{}".format(sType)))
|
||||
with Image.open(picList[0]) as img :
|
||||
picList = sorted(Path(tmpdirname).glob("*.{}".format(sType)))
|
||||
if not picList:
|
||||
raise ValueError("No images generated from PPT conversion.")
|
||||
|
||||
with Image.open(picList[0]) as img:
|
||||
if isinstance(width, str):
|
||||
qw = float(width[:-1])/100.0
|
||||
nwidth, nheight = (int(img.width*qw), int(img.height*qw))
|
||||
qw = float(width.rstrip('%')) / 100.0
|
||||
nwidth, nheight = (int(img.width * qw), int(img.height * qw))
|
||||
elif width is None:
|
||||
nwidth, nheight = img.size
|
||||
else:
|
||||
nwidth, nheight = (width, int(img.height*width/img.width))
|
||||
nwidth, nheight = (width, int(img.height * width / img.width))
|
||||
canvas = Image.new(img.mode, (nwidth, nheight * len(picList)))
|
||||
for i in range(1,len(picList)+1) :
|
||||
with Image.open(Path(tmpdirname).joinpath("幻灯片{}.{}".format(i,sType))) as img :
|
||||
new_img = img.resize((nwidth,nheight), resample=Image.Resampling.LANCZOS)
|
||||
canvas.paste(new_img, box=(0, (i-1) * nheight))
|
||||
if Path(saveto) == Path('.') :
|
||||
filepath = Path(pptFile).parent
|
||||
else :
|
||||
filepath = Path(saveto)
|
||||
if saveName :
|
||||
if '.' in saveName :
|
||||
canvas.save(filepath.joinpath(saveName))
|
||||
else:
|
||||
canvas.save(filepath.joinpath(saveName), format=sType)
|
||||
else:
|
||||
canvas.save(filepath.joinpath(Path(pptFile).stem), format=sType)
|
||||
|
||||
def imageOptimization(imageFile:str|Path|Image.Image, saveFile:str|Path|None = None,
|
||||
max_width:int = None, max_height:int = None,
|
||||
engine:str|None = "pngquant",
|
||||
engine_conf:str|None = None) -> Image.Image|None :
|
||||
for i, picPath in enumerate(picList, 1):
|
||||
with Image.open(picPath) as img:
|
||||
new_img = img.resize((nwidth, nheight), resample=Image.Resampling.LANCZOS)
|
||||
canvas.paste(new_img, box=(0, (i - 1) * nheight))
|
||||
|
||||
filepath = pptFile.parent if Path(saveto) == Path('.') else Path(saveto)
|
||||
if saveName:
|
||||
if Path(saveName).suffix:
|
||||
canvas.save(filepath / saveName)
|
||||
else:
|
||||
canvas.save(filepath / saveName, format=sType)
|
||||
else:
|
||||
canvas.save(filepath / pptFile.name, format=sType)
|
||||
|
||||
def imageOptimization(imageFile: str | Path | Image.Image, saveFile: str | Path | None = None,
|
||||
max_width: int = None, max_height: int = None,
|
||||
engine: str | None = "pngquant",
|
||||
engine_conf: str | None = None) -> Image.Image | None:
|
||||
"""图片优化、无损压缩
|
||||
默认建议使用pngquant进行无损压缩，也可以设置为其他图片无损压缩引擎，
|
||||
不需要针对性压缩，可设定engine为None。
|
||||
@@ -126,42 +134,50 @@ def imageOptimization(imageFile:str|Path|Image.Image, saveFile:str|Path|None = N
|
||||
if isinstance(imageFile, (str, Path)):
|
||||
imageFile = Image.open(imageFile)
|
||||
img = cv2.cvtColor(np.array(imageFile), cv2.COLOR_RGB2BGR)
|
||||
if max_width or max_height :
|
||||
|
||||
if max_width or max_height:
|
||||
height, width, _ = img.shape
|
||||
oShape = (width, height)
|
||||
if max_width :
|
||||
if width > max_width :
|
||||
height = int(height * max_width / width)
|
||||
width = max_width
|
||||
if max_width < 1 :
|
||||
width = int(width * max_width)
|
||||
height = int(height * max_width)
|
||||
if max_height :
|
||||
if height > max_height :
|
||||
width = int(width * max_height / height)
|
||||
height = max_height
|
||||
if max_height < 1 :
|
||||
if width < oShape[0] :
|
||||
if height / oShape[1] > max_height :
|
||||
height = int(oShape[1] * max_height)
|
||||
width = int(oShape[0] * max_height)
|
||||
new_width, new_height = width, height
|
||||
|
||||
if max_width:
|
||||
if max_width < 1:
|
||||
new_width = int(width * max_width)
|
||||
new_height = int(height * max_width)
|
||||
elif width > max_width:
|
||||
new_width = max_width
|
||||
new_height = int(height * max_width / width)
|
||||
|
||||
if max_height:
|
||||
if max_height < 1:
|
||||
if new_width < width:
|
||||
if new_height / height > max_height:
|
||||
new_height = int(height * max_height)
|
||||
new_width = int(width * max_height)
|
||||
else:
|
||||
height = int(height * max_height)
|
||||
width = int(width * max_height)
|
||||
img = cv2.resize(img, (width, height))
|
||||
new_height = int(new_height * max_height)
|
||||
new_width = int(new_width * max_height)
|
||||
elif new_height > max_height:
|
||||
new_width = int(new_width * max_height / new_height)
|
||||
new_height = max_height
|
||||
|
||||
img = cv2.resize(img, (new_width, new_height))
|
||||
|
||||
res = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
|
||||
with TemporaryDirectory(prefix="pytoolsz") as tmpFolder :
|
||||
res.save(Path(tmpFolder)/"tmp.png", compress_level=2, quality=100)
|
||||
if engine :
|
||||
with TemporaryDirectory(prefix="pytoolsz") as tmpFolder:
|
||||
tmpPath = Path(tmpFolder) / "tmp.png"
|
||||
res.save(tmpPath, compress_level=2, quality=100)
|
||||
|
||||
if engine:
|
||||
try:
|
||||
subprocess.run([engine, engine_conf, Path(tmpFolder)/"tmp.png"], shell=True, check=True)
|
||||
subprocess.run([engine, engine_conf, tmpPath], shell=True, check=True)
|
||||
except Exception as e:
|
||||
print("未安装pngquant，不能进行图片优化压缩。\n可使用`scoop install pngquant`进行安装。")
|
||||
raise e
|
||||
else:
|
||||
res.save(Path(tmpFolder)/"tmp.png", compress_level=7, quality=95)
|
||||
if saveFile is None :
|
||||
res.save(tmpPath, compress_level=7, quality=95)
|
||||
|
||||
if saveFile is None:
|
||||
res = Image.open(lastFile(Path(tmpFolder), "*.*"))
|
||||
return res
|
||||
else :
|
||||
else:
|
||||
shutil.copyfile(lastFile(Path(tmpFolder), "*.*"), saveFile)
|
||||
Reference in New Issue
Block a user