Add ball which Snake can to shot

Author: GoToCoding

__pycache__/constants.cpython-35.pyc

@@ -0,0 +1,63 @@
+import copy, pygame
+from constants import *
+
+class Ball:
+	def __init__(self, screen, color, enemy):
+		self.ready = True
+		self.x = 0
+		self.y = 0
+		self.dx = 0
+		self.dy = 0
+		self.screen = screen
+		r = color[0] - 100
+		if r < 0:
+			r = 0
+		g = color[1] - 100
+		if g < 0:
+			g = 0
+		b = color[2] - 100
+		if b < 0:
+			b = 0		
+		self.color = (r, g, b)
+		self.enemy = enemy
+	
+	def shot(self, x, y, direction):
+		if self.ready != False:
+			self.x = x
+			self.y = y
+			self.ready = False
+			if direction == 'right':
+				self.dx = 1
+				self.dy = 0
+			if direction == 'left':
+				self.dx = -1
+				self.dy = 0
+			if direction == 'up':
+				self.dx = 0
+				self.dy = -1
+			if direction == 'down':
+				self.dx = 0
+				self.dy = 1
+
+			self.go()
+
+	def go(self):
+		self.x += self.dx
+		self.y += self.dy
+		if self.x < 0 or self.x >= W or self.y < 0 or self.y >= H:
+			self.ready = True
+
+		if [self.x, self.y] in self.enemy.body:
+			self.ready = False
+			if self.enemy.body[0][0] == self.x and self.enemy.body[0][1] == self.y:
+				self.enemy.isAlive = False
+			else:
+				newBody = []
+				for b in self.enemy.body:
+					if b[0] == self.x and b[1] == self.y:
+						break
+					newBody.append(b)
+				self.enemy.body = newBody
+
+	def draw(self):
+		pygame.draw.ellipse(self.screen, self.color, (self.x * blockW, self.y * blockH, blockW, blockH))

ball.py

@@ -2,6 +2,7 @@
 from snake import Snake
 from food import Food
 from constants import *
+from ball import Ball
 
 
 pygame.init()
@@ -84,6 +85,10 @@ def game():
 
     snake1 = Snake(2, 2, screen, (255, 255, 0))
     snake2 = Snake(20, 20, screen, (255, 0, 255))
+
+    snake1.ball = Ball(screen, snake1.color, snake2)
+    snake2.ball = Ball(screen, snake2.color, snake1)
+
     yummy = Food(25, screen)
 
     while True:
@@ -101,6 +106,8 @@ def game():
                     snake1.tryToDirect('left')
                 if e.key == pygame.K_RIGHT:
                     snake1.tryToDirect('right')
+                if e.key == pygame.K_RCTRL:
+                    snake1.ball.shot(snake1.body[0][0], snake1.body[0][1], snake1.direction)
 
                 #second player
                 if e.key == pygame.K_w:
@@ -111,6 +118,8 @@ def game():
                     snake2.tryToDirect('down')
                 if e.key == pygame.K_d:
                     snake2.tryToDirect('right')
+                if e.key == pygame.K_r:
+                    snake2.ball.shot(snake2.body[0][0], snake2.body[0][1], snake2.direction)
 
                 #pause
                 if e.key == pygame.K_p:
@@ -131,7 +140,16 @@ def game():
         snake2.fixDirection()
 
         fl1 = snake1.die(snake2.body)
+        if snake1.isAlive == False:
+            fl1 = False
         fl2 = snake2.die(snake1.body)
+        if snake2.isAlive == False:
+            fl2 = False
+
+        if not snake1.ball.ready:
+            snake1.ball.go()
+        if not snake2.ball.ready:
+            snake2.ball.go()
 
         snake1.go()
         snake2.go()
@@ -174,6 +192,10 @@ def game():
 
         snake1.draw()
         snake2.draw()
+        if not snake1.ball.ready:
+            snake1.ball.draw()
+        if not snake2.ball.ready:
+            snake2.ball.draw()
         yummy.draw()
 
         text = FONT_low.render('Player1 score: ' + str(snake1.score), True, (255, 255, 0))

main.py

@@ -1,5 +1,6 @@
 import copy, pygame
 from constants import *
+from ball import Ball
 
 class Snake:
     def __init__(self, x, y, screen, color):
@@ -11,6 +12,7 @@ def __init__(self, x, y, screen, color):
         self.tail = [x, y]
         self.screen = screen
         self.color = color
+        self.isAlive = True
 
     def fixDirection(self):
         if self.dx == 1: self.direction = 'right'
@@ -48,7 +50,8 @@ def die(self, enemy):
 
 
     def go(self):
-
+        if self.isAlive == False:
+            return 1
         self.tail = copy.deepcopy(self.body[-1])
         i = len(self.body) - 1
         while (i > 0):