Denys Volokh
06/09/2023, 5:51 AMclass SFTPClient(Base):
def __init__(self, block_name: str, broker_name: str = None) -> None:
self.rfs = RemoteFileSystem.load(block_name)
def load_file_data(self, remote_path: str) -> pd.DataFrame:
...
@flow(
name="Transfer files to S3 bucket",
retries=20,
retry_delay_seconds=5,
)
def flowrun(date: str = None):
context = get_run_context()
sftclient = SFTPClient(block_name="rjo-sftp", broker_name="RJO")
data = sftclient.load_file_data(remote_path="/tmp/test.csv")
if not data:
sftclient.rfs.filesystem.client.close()
sftclient = None
raise Exception("No data")