@@ -667,21 +667,21 @@ def __post_init_post_parse__(self):
667667
668668 @classmethod
669669 def format_key_import (cls , package : str , module_name : str , path : str ) -> Tuple [str , str ]:
670- key_cls = f'{ snake_to_pascal (module_name )} Key'
671- key_module = f'{ path } _key'
670+ key_cls = f'{ snake_to_pascal (path )} Key'
671+ key_module = f'{ pascal_to_snake ( path ) } _key'
672672 return f'{ package } .types.{ module_name } .big_map.{ key_module } ' , key_cls
673673
674674 @classmethod
675675 def format_value_import (cls , package : str , module_name : str , path : str ) -> Tuple [str , str ]:
676- value_cls = f'{ snake_to_pascal (module_name )} Value'
677- value_module = f'{ path } _value'
676+ value_cls = f'{ snake_to_pascal (path )} Value'
677+ value_module = f'{ pascal_to_snake ( path ) } _value'
678678 return f'{ package } .types.{ module_name } .big_map.{ value_module } ' , value_cls
679679
680680 @classmethod
681- def format_big_map_diff_argument (cls , module_name : str ) -> Tuple [str , str ]:
682- key_cls = f'{ snake_to_pascal (module_name )} Key'
683- value_cls = f'{ snake_to_pascal (module_name )} Value'
684- return module_name , f'BigMapDiff[{ key_cls } , { value_cls } ]'
681+ def format_big_map_diff_argument (cls , path : str ) -> Tuple [str , str ]:
682+ key_cls = f'{ snake_to_pascal (path )} Key'
683+ value_cls = f'{ snake_to_pascal (path )} Value'
684+ return pascal_to_snake ( path ) , f'BigMapDiff[{ key_cls } , { value_cls } ]'
685685
686686 def iter_imports (self , package : str ) -> Iterator [Tuple [str , str ]]:
687687 yield 'dipdup.context' , 'HandlerContext'
@@ -693,7 +693,7 @@ def iter_imports(self, package: str) -> Iterator[Tuple[str, str]]:
693693
694694 def iter_arguments (self ) -> Iterator [Tuple [str , str ]]:
695695 yield 'ctx' , 'HandlerContext'
696- yield self .format_big_map_diff_argument (self .contract_config . module_name )
696+ yield self .format_big_map_diff_argument (self .path )
697697
698698 @property
699699 def contract_config (self ) -> ContractConfig :
0 commit comments