Exchange variable between handle_output and load_input

[dridk]

Hi,

I try to create a custom IOManager to send/load a path to a S3 storage. I want my asset return a local path and get a local path. IOManager will load the local path to the S3 storage. To create a URI endpoint, I need to extract the filename in handle_output and get back the filename from load_input.

I tried this :

@asset 
def asset1() -> Path:
    return Path("tmp/test.png")

def asset2(asset1: Path) : 
      print(asset1)

class MyIoManager:

       def handle_output(self, context, path: Path):
                    filename = os.path.basename(path)
                    context.add_input_metadata({"filename", filename)

     def load_input(self, context) -> Path:
             
               metadata = context.upstream_output.metadata
               assert "filename" is in metadata  #DOESNT WORK . How to access filename ??

I tried to exchange data between handle_output and load_input using metadata. But it doesn't work. context.upstream_output.metdata only returns data set from @asset decorator.

How to exchange data between handle_output and load_input ?