class Control:
def __init__(self,
bgcolor=pygame.Color('White'),
size=(400, 400),
foodspawnrange=(200, 700)):
pygame.init()
self.window = pygame.display.set_mode(size)
self.window.fill(bgcolor)
pygame.display.set_icon(pygame.image.load('images/favicon.png'))
pygame.display.set_caption("Snake")
self.objects = []
self.clock = Clock()
self.bgcolor = bgcolor
self.eventhandler = EventHandler()
self.run = True
self.foodspawn = foodspawnrange
self.nextfood = -randint(*self.foodspawn)
self.size = size
self.pause = False
self.deaths = {}
self.events = []
def __iadd__(self, obj):
self.objects.append(obj)
return self
def __call__(self):
while self.run:
self.events = pygame.event.get()
self.eventhandler(self)
if self.run and not self.pause:
if self.nextfood >= 0:
self.nextfood = -randint(*self.foodspawn)
while 1:
new_f = Food(pos=[
randint(0, self.size[0] // 10) * 10,
randint(0, self.size[1] // 10) * 10
])
if not (new_f.pos in self.getproperty(attr='pos')):
break
self += new_f
self.nextfood += self.clock.tick()
for obj in self.objects:
obj(self)
for i in self.getproperty(type='snake', attr='pn'):
cur = self.getbyattr(type='snake', attr='pn', value=i)
self.deaths[i] = cur.deaths
pygame.display.flip()
else:
pass
def __getitem__(self, index):
return self.objects[index]
def __isub__(self, obj):
obj.destruct(self)
return self
def getproperty(self, type=None, attr='pos'):
for i in self.objects:
if (i.type == type) or (type == None):
if hasattr(i, attr):
yield getattr(i, attr)
def getbyattr(self, type=None, attr=None, value=None):
for i in self.objects:
if ((i.type == type) or (type == None)) and hasattr(
i, attr) and (getattr(i, attr) == value):
return i
return None
def getobjects(self, type=None):
for i in self.objects:
if (i.type == type) or (type == None):
yield i
def addAutoSnake(self, scriptname: str, color):
while 1:
try:
f = open(scriptname + '.py', 'r')
lines = ''.join(
f.readlines()
) + '\nControl.script = Behavior()' #+'\nautosnake.script = run'
f.close()
exec(lines)
autosnake = Snake.spawn(self,
color,
'auto' + color,
Control.script,
stdir=RIGHT)
self += autosnake
return
except FileNotFoundError as e:
errorAlert(
*traceback.format_exception(type(e), e, e.__traceback__))
scriptname = self.eventhandler.file_dialog(scriptname)
except Exception as e:
errorAlert(
*traceback.format_exception(type(e), e, e.__traceback__))
