import pygame
import random
from time import sleep
if __name__=='__main__':
    pygame.init()
    window = pygame.display.set_mode((400,600))
    window.fill((255,255,255))
    pygame.display.flip()
    list_balls=[]
    while True:

        if list_balls:
            for ball in list_balls:
                x,y=ball['pos']
                r=ball['re']
                index=list_balls.index(ball)
                if index<len(list_balls)-1:
                    while index+1<len(list_balls):
                        ball2=list_balls[index+1]
                        x2,y2=ball2['pos']
                        if (x2-x)**2+(y2-y)**2<=(2*r)**2:
                            if x>=x2:
                                ball['xs']=1
                                ball2['xs']=-1
                            if x<=x2:
                                ball['xs'] = -1
                                ball2['xs'] = 1

                            if y>=y2:
                                ball['ys']=-1
                                ball2['ys']=1
                            if y<=y2:
                                ball['ys']=1
                                ball2['ys']=-1
                        index +=1

                if x+r>=800:
                    xs = -1
                    ball['xs']=xs
                if x-r<=0:
                    xs = 1
                    ball['xs'] = xs
                if y+r>=600:
                    ys = -1
                    ball['ys']=ys
                if y-r<=0:
                    ys=1
                    ball['ys'] = ys
                x += ball['xs']
                y += ball['ys']
                ball['pos']=(x,y)
                pygame.draw.circle(window,ball['rgb'],(x,y),ball['re'],ball['width'])
            sleep(0.01)
            pygame.display.flip()

            window.fill((255,255,255))