def main(): # Main loop while True: events = pygame.event.get() for event in events: if event.type == pygame.QUIT: pygame.quit() sys.exit() # Draw on screen screen.blit(bg, (0, 0)) draw_ground() tam_br = 8 if main_br.get_len() > 5: length = main_br.get_len() # if length branch = Branch(main_br.get_rect(length - 1).x, main_br.get_rect(length - 1).y, color=GREEN, tam=8) branch.set_direction(Branch.RIGHT) tree.append(branch) if main_br.get_len() > 5: length = main_br.get_len() branch = Branch(main_br.get_rect(length - 1).x, main_br.get_rect(length - 1).y, color=GREEN, tam=8) branch.set_direction(Branch.LEFT) tree.append(branch) # # Drawing branches and main # if main_br.get_len() in [5, 15, 25, 35, 45]: # length = main_br.get_len() # branch = Branch(main_br.get_rect(length-1).x, main_br.get_rect(length-1).y, color=GREEN, tam=8) # branch.set_direction(Branch.RIGHT) # tree.append(branch) # if main_br.get_len() in [10, 20, 30, 40, 50]: # length = main_br.get_len() # branch = Branch(main_br.get_rect(length-1).x, main_br.get_rect(length-1).y, color=GREEN, tam=8) # branch.set_direction(Branch.LEFT) # tree.append(branch) for br in tree: br.draw_branch(screen) pygame.display.update() sleep(0.1) clock.tick(30)