84 lines
2.9 KiB
Python
84 lines
2.9 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'
|
||
|
||
@staticmethod
|
||
def get_image_thumb_and_url(media_item) -> Tuple[str, str]:
|
||
""" 获取图片的缩略图与大图的链接
|
||
"""
|
||
thumb = None
|
||
url = None
|
||
# 普通图片
|
||
if media_item.type == "2":
|
||
thumb = media_item.thumb.text
|
||
url = media_item.url.text
|
||
# 微信音乐
|
||
if media_item.type == "5":
|
||
thumb = media_item.thumb.text
|
||
url = media_item.thumb.text
|
||
|
||
return thumb, url
|
||
|
||
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:
|
||
thumb, url = self.get_image_thumb_and_url(media_item)
|
||
if thumb and url:
|
||
if download_pic:
|
||
thumb_path = self.save_image(thumb, 'thumb')
|
||
image_path = self.save_image(url, 'image')
|
||
else:
|
||
thumb_path = thumb
|
||
image_path = url
|
||
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
|