def basic(self, n=1, commit=False, user=None):
if n > 1:
return [self.basic(commit=commit, user=user) for _ in range(n)]
snake = Snake(name="test", url="http://foo.bar")
if user:
UserSnake.objects.create(user=user, snake=snake)
if commit:
snake.save()
return snake
def load_snake_data(self):
if self.instance and self.instance.snake_id:
self.snake = self.instance.snake
self.initial['snake_url'] = self.instance.snake.url
else:
self.snake = Snake()