PGSurface2.py

import pygame
from expyriment.stimuli import Canvas

class PGSurface(Canvas):

    def get_surface_copy(self):
        return self._get_surface().copy()

    def set_surface(self, surface):
        """surface or pixel_map"""
        if isinstance(surface, pygame.PixelArray):
            return self._set_surface(surface.make_surface())
        elif isinstance(surface, pygame.Surface):
            return self._set_surface(surface)
        else:
            return False

    def get_pixel_array(self):
        return pygame.PixelArray(self.get_surface_copy())