a C g%;@spddlmZddlmZer(ddlmZGdddeZGdddeZGd d d eZ Gd d d Z e Z d S)) defaultdict) TYPE_CHECKING) OptimizerBasec@seZdZdZdS)UnrecognisedOperationErrorzL Raised when the operation isn't in any of the known image classes. N__name__ __module__ __qualname____doc__r r N/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/willow/registry.pyrsrc@seZdZdZdS)UnavailableOperationErrorz Raised when all the image classes the operation exists in are not available. (most likely due to a missing image library.) Nrr r r r rsrc@seZdZdZdS)UnroutableOperationErrorzo Raised when there is no way to convert the image into an image class that supports the operation. Nrr r r r rsrc@seZdZddZddZd)ddZdd Zd d Zd d ddZddZ ddZ ddZ ddZ d*ddZ eed dddZddZgefdd Zd!d"Zd#d$Zd%d&Zd'd(ZdS)+WillowRegistrycCs.t|_i|_tt|_i|_i|_g|_dSN) set_registered_image_classes_unavailable_image_classesrdict_registered_operations_registered_converters_registered_converter_costs_registered_optimizersselfr r r __init__#s  zWillowRegistry.__init__cCs||j||<dSrr)r image_classoperation_namefuncr r r register_operation+sz!WillowRegistry.register_operationNcCs(||j||f<|dur$||j||f<dSr)rr)rfrom_image_classto_image_classr costr r r register_converter.sz!WillowRegistry.register_converterc Cs|j|z |Wn.tyF}z||j|<WYd}~n d}~00t|D]}t||}t|drz|||j |qPt|dr|j ||j d||j ddqPt|drP|j D]\}}|j ||||dqqPdS)NZ_willow_operation_willow_converter_torr)r$_willow_converter_from) raddcheck Exceptionrdirgetattrhasattrr!rr%r&r')rreattrvalZconverter_fromr$r r r register_image_class4s&        z#WillowRegistry.register_image_classcCst|dg}t|dg}t|dg}|D]}||q(|D]}||d|d|dq<|D]}||d|d|dq`dS)NZwillow_image_classesZwillow_operationsZwillow_convertersrr)r,r1r!r%)rZplugin image_classes operations convertersrZ operation converterr r r register_pluginMs    zWillowRegistry.register_pluginr)optimizer_classcCszddlm}t|dd}Wn(tyDddl}|jdd}Yn0|sNdSt|tr| dkrhdS| dkrzd}n | d }|durd}n |j |v}|r| r||j vr|j |dS) zRegisters an optimizer class.r)settingsZWILLOW_OPTIMIZERSFNfalsetrueT,)Z django.confr9r, ImportErrorosenvironget isinstancestrlowersplitZ library_nameZ check_libraryrappend)rr8r9Zenabled_optimizersr>Z add_optimizerr r r register_optimizer[s0       z!WillowRegistry.register_optimizercCs|j||Srr)rrrr r r get_operationszWillowRegistry.get_operationcCs"|jD]}||vr dSq dS)NTF)rvalues)rrZimage_class_operationsr r r operation_existsszWillowRegistry.operation_existscCs|j||fSr)rrr"r#r r r get_converterszWillowRegistry.get_convertercCs|j||fdS)Nd)rr@rJr r r get_converter_costsz!WillowRegistry.get_converter_costcsj}r:ttfdd|}|s:tdd|r|tj}|s~tdddgfdd |D|S|SdS) Ncs|jvoj|vSrr)rrwith_operationr r s z2WillowRegistry.get_image_classes..z%Could not find image class with the 'z ' operation zThe operation 'zI' is available in the following image classes but they all raised errors:c s*g|]"}dj|jtj|ddqS)z#{image_class_name}: {error_message}z Unknown error)Zimage_class_name error_message)formatrrBrr@.0rrr r s z4WillowRegistry.get_image_classes..) rcopyrfilterrrkeysrjoin)rrO availabler3Zavailable_image_classesr rNr get_image_classess:     z WillowRegistry.get_image_classes) image_formatreturncCs(g}|jD]}||r ||q |Sr)rZ applies_torE)rr] optimizersZ optimizerr r r get_optimizers_for_formats    z(WillowRegistry.get_optimizers_for_formatccs.|jD]\\}}}||ur ||fVq dS)a Yields a tuple for each image class that can be directly converted from the specified image classes. The tuple contains the converter function and the image class. For example: >>> list(registry.get_converters_from(Pillow)) [ (convert_pillow_to_wand, Wand), (save_as_jpeg, JpegFile) ... ] N)ritems)rr"Zc_fromZc_tor6r r r get_converters_fromsz"WillowRegistry.get_converters_fromc Cs||kr|gS||vrgS||jvs.||jvr2gSg}||D]<\}}||vr@||||||fg||h}||q@|S)a Returns all paths between two image classes. Each path is a list of tuples representing the steps to take in order to convert to the new class. Each tuple contains two items: The converter function to call and the class that step converts to. The order of the paths returned is undefined. For example: >>> registry.find_all_paths(JpegFile, OpenCV) [ [ (load_jpeg_into_pillow, Pillow), (convert_pillow_to_opencv, OpenCV) ], [ (load_jpeg_into_wand, Wand), (convert_wand_to_opencv, OpenCV) ] ] )rrrbfind_all_pathsunionextend) rstartendpathZ seen_classespathsr6 next_classZnewpathsr r r rcs(   zWillowRegistry.find_all_pathscCs.|}d}|D]\}}||||7}|}q |S)zE Costs up a path and returns the cost as an integer. r)rM)rrfrhZ last_classZ total_costr6rjr r r get_path_costs  zWillowRegistry.get_path_costcCsFd}d}|||D](}|||}|dus4||kr|}|}q||fS)z Finds the shortest path between two image classes. This is similar to the find_all_paths function, except it only returns the path with the lowest cost. N)rcrk)rrfrg current_path current_costrhr$r r r find_shortest_path)s z!WillowRegistry.find_shortest_pathc CsVd}d}d}|D]:}|||\}}|dur.q|dus>||kr|}|}|}q|||fS)z Finds which of the specified image classes is the closest, based on the sum of the costs for the conversions needed to convert the image into it. N)rn) rrfr3Z current_classrlrmrrhr$r r r find_closest_image_class<sz'WillowRegistry.find_closest_image_classc Csz|||}|}g}d}Wnjty|j|dd}|||\}}}|durvtd|ddd|D|j|||}Yn0||||fS) a Finds an operation that can be used by an image in the specified from_class. This function returns four values: - The operation function - The class which the operation is implemented on - A path to convert the image into the correct class for the operation - The total cost of all the conversions The path (third value) is a list of two-element tuple. Each tuple contains a function to call and a reference to the class that step converts to. See below for an example. How it works: If the specified operation_name is implemented for from_class, that is returned with an empty conversion path. If the specified operation_name is implemented on another class (but not from_class) that operation is returned with the conversion path to that new class. If it's implemented on multiple image classes, the closest one is chosen (based on the sum of the costs of each conversion step). If the operation_name is not implemented anywhere, there is no route to any image class that implements it or all the image classes that implement it are unavailable, a LookupError will be raised. Basic example: >>> func, cls, path, cost = registry.find_operation(JPEGImageFile, 'resize') >>> func PillowImage.resize >>> cls PillowImage >>> path [ (PillowImage.open, PillowImage) ] >>> cost 100 To run the found operation on an image, run each conversion function on that image then run the operation function: >>> image = Image.open(...) >>> func, cls, path, cost = registry.find_operation(type(image), operation_name) >>> for converter, new_class in path: ... image = converter(image) ... >>> func(image, *args, **kwargs) rT)rOr[Nz^The operation '{}' is available in the image class '{}' but it can't be converted to from '{}'z, css|] }|jVqdSr)rrTr r r sz0WillowRegistry.find_operation..)rG LookupErrorr\rorrSrZr)rZ from_classrr clsrhr$r3r r r find_operationSs,5    zWillowRegistry.find_operation)N)NN)rr r rr!r%r1r7rFrGrIrKrMr\rBlistr`rbrrcrkrnrorsr r r r r"s$ % 03 rN) collectionsrtypingrr_rrqrrrrregistryr r r r s