124 lines
4.6 KiB
Python
124 lines
4.6 KiB
Python
#!/usr/bin/env python
|
|
|
|
from media import Media
|
|
from get_image_tag import get_exif_tag
|
|
from PIL import Image
|
|
from raw_photo import get_raw_image_dimensions, extract_jpg_thumb
|
|
from _video import Video
|
|
|
|
class ImgPreview:
|
|
def __init__(self,*args,**kwargs):
|
|
super(ImgPreview,self).__init__()
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
self.file = kwargs['file']
|
|
self.event = kwargs['event']
|
|
self.config = kwargs['config']
|
|
self.m = Media(self.file,
|
|
self.event,
|
|
self.config)
|
|
self.file_type = self.m.file_type
|
|
self.is_jpg = False
|
|
self.is_raw = False
|
|
self.path_hash = self.m.source_path_hash
|
|
self.dtc = f'{self.m.capture_date[0]}/{self.m.capture_date[1]}/{self.m.capture_date[2]}'
|
|
self.size = None
|
|
self.mpixels = None
|
|
self.thumbnail = 'thumbnail.jpg'
|
|
self.ratio = None
|
|
self.thumbnail_width = None
|
|
self.thumbnail_height = None
|
|
self.thumbnail_ratio = None
|
|
self.video_framerate = None
|
|
self.video_bit_depth = None
|
|
self.video_duration = None
|
|
self.video_encoding = None
|
|
self.video_codec = None
|
|
self.video_profile = None
|
|
self.video_pix_format = None
|
|
|
|
if self.file_type == 'image':
|
|
self._img_preview()
|
|
print(f'size: {self.size}')
|
|
print(f'dpi: {self.dpi}')
|
|
print(f'iso: {self.iso}')
|
|
print(f'lens: {self.lens}')
|
|
print(f'zoom: {self.zoom}')
|
|
print(f'camera: {self.camera}')
|
|
print(f'aperture: {self.aperture}')
|
|
print(f'mpixels: {self.mpixels}')
|
|
elif self.file_type == 'video':
|
|
self._video_preview()
|
|
|
|
self.thumb_ratio()
|
|
self.size = f'{self.width}x{self.height}'
|
|
if self.width is not None \
|
|
and self.height is not None:
|
|
self.mpixels = round((self.width * self.height) / 1000000, 1)
|
|
else:
|
|
self.mpixels = 'Unknown :('
|
|
|
|
def _img_preview(self):
|
|
self.dpi = get_exif_tag(self.file, "xresolution")
|
|
self.iso = get_exif_tag(self.file, 'iso')
|
|
self.aperture = get_exif_tag(self.file, 'fnumber')
|
|
self.camera = get_exif_tag(self.file,'cameramodelname')
|
|
if self.camera is None:
|
|
self.camera = get_exif_tag(self.file,'image model')
|
|
self.lens = get_exif_tag(self.file,'lensmodel')
|
|
self.zoom = get_exif_tag(self.file,'focallength')
|
|
|
|
if self.file.lower().endswith("jpg") \
|
|
or self.file.lower().endswith("jpeg"):
|
|
self._jpg_preview()
|
|
else:
|
|
self._raw_preview()
|
|
|
|
def _jpg_preview(self):
|
|
self.is_jpg = True
|
|
img = Image.open(self.file)
|
|
self.width = img.width
|
|
self.height = img.height
|
|
self.gen_thumb_from_jpg()
|
|
|
|
def gen_thumb_from_jpg(self):
|
|
"""Generates a thumbnail image from the given input image."""
|
|
thumb_width = 500
|
|
thumb_size = (thumb_width, int(thumb_width // 1.5))
|
|
try:
|
|
with Image.open(self.file) as img:
|
|
img.thumbnail(thumb_size)
|
|
img.save(self.thumbnail, "JPEG")
|
|
except IOError:
|
|
print(f"Error: Cannot create thumbnail for '{self.file}'")
|
|
|
|
def _raw_preview(self):
|
|
self.is_raw = True
|
|
self.width = get_raw_image_dimensions(self.file)[1]
|
|
self.height = get_raw_image_dimensions(self.file)[0]
|
|
self.thumbnail = extract_jpg_thumb(self.file)
|
|
|
|
def _video_preview(self):
|
|
vid = Video(file=self.file)
|
|
self.thumbnail = vid.gen_video_thumbnail()
|
|
video_meta = vid.get_video_meta()
|
|
self.width = video_meta['video']['size']['width']
|
|
self.height = video_meta['video']['size']['height']
|
|
self.video_framerate = round(
|
|
int(video_meta['video']['r_frame_rate'].split('/')[0]) /
|
|
int(video_meta['video']['r_frame_rate'].split('/')[1]),2)
|
|
self.video_bit_depth = video_meta['video']['bits_per_raw_sample']
|
|
self.video_duration = video_meta['video']['duration']
|
|
self.video_encoding = video_meta['video']['encoding_brand']
|
|
self.video_codec = video_meta['video']['codec_long_name']
|
|
self.video_profile = video_meta['video']['profile']
|
|
self.video_pix_format = video_meta['video']['pix_fmt']
|
|
|
|
def thumb_ratio(self):
|
|
img = Image.open(self.thumbnail)
|
|
self.thumbnail_width = img.width
|
|
self.thumbnail_height = img.height
|
|
self.thumbnail_ratio = float(self.thumbnail_width / self.thumbnail_height)
|
|
print(self.thumbnail_width)
|
|
print(self.thumbnail_height)
|
|
print(self.thumbnail_ratio) |