Proper codegen for single default entrypoint

Author: igorsereda

src/dipdup/codegen.py

@@ -148,6 +148,13 @@ async def init(self, overwrite_types: bool = False, keep_schemas: bool = False)
             await self.cleanup()
         await self.verify_package()
 
+    @staticmethod
+    def __match_entrypoint_schema(entrypoint_name, entrypoint_schemas) -> dict:
+        if entrypoint_name == 'default' and len(entrypoint_schemas) == 1:
+            return entrypoint_schemas[0]['parameterSchema']
+
+        return next(ep['parameterSchema'] for ep in entrypoint_schemas if ep['name'] == entrypoint_name)
+
     async def _fetch_operation_pattern_schema(
         self,
         operation_pattern_config: PatternConfigU,
@@ -195,8 +202,9 @@ async def _fetch_operation_pattern_schema(
         entrypoint = cast(str, operation_pattern_config.entrypoint)
 
         try:
-            entrypoint_schema = next(
-                ep['parameterSchema'] for ep in contract_schemas['entrypoints'] if ep['name'] == entrypoint
+            entrypoint_schema = self.__match_entrypoint_schema(
+                entrypoint,
+                contract_schemas['entrypoints'],
             )
         except StopIteration as e:
             raise ConfigurationError(f'Contract `{contract_config.address}` has no entrypoint `{entrypoint}`') from e