Function download_workspace

src/FileTransmit.py:34–63

(username: str)

@file_router.get('/download/{username}')
async def download_workspace(username: str):
    try:
        user_workspace = get_or_create_workspace(username)

        # Create a zip file from the user's workspace directory
        zip_filename = f'{username}_workspace.zip'
        zip_buffer = io.BytesIO()  # in-memory buffer to hold the zip file

        # Create a ZipFile object in write mode
        with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
            # Walk through the workspace directory and add files to the zip
            for root, dirs, files in os.walk(user_workspace):
                for file in files:
                    file_path = os.path.join(root, file)
                    arcname = os.path.relpath(file_path, user_workspace)  # Store files relative to the workspace
                    zip_file.write(file_path, arcname)

        # Seek to the beginning of the buffer before sending it
        zip_buffer.seek(0)

        # Return the zip file as a Response, without triggering stat checks
        return Response(
            zip_buffer.read(),  # Read the content of the BytesIO object
            media_type="application/zip",  # Set the media type to zip file
            headers={"Content-Disposition": f"attachment; filename={zip_filename}"}
        )
    
    except Exception as e:
        print(f"Error creating zip: {e}")
        raise HTTPException(status_code=500, detail=f"Failed to create zip file: {str(e)}")

# Route to handle file uploads with username
@file_router.post('/upload/{username}')