66 lines
2.4 KiB
Python
66 lines
2.4 KiB
Python
import os
|
||
import re
|
||
from typing import Tuple, Optional
|
||
|
||
from entity.moment_msg import MomentMsg, Media
|
||
import requests
|
||
import uuid
|
||
|
||
|
||
class ImageExporter:
|
||
def __init__(self, dir_name: str):
|
||
self.dir_name = dir_name
|
||
if not os.path.exists(f'output/{self.dir_name}/thumbs/'):
|
||
os.mkdir(f'output/{self.dir_name}/thumbs/')
|
||
if not os.path.exists(f'output/{self.dir_name}/images/'):
|
||
os.mkdir(f'output/{self.dir_name}/images/')
|
||
|
||
def save_image(self, url: str, img_type: str) -> str:
|
||
""" 下载图片
|
||
"""
|
||
if not (img_type == 'image' or img_type == 'thumb'):
|
||
raise Exception("img_type 参数非法")
|
||
file_name = uuid.uuid4()
|
||
response = requests.get(url)
|
||
if response.ok:
|
||
with open(f'output/{self.dir_name}/{img_type}s/{file_name}.jpg', 'wb') as file:
|
||
file.write(response.content)
|
||
return f'{img_type}s/{file_name}.jpg'
|
||
|
||
def get_images(self, msg: MomentMsg, download_pic: int) -> list[Tuple]:
|
||
""" 获取一条朋友圈的全部图像， 返回值是一个元组列表
|
||
[(缩略图路径，原图路径)，(缩略图路径，原图路径)]
|
||
"""
|
||
results = []
|
||
if not msg.timelineObject.ContentObject.mediaList:
|
||
return results
|
||
|
||
media = msg.timelineObject.ContentObject.mediaList.media
|
||
for media_item in media:
|
||
if media_item.type == "2":
|
||
if download_pic:
|
||
thumb_path = self.save_image(media_item.thumb.text, 'thumb')
|
||
image_path = self.save_image(media_item.url.text, 'image')
|
||
else:
|
||
thumb_path = media_item.thumb.text
|
||
image_path = media_item.url.text
|
||
if thumb_path and image_path:
|
||
results.append((thumb_path, image_path))
|
||
|
||
return results
|
||
|
||
def get_finder_images(self, msg: MomentMsg) -> Optional[str]:
|
||
""" 获取视频号的封面图
|
||
"""
|
||
results = None
|
||
if not msg.timelineObject.ContentObject.finderFeed:
|
||
return results
|
||
|
||
if not msg.timelineObject.ContentObject.finderFeed.mediaList:
|
||
return results
|
||
|
||
media = msg.timelineObject.ContentObject.finderFeed.mediaList.media
|
||
for media_item in media:
|
||
thumb_path = self.save_image(media_item.thumbUrl, 'thumb')
|
||
return thumb_path
|