forked from LiveCarta/BookConverter
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
import os
|
|
import pathlib
|
|
from shutil import copyfile
|
|
|
|
|
|
def process_images(body_tag, access, html_path, book_id):
|
|
"""
|
|
Function to process <img> tag. Img should be sent Amazon S3 and then return new tag with valid link.
|
|
For now images are moved to one folder.
|
|
"""
|
|
img_tags = body_tag.find_all('img')
|
|
|
|
if len(img_tags):
|
|
if access is None:
|
|
folder_path = os.path.dirname(
|
|
os.path.dirname(os.path.abspath(__file__)))
|
|
new_path = pathlib.Path(os.path.join(
|
|
folder_path, f'json/img_{book_id}/'))
|
|
new_path.mkdir(exist_ok=True)
|
|
|
|
for img in img_tags:
|
|
img_name = img.attrs.get('src')
|
|
# quick fix for bad links
|
|
if (len(img_name) >= 3) and img_name[:3] == '../':
|
|
img_name = img_name[3:]
|
|
|
|
img_path = pathlib.Path(f'{html_path.parent}', f'{img_name}')
|
|
|
|
if access is not None:
|
|
link = access.send_image(img_path, doc_id=book_id)
|
|
img.attrs['src'] = link
|
|
else:
|
|
img_size = os.path.getsize(img_path)
|
|
new_img_path = new_path / img_name
|
|
copyfile(img_path, new_img_path)
|
|
img.attrs["src"] = str(new_img_path)
|
|
|
|
return img_tags |