Tips for creating a Main Menu screen, that once pressing enter

starts the game

import pygame

SCR$\_$WID, SCR$\_$HEI $=640,480$

class Player$()$:

$\_$counter $=0$

def $\_\_$init$\_\_($self$,$ x$,$ num$)$:

Player.$\_$counter $+=1$

self.id $=$ Player.$\_$counter

self.num $=$ num

self.x$,$ self.y $=$ x$,$ SCR$\_$HEI$/2$

self.speed $=3$

self.padWid, self.padHei $=8,64$

self.score $=0$

self.score$2=0$

self.scoreFont $=$ pygame.font.Font$("$imagine$\_$font.ttf$",64)$

print $($Player$.\_$counter$)$

def scoring$($self$)$:

scoreBlit $=$ self.scoreFont.render$($str$($self$.$score$),1,(255,255,255))$

screen.blit$($scoreBlit$,(32,16))$

if self.score $==10$:

print $("$player $1$ wins!"$)$

exit$()$

scoreBlit $=$ self.scoreFont.render$($str$($self$.$score$2),1,(255,255,255))$

screen.blit$($scoreBlit$,($SCR$\_$HEI$+92,16))$

if self.score$2==10$:

print $("$Player $2$ wins!"$)$

exit$()$

def movement$($self$)$:

keys $=$ pygame.key.get$\_$pressed$()$

if self.num $==1$:

if keys$[$pygame$.$K$\_$w$]$:

self.y $-=$ self.speed

elif keys$[$pygame$.$K$\_$s$]$:

self.y $+=$ self.speed

if self.y $<=0$:

self.y $=0$

elif self.y $>=$ SCR$\_$HEI$-64$:

self.y $=$ SCR$\_$HEI$-64$

if self.num $==2$:

if keys$[$pygame$.$K$\_$UP$]$:

self.y $-=$ self.speed

elif keys$[$pygame$.$K$\_$DOWN$]$:

self.y $+=$ self.speed

if self.y $<=0$:

self.y $=0$

elif self.y $>=$ SCR$\_$HEI$-64$:

self.y $=$ SCR$\_$HEI$-64$

def draw$($self$)$:

pygame.draw.rect$($screen$,(255,255,255),($self$.$x$,$ self.y$,$ self.padWid, self.padHei$))$

class Ball$()$:

def $\_\_$init$\_\_($self$)$:

self.x$,$ self.y $=$ SCR$\_$WID$/2,$ SCR$\_$HEI$/2$

self.speed$\_$x $=-3$

self.speed$\_$y $=3$

self.size $=8$

def movement$($self$)$:

self.x $+=$ self.speed$\_$x

self.y $+=$ self.speed$\_$y

#wall col

if self.y $<=0$:

self.speed$\_$y $*=-1$

elif self.y $>=$ SCR$\_$HEI$-$self.size:

self.speed$\_$y $*=-1$

if self.x $<=0$:

self.$\_\_$init$\_\_()$

player.score$2+=1$

elif self.x $>=$ SCR$\_$WID$-$self.size:

self.$\_\_$init$\_\_()$

self.speed$\_$x $=3$

player.score $+=1$

##wall col

#paddle col

#player

for n in range$(-$self.size, player.padHei$)$:

if self.y $==$ player.y $+$ n:

if self.x $<=$ player.x $+$ player.padWid:

self.speed$\_$x $*=-1$

break

n $+=1$

#enemy

for n in range$(-$self.size, enemy.padHei$)$:

if self.y $==$ enemy.y $+$ n:

if self.x $>=$ enemy.x $-$ enemy.padWid:

self.speed$\_$x $*=-1$

break

n $+=1$

##paddle col

def draw$($self$)$:

pygame.draw.rect$($screen$,(255,255,255),($self$.$x$,$ self.y$,8,8))$

SCR$\_$WID, SCR$\_$HEI $=640,480$

screen $=$ pygame.display.set$\_$mode$(($SCR$\_$WID, SCR$\_$HEI$))$

pygame.display.set$\_$caption$("$Pong$")$

pygame.font.init$()$

clock $=$ pygame.time.Clock$()$

FPS $=60$

player $=$ Player$(16,1)$

ball $=$ Ball$()$

#player $=$ Player$()$

enemy $=$ Player$(624,2)$

def main$()$:

while True:

#process

for event in pygame.event.get$()$:

if event.type $==$ pygame.QUIT:

print $("$Game exited by user"$)$

exit$()$

##process

#logic

ball.movement$()$

player.movement$()$

enemy.movement$()$

##logic

#draw

screen.fill$((0,0,0))$

ball.draw$()$

player.draw$()$

player.scoring$()$

enemy.draw$()$

##draw

#$\_\_\_\_\_\_\_$

pygame.display.flip$()$

clock.tick$($FPS$)$

main$()$