def METHOD_NAME(
    self, upload_file: str, upload_stage: Optional[str] = None
) -> Optional[str]:
    upload_location = self._uploads.get(upload_file)
    if upload_location is None:
        return
    if upload_stage is not None:
        if upload_stage not in upload_location:
            return
    return upload_location["location"]