使用pygame做多个触壁反弹小球

代码少，就把类文件和函数文件放一块了。颜色随机，小球个数number适当调整，在这种算法下比较吃显卡

import pygame
import sys
from pygame.sprite import Group
from pygame.sprite import Sprite
import random

#小球个数
number = 20
move_speed = 5
up_left = 7
up_right = 9
down_left = 3
down_right = 1

black = (0,0,0)

windows_width = 1200
windows_height = 600

#小球类
class Bound(Sprite):
    def __init__(self,direction,start_x,start_y,color):
        super(Bound,self).__init__()
        self.color = color
        self.start_direction = direction
        self.rect = pygame.Rect(start_x,start_y,20,20)

    #检查边界，并设置反弹
    def check_border(self):
        if self.rect.top <= 0:
            #左上
            if self.start_direction == up_left:
                self.start_direction = down_left
            #右上
            elif self.start_direction == up_right:
                self.start_direction = down_righ