o NDi@sLddlmZddlmZddlmZeZddZGdddeZ e Z dS) ) OrderedDict)warn) namespacecCs.|r |jjt|jidS|jjt|jdS)N)typeroptions)rr) __class____name__r)objpolicyr ]/var/www/Datamplify/venv/lib/python3.10/site-packages/cassandra/datastax/insights/registry.py_default_serializer_for_objectsr c@sNeZdZdZdddZdedfddZddZd d Zd d Z e d dZ dS)InsightsSerializerRegistryFNcs.pi|}tfdd|D|_dS)Nc3s|] }||fVqdSNr .0cls mapping_dictr r +sz6InsightsSerializerRegistry.__init__..)_class_topological_sortr _mapping_dict)selfr class_orderr rr __init__'s   z#InsightsSerializerRegistry.__init__cCsTz||dur |n|j|WSty)|tur#t||}|YS|}|YSwr)_get_serializerr Exception_NOT_SETr )rr r defaultrresultr r r serialize.s  z$InsightsSerializerRegistry.serializecCsRz|j|WSty(|jD]\}}t||r$|j|YSqYtwr)rKeyErroritems issubclass ValueError)rrregistered_cls serializerr r r r9s   z*InsightsSerializerRegistry._get_serializercs.|j|<tfddjD_dS)Nc3s|] }|j|fVqdSr)rrrr r rEsz6InsightsSerializerRegistry.register..)rrr)rrr&r r'r registerBs    z#InsightsSerializerRegistry.registercsfdd}|S)z Parameterized registration helper decorator. Given a class `cls`, produces a function that registers the decorated function as a serializer for it. cs||Sr)r()r&rrr r decoratorOs zEInsightsSerializerRegistry.register_serializer_for..decoratorr )rrr*r r)r register_serializer_forIsz2InsightsSerializerRegistry.register_serializer_forcst|g}}|r.|d|dd}tfdd|Ds%|n||}|s t|D]\}||ddD]}t|rNtdj||dq>q2|S)a A simple topological sort for classes. Takes an iterable of class objects and returns a list A of those classes, ordered such that A[X] is never a superclass of A[Y] for X < Y. This is an inefficient sort, but that's ok because classes are infrequently registered. It's more important that this be maintainable than fast. We can't use `.sort()` or `sorted()` with a custom `key` -- those assume a total ordering, which we don't have. rNc3s|]}t|VqdSr)r#)rxheadr r rgszEInsightsSerializerRegistry._class_topological_sort..zKSorting classes produced an invalid ordering. In: {classes} Out: {sorted_})classessorted_)listanyappend enumerater#rformat)r0unsortedr1tailiafter_head_valuer r.r rUs"    z2InsightsSerializerRegistry._class_topological_sortr) r __module__ __qualname__ initializedrrr rr(r+ staticmethodrr r r r r#s    rN) collectionsrwarningsr cassandra.datastax.insights.utilrobjectrr rinsights_registryr r r r s    W