computer.py

import random
import tutorial
import pygame

def computer_choice(choices, bullets, shields):
    possible_choices = ["reload"]
    if choices["shield"] and bullets > 0:
        possible_choices.append("shield")
    if choices["shoot"]:
        if bullets == 0 and shields == 0:
            return "shoot"
        possible_choices.append("shoot")
    
    selected_choice = random.choice(possible_choices)
    
    return selected_choice

def cp_move(cp_move, clock, main_scene, cp_x, cp_y, background, cp_anim):
    for frame in cp_move:
        main_scene.blit(background, (0, 0))
        update_cp_anim(cp_anim, clock, main_scene, cp_x, cp_y)
        pygame.time.delay(300)

def update_cp_anim(cp_anim, clock, main_scene, cp_anim_x, cp_anim_y):
    cp_anim.update(clock.tick(60) / 1000.0)
    main_scene.blit(cp_anim.current_frame, (cp_anim_x, cp_anim_y))
    pygame.display.update()

def start_cp_anim(idle):
    cp_anim = tutorial.tutorial_animations(idle)
    cp_anim.start(3)
    return cp_anim