lgcosta/paperless-ngx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Support rotating image types

Browse Source
This commit is contained in:
martin 2024-11-07 04:06:48 +08:00
parent 2f06680f76
commit 304611073a
2 changed files with 34 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src-ui/src/app/components/document-detail/document-detail.component.html
View File

@ -57,7 +57,8 @@
<i-bs width="1em" height="1em" name="scissors"></i-bs>&nbsp;<span i18n>Split</span>
</button>
<button ngbDropdownItem (click)="rotateDocument()" [disabled]="!userIsOwner || originalContentRenderType !== ContentRenderType.PDF">
<button ngbDropdownItem (click)="rotateDocument()"
[disabled]="!userIsOwner || (originalContentRenderType !== ContentRenderType.PDF && originalContentRenderType !== ContentRenderType.Image)">
<i-bs name="arrow-clockwise"></i-bs>&nbsp;<ng-container i18n>Rotate</ng-container>
</button>

49
src/documents/bulk_edit.py
View File

@ -212,30 +212,45 @@ def rotate(doc_ids: list[int], degrees: int):
qs = Document.objects.filter(id__in=doc_ids)
affected_docs = []
import pikepdf
from PIL import Image
rotate_tasks = []
image_mime_types = [
"image/png",
"image/jpeg",
"image/tiff",
"image/bmp",
"image/gif",
"image/webp",
]
for doc in qs:
if doc.mime_type != "application/pdf":
is_image = doc.mime_type in image_mime_types
if doc.mime_type != "application/pdf" and not is_image:
logger.warning(
f"Document {doc.id} is not a PDF, skipping rotation.",
f"Document {doc.id} is not a PDF or image, skipping rotation.",
)
continue
try:
with pikepdf.open(doc.source_path, allow_overwriting_input=True) as pdf:
for page in pdf.pages:
page.rotate(degrees, relative=True)
pdf.save()
doc.checksum = hashlib.md5(doc.source_path.read_bytes()).hexdigest()
doc.save()
rotate_tasks.append(
update_document_archive_file.s(
document_id=doc.id,
),
)
logger.info(
f"Rotated document {doc.id} by {degrees} degrees",
)
affected_docs.append(doc.id)
if is_image:
im = Image.open(doc.source_path)
rotated = im.rotate(-degrees, expand=True)
format_from_mime = doc.mime_type.split("/")[1]
dpi = im.info.get("dpi", None)
save_kwargs = {"format": format_from_mime.upper()}
if dpi is not None:
save_kwargs["dpi"] = dpi
rotated.save(doc.source_path, **save_kwargs)
else:
with pikepdf.open(doc.source_path, allow_overwriting_input=True) as pdf:
for page in pdf.pages:
page.rotate(degrees, relative=True)
pdf.save()
doc.checksum = hashlib.md5(doc.source_path.read_bytes()).hexdigest()
doc.save()
rotate_tasks.append(update_document_archive_file.s(document_id=doc.id))
logger.info(f"Rotated document {doc.id} by {degrees} degrees")
affected_docs.append(doc.id)
except Exception as e:
logger.exception(f"Error rotating document {doc.id}: {e}")