o <:eT@sddlmZddlZddlZddlZddlZddlZddlmZddlmZ ddl Tddl TdZ ej dd Zd ZGd d d ZdS) )datetimeN)BytesIO)Image)*z./temp/~z /Downloads/z.zipc@seZdZefddZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZddZddZd=d d!Zd"d#Zd$d%Zd>d&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd?d1d2Zd3d4Zd5d6Zd7d8Zd9d:Zd;d<Z dS)@ ImageHandlercCsdd|_d|_d|_d|_g|_g|_g|_d|_d|_d|_ g|_ g|_ ||_ d|_ d|_t|_dSN)imageIDfolderID uploadtimeoriginalfolderpathoriginalimagenameoriginalimagetypeoriginalimagestreamedittedfolderpathedittedimagenameedittedimagetypeedittedimagestreameditinghistorytemp_folderpathtemp_input_filepathtemp_output_filepathDEFAULT_SAVE_FOLDER_PATHsave_folderpath)selftemp_folder_pathrM/var/www/html/file-convertor/convertpro/convertor/imageeditor/imagehandler.py__init__s  zImageHandler.__init__cCsLd|_d|_d|_g|_g|_g|_d|_d|_d|_g|_ g|_ d|_ dSr ) r r rrrrrrrrrr)rrrr _reset_values&s zImageHandler._reset_valuescCs,tjjtjjtjjf}||rdSdS)NTF) FileFormatJPEGvaluePNGBMPlowerendswith)rfilenamesupported_formatsrrrcheck_file_extension5sz!ImageHandler.check_file_extensioncCs|dkrAtj|}||r;|tj|d|_tj|dgtj|dg|_|_ |j |_ t ddSt ddSt d dS) Nr /rrValid filetypeTInvalid filetypeFFilepath is empty) ospathbasenamer+r!dirnamersplitextrrrprint)rfilepathr)rrrcheck_file_type<s  *zImageHandler.check_file_typecCs*|dkrt|}|gkrdStddS)Nr Tz Empty Folder.F)r0listdirr5)r folderpathitemsrrrcheck_folder_dataMs  zImageHandler.check_folder_datac sdkr||ddkrt|ndt|}g}g}t}fdd|D}td||D]}||rB||q5||q5td|td|d d |}t|g} g} |D]}| |rv| |qg| |qgd d | } t| | |} d rnd |_ d d| D|_ dd| D|_ td|j td|j | gkrdnd} | | fSdS)Nr r.cs&g|]}tjtj|r|qSr)r0r1isfilejoin).0itemr9rr `s&z2ImageHandler.check_folder_type..z Filenames: zSupported files: zUnsupported files: zPThe following files have been skipped since they are not in valid file formats: z, zTThe following files have been skipped since they are already in output file format: r,cSg|] }tj|dqS)rr0r1r4r?r)rrrrB~cSrCrrDrErrrrBrFzInput filenames: zInput filetype: TF)r!strr'r0r8r5r+appendr>r(extendinputfolderpath inputfilename inputfiletype) rr9output_filetypesupported_filesunsupported_filesr: filenamesr)unsupported_files_info valid_files skipped_filesskipped_files_inforesultrrArcheck_folder_typeVsB$       zImageHandler.check_folder_typecC@|tjjkr tjjS|tjjkrtjjS|tjjkrtjjSdSN)r"r#r$ ImageFormatr%r&)r file_formatrrr_get_image_format   zImageHandler._get_image_formatcCrXrY)rZr#r$r"r%r&)r image_formatrrr_get_file_formatr]zImageHandler._get_file_formatcCs2|dks|dks|dddkrtddSdS)Nr zdata:,rzData is empty.FT)splitr5)rimagerrrcheck_image_datas"zImageHandler.check_image_datacCs|dkrB|dddd}tjjtjjtjjf}||vr<|||g|_|j|_ t d|jt ddSt d d St d d S) Nr ;r:rzOriginal filetype: r-Tr.Fr/) rarZr#r$r%r&r!r_rrr5)rdocdoc_typer*rrrcheck_doc_typeszImageHandler.check_doc_typecCst}|d}|tjkr|jddn|jd}t ||}t dd||_ ||_ td|j td|j |tjkrI|j g|_|j|_dSdS) Nz %d%m%y%H%M%Sr,rz [^a-zA-Z0-9]r zImageID:z Upload Time:)rnowstrftime FileCategoryINPUT_FOLDERPATHrrarrHupperresubr r r5 INPUT_BASE64rr)r file_categoryrjformatted_datetime first_partimage_IDrrrget_imageID_uploadtimes $     z#ImageHandler.get_imageID_uploadtimecCs$td||dkrt|SdS)NzBase64 data to convert: r )r5base64 b64decode)r base64_datarrr_convert_base64_to_bytestreams z*ImageHandler._convert_base64_to_bytestreamcCs|dkr t|dSdS)Nr utf-8)rw b64encodedecode)r bytestreamrrr_convert_bytestream_to_base64sz*ImageHandler._convert_bytestream_to_base64cCs,|dkrt|}t|}t|}|SdSr )rwrxrimgopen)rry decoded_data image_datarbrrr_convert_base64_to_images   z%ImageHandler._convert_base64_to_imagecCst||dkrT|tjjtjjtjjfvrVt}td|td|jtd||tjjkrB| d}td|jd| ||t | d}|SdSdS)Nz Input File Format: z Image Mode: zOutput File Format: RGB r{)r5rZr#r$r%r&rmoder"convertsaverwr|getvaluer})rr input_format output_format output_bufferryrrr_convert_image_to_base64s   z%ImageHandler._convert_image_to_base64Nc CsB|tjks |tjkrl|jdkrltt|jD]N}t|j|j||j|d}|j | | |j||j|fWdn1sHwY|j d|j |dd|jd|jdqt|j dS|tjkr|dkr|d d}||j|jfg|_ |j d|j d dd|jd|jddSd S) Nr rbz Uploaded rz image in z directory with image ID r<Tr`rF)rlINPUT_FILEPATHrmrrangelenrrrrrIrreadrr r5rqra)rrrrfnumfiledatarrrconvert_file_to_base64s  ",0  .z#ImageHandler.convert_file_to_base64cCs.|dr t||_dSdt||_dS)Nr<) startswithrHr'r)rrNrrr_assign_output_filetypes z$ImageHandler._assign_output_filetypec CsB|||_||_||_||_||_||_||_||_| |_ dSrY) r!r r rrrrrrr) rr r original_folderpathoriginal_filenameoriginal_filetypeoriginal_filestreameditted_filetypeeditted_filestreamediting_historyrrrassign_select_item_detailss z'ImageHandler.assign_select_item_detailscCsx|jdkr:|durd|j|_n||_td|j|dur$|j|_n |dr+|n|d|_td|jdSdS)Nr edittedimages_zEditted Filename: r,zEditted Filepath: )rr rr5rrr()routput_filenameoutput_filepathrrrdeclare_output_filepath_detailss  z,ImageHandler.declare_output_filepath_detailsc CsLt}t|dtjF}|jD]}td|dd||jd|d| |dqtd|j d|jd |j d|j d|jd Wdn1sUwY|j d krtj|j slt|j t|j |jtd }||Wdn1swYtd |jd |j d |jd d SdS)Nwz!Initiating image compression for rr<r,rzCompressed editted images in  directory to z folder in ZIP format.r wbz Downloaded TF)iorzipfileZipFile ZIP_DEFLATEDrr5writestrrrzrrrIrr0r1existsmakedirsrZIP_FILE_EXTENSIONwriter)r zip_bufferzipfstreamzip_filerrrsave_to_output_folder#s$ &   z"ImageHandler.save_to_output_folderc sN|r|jgkr|j|_|jD]n}tj|ddkrc|||dtj|dd||d<tj|dd|d<|j dtj|ddd|j d|dq|j dtj|ddd|j d|dqfd d |j D|_ t d d d d |jDdSt dddS)NrrzConverted image file named  in rz file format.z Didn't convert image file named z" directory since it is already in csg|]}qSrr)r? imagetypefiletyperrrBHsz5ImageHandler.convert_image_format..z4Editted Image Stream File Names after Conversion to z: cSsg|]}|dqSrGr)r?rrrrrBIsTz File format z is not supported.F)r+rrr0r1r4rrr\rrIrrr5)rrrrrrconvert_image_format9s   2:: z!ImageHandler.convert_image_formatcCs4|dkrt}|||||j}|SdSrY)rrrr\ getbuffernbytes)rrbrbuffer file_sizerrr_get_image_file_sizePs  z!ImageHandler._get_image_file_sizecCs\|tjjkr|S|tjjkr|dS|tjjkr|ddS|tjjkr,|dddSdS)Ni) FileSizeUnitBYTESr$ KILO_BYTES MEGA_BYTES GIGA_BYTES)r targetsizeunitrrr_target_size_in_bytesXs     z"ImageHandler._target_size_in_bytesrc Cs|jgkr |j|_|jD]}||d}|j\}} td|dtd|td| |tjkr|tjkr|r|t j krW|dt | |d|}} tdnp|t j krpt ||d| |d}} tdnWtd |d |j d |d d S|d|d}} tdn4|tjkrt ||ddt | |dd}} tdntd|d |j d|d d Std|td| d||| f}n|tjkr||tj|dd||d|dkr(td|dd|dd|dd |j d|dd|dd|dd q d } tddd D]A} t || dt | | d} } || | f}||tj|dd}|||d|dkrp| | }} |}d!} nq0| s|j d"tj|ddd#|jd$q ntd%|d |j d&|d d S||tj|dd|tj|dd|d<|j d'tj|ddd#|jd(|| fd q d!S))Nrz Filename: rzWidth: zHeight: r$z)Changed dimensions with respect to width.z*Changed dimensions with respect to height.zInvalid Dimension: r<z%Didn't resize images since Dimension is invalid.FwidthheightzChanged dimensions. percentagedz.Changed dimensions with respect to percentage.z%Invalid Resize By Image Size Option: z7Didn't resize images since Resize By Image Size Option z New Width: z New Height: r targetSizerz Filesize of z is already within  z*Didn't resize image since the filesize of _Tz!Couldn't resize image file named rz1 directory since it did not achieve target size).zInvalid Resize Option: z)Didn't resize images since Resize Option zResized image file named z directory to new size )rrrsizer5 ResizeOptions BY_IMAGESIZEResizeByImageSizeOptions BY_DIMENSIONS DimensionWIDTHroundHEIGHTrrI BY_PERCENTAGEresize BY_FILESIZErr0r1r4rrrrr\)rr$ resize_byresize_image_size_byaspect_ratio_lockresize_by_dimensionrrbrrresizedr new_width new_height resized_imagefilesizerrr resize_imagecsr           *  2(," ,88zImageHandler.resize_imagecC|tjjkrdSdS)N ClockwisezAnti-clockwise)RotationOrientation CLOCK_WISEr$)r orientationrrr_get_rotation_directionz$ImageHandler._get_rotation_directionc Cs|jgkr |j|_|tjjks|tjjkro|tjjkrtjjntjj }|jD]F}| |d}| |}| |t j|dd|t j|dd|d<|jdt j|ddd|jd||dq&dStd|d |jd |d d S) NrrzRotated image file named r directory in direction.TzInvalid Orientation: r<z.Didn't rotate images since the Orientation of rF)rrrrr$ANTI_CLOCK_WISEr Transpose ROTATE_270 ROTATE_90r transposerr0r1r4r\rrIrrr5)rrrrrbrrr rotate_images   8:zImageHandler.rotate_imagecCr)N HorizontalVertical)FlippingDirection HORIZONTALr$)r directionrrr_get_flipping_directionrz$ImageHandler._get_flipping_directionc Cs|jgkr |j|_|tjjks|tjjkrm|tjjkrtjntj}|jD]F}| |d}| |}| |t j |dd|t j |dd|d<|jdt j |ddd|jd||dq$dStd|d |jd |d d S) NrrzFlipped image file named rrrTzInvalid Direction: r<z*Didn't flip images since the Direction of rF)rrrrr$VERTICALrFLIP_LEFT_RIGHTFLIP_TOP_BOTTOMrrrr0r1r4r\rrIrrr5)rrfliprrbrrr flip_images   8:zImageHandler.flip_imagec Cs|jgkr |j|_|jD]}||d}|j\}}||krc||krc|dkrc|dkrctd||fd||||fd|jdtj |ddd|j d|dd ||fd||||fd q |dkri|nd}|dkrq|nd}||kry|n|}||kr|n|}| ||||f}| |tj |dd| tj |dd|d<|jd tj |ddd|j d ||||fdq d S) Nrz Image size z( is not compatible with cropping window r<zDidn't crop image file named rrz# directory since the image size of rzCropped image file named z directory with cropping window T)rrrrr5rrIr0r1r4rcroprr\) rlefttoprightbottomrrbrrrrr crop_images"    "N8<zImageHandler.crop_imagerY)NN)NrN)!__name__ __module__ __qualname__DEFAULT_TEMP_FOLDER_PATHr r!r+r7r;rWr\r_rcrhrvrzrrrrrrrrrrrrrrrrrrrrrrs<  2     Y r)rror0rrwrrPILrrenums serializersr r1 expanduserrrrrrrrs