Saket Talware

'

def extract_zip_from_sftp(sftp_host, sftp_port, sftp_username, sftp_password, zip_file_path):
    # Connect to SFTP server
    transport = paramiko.Transport((sftp_host, sftp_port))
    transport.connect(username=sftp_username, password=sftp_password)
    sftp = paramiko.SFTPClient.from_transport(transport)
    
    # Download the ZIP file into memory
    with sftp.open(zip_file_path, 'rb') as remote_zip:
        zip_data = BytesIO(remote_zip.read())
    
    # Extract ZIP file contents
    extracted_files = {}
    with zipfile.ZipFile(zip_data) as zip_ref:
        for file_name in zip_ref.namelist():
            if file_name.endswith('.csv'):  # Assuming you want to extract CSV files
                extracted_files[file_name] = zip_ref.read(file_name)
    
    # Close SFTP connection
    sftp.close()
    transport.close()
    
    return extracted_files

# Example usage
extracted_files = extract_zip_from_sftp(sftp_host, sftp_port, sftp_username, sftp_password, zip_file_path)