BattleShips.py

import os
os.environ['PYGAME_HIDE_SUPPORT_PROMPT'] = '1'
import pygame
import logging
from Client import Game, Constants
from Shared.Enums import STAGES
from Shared.Helpers import runFuncLogged, initLogging
from Client import Frontend

def game():
	initLogging('client_log.txt')
	pygame.fastevent.init()
	pygame.time.set_timer(pygame.event.Event(pygame.USEREVENT), Constants.ANIMATION_TIMING)

	game = Game.Game()

	clockObj = pygame.time.Clock()
	while game.gameStage != STAGES.CLOSING:
		for event in pygame.fastevent.get():
			if event.type == pygame.QUIT:
				game.quit()
			elif event.type == pygame.USEREVENT:
				game.advanceAnimations()
			elif event.type == pygame.KEYDOWN:
				assert STAGES.COUNT == 11
				if game.gameStage in [STAGES.MAIN_MENU, STAGES.MULTIPLAYER_MENU, STAGES.GAME_END, STAGES.END_GRID_SHOW]:
					game.keydownInMenu(event)
				elif event.key == pygame.K_r:
					game.rotateShip()
				elif event.key == pygame.K_q:
					game.changeCursor()
				elif event.key == pygame.K_g:
					game.toggleGameReady()
			elif event.type == pygame.MOUSEBUTTONDOWN:
				if event.button == 1:
					game.mouseClick(event.pos)
				elif event.button == 3:
					game.mouseClick(event.pos, rightClick=True)
				elif event.button == 4: # scroll up
					game.changeShipSize(+1)
				elif event.button == 5: # scroll down
					game.changeShipSize(-1)
			elif event.type == pygame.MOUSEBUTTONUP:
				if event.button == 1:
					Frontend.Runtime.windowGrabbedPos = None
			elif event.type == pygame.MOUSEMOTION:
				game.mouseMovement(event)
			elif event.type in [pygame.WINDOWFOCUSGAINED, pygame.WINDOWFOCUSLOST, pygame.WINDOWRESTORED]:
				Frontend.Runtime.windowHasFocus = event.type != pygame.WINDOWFOCUSLOST
				if not Frontend.Runtime.windowHasFocus: game.options.inputActive = False
				game.redrawNeeded = True

		game.handleConnections()
		transitionOffset = game.updateTransition()
		game.drawGame(transitionOffset)
		clockObj.tick(Constants.FPS)

def main():
	runFuncLogged(game)
	Frontend.quit()
if __name__ == '__main__':
	main()