import pygame
import os
import random

black = (0,0,0)
blue = (0,0,255)
w, h = 800, 600
fps = 60

game_fold = os.path.dirname(__file__)
img_fold = os.path.join(game_fold, "img")
pl_img = pygame.image.load(os.path.join(img_fold, "player.png"))

class Player(pygame.sprite.Sprite):
    def __init__(self):
        pygame.sprite.Sprite.__init__(self)
        self.image = pl_img
        self.rect = self.image.get_rect()
        self.rect.center = (w / 2, h / 2)
        self.speedx = 0
        self.speedy = 0

    def update(self):
        self.speedx = 0
        self.speedy = 0
        keystate = pygame.key.get_pressed()
        if keystate[pygame.K_LEFT]:
            self.speedx = -5
        if keystate[pygame.K_RIGHT]:
            self.speedx = 5
            self.rect.x += self.speedx
        if self.rect.right > w:
            self.rect.right = w
        if self.rect.left < 0:
            self.rect.left = 0
        if keystate[pygame.K_DOWN]:
            self.speedy = 5
        if keystate[pygame.K_UP]:
            self.speedy = -5
        if self.rect.bottom > h:
            self.rect.bottom = h
        if self.rect.top < 0:
            self.rect.top = 0
        self.rect.x += self.speedx
        self.rect.y += self.speedy

class Mob(pygame.sprite.Sprite):
    def __init__(self):
        pygame.sprite.Sprite.__init__(self)
        self.image = pygame.Surface((50, 50))
        self.image.fill(black)
        self.rect = self.image.get_rect()
        self.rect.x = random.randrange(0, w - self.rect.width)
        self.rect.y = random.randrange(-50, -30)
        self.speedx = random.randrange(-8, 8)
        self.speedy = random.randrange(-8, 8)
    def update(self):
        self.rect.x += self.speedx
        self.rect.y += self.speedy
        if self.rect.top > h + 10 or self.rect.left < 30 or self.rect.right > w + 30:
            self.rect.x = random.randrange(w - self.rect.width)
            self.rect.y = random.randrange(-50, -30)
            self.speedy = random.randint(-8, 8)
            self.speedx = random.randint(0, 8)


       #if self.rect.right <= w:
       #    self.rect.x += 5
       #elif self.rect.right >= w:
       #    self.rect.y -= 5

       #elif self.rect.left <= w:
       #    self.rect.x += 5
       #elif self.rect.left >= w:
       #    self.rect.y += 5

       #elif self.rect.bottom <= h:
       #    self.rect.x += 5
       #elif self.rect.bottom >= h:
       #    self.rect.y += 5

       #elif self.rect.top <= h:
       #    self.rect.x += 5
       #elif self.rect.bottom >= h:
       #    self.rect.y += 5


pygame.init()
pygame.mixer.init()
screen = pygame.display.set_mode((w,h))
pygame.display.set_caption("My Game")
clock = pygame.time.Clock()

all_sprites = pygame.sprite.Group()
player = Player()
play = Mob()
all_sprites.add(player, play)


running = True
while running:
    clock.tick(fps)
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

    all_sprites.update()

    screen.fill(blue)
    all_sprites.draw(screen)

    #red = random.randint(0, 255)
    #green = random.randint(0, 255)
    #blue = random.randint(0, 255)
    #screen.fill((red, green, blue))
    pygame.display.flip()

pygame.quit()