level.py

import pygame
import globals
import utils

class Level:
    def __init__(self, size, platforms=None, entities=None, winFunc=None, loseFunc=None, powerupSpawnPoints=None):
        self.size = size
        self.platforms = platforms
        self.entities = entities
        self.winFunc = winFunc
        self.loseFunc = loseFunc
        self.powerupSpawnPoints = powerupSpawnPoints
    def isWon(self):
        if self.winFunc is None:
            return False
        return self.winFunc(self)
    def isLost(self):
        if self.loseFunc is None:
            return False
        return self.loseFunc(self)

# lose if no players have lives remaining
def lostLevel(level):
    # level isn't lost if any player has a life left
    for entity in level.entities:
        if entity.type == 'player':
            if entity.battle is not None:
                if entity.battle.lives > 0:
                    return False
    # level is lost otherwise
    return True

# win if no collectable items left
def wonLevel(level):
    # level isn't won if any collectable exists
    for entity in level.entities:
        if entity.type == 'collectable':
            return False
    # level isn't won otherwise
    return True

def loadLevel(levelNumber):
    if levelNumber == 1:
        # load level 1
        globals.world = Level(
            size=(3000,600),
            platforms = [
                pygame.Rect(0,580,3000,20),
                pygame.Rect(0,400,100,200),
                pygame.Rect(2900,400,100,200),
                pygame.Rect(500,500,100,20),
                pygame.Rect(600,450,100,20),
                pygame.Rect(800,450,200,20)
            ],
            entities = [
                utils.makeCoin(520,450),
                utils.makeCoin(120,530),
                utils.makeCoin(1700,550),
                utils.makeEnemy(1500,580-26)
            ],
            winFunc = wonLevel,
            loseFunc = lostLevel,
            powerupSpawnPoints = [(900,350),(200,500)]
        )
    if levelNumber == 2:
        # load level 2
        globals.world = Level(
            size=(1200,1200),
            platforms = [
                # middle
                pygame.Rect(0,300,1200,20)
            ],
            entities = [
                utils.makeCoin(100,200)
            ],
            winFunc = wonLevel,
            loseFunc = lostLevel,
            powerupSpawnPoints = [(400,260),(300,100)]
        )

    # add players
    for player in globals.players:
        globals.world.entities.append(player)     

    # reset players
    for entity in globals.world.entities:
        entity.reset(entity)