Box.py

import pygame

import conf


class Box(object):
 def __init__(self, screen, blockSize, x, y, bType='body'):
 self.screen = screen
 self.blockSize = blockSize
 self.direction = 2
 self.bType = bType
 self.x = x
 self.y = y
 self.color = conf.SNAKE_COLOR
 self.back = None
 self.directionChanged = 0

 def getDirection(self):
 return self.direction

 def setDirection(self, direction):
 # self.directionChanged = 1
 self.direction = direction

 def update(self):
 if self.direction == 1:
 self.y -= 1
 elif self.direction == 2:
 self.x += 1
 elif self.direction == 3:
 self.y += 1
 elif self.direction == 4:
 self.x -= 1

 def blit(self):
 relX = self.x * self.blockSize
 relY = self.y * self.blockSize
 pygame.draw.rect(self.screen, self.color, (relX, relY, self.blockSize, self.blockSize))
 if self.back is not None:
 self.back.setDirection(self.direction)