def add_story(self, title):
"""Add a story to the Sketch.
Args:
title: The name of the view.
Raises:
ValueError: If both query_string an query_dsl are missing.
Returns:
An instance of a Story object.
"""
story = SQLStory.query.filter_by(title=title,
sketch=self.sql_sketch,
user=None).first()
if story:
return Story(story)
story = SQLStory.get_or_create(title=title,
content='[]',
sketch=self.sql_sketch,
user=None)
db_session.add(story)
db_session.commit()
return Story(story)
def _create_story(self, sketch, user):
"""Create a story in the database.
Args:
sketch: A sketch (instance of timesketch.models.sketch.Sketch)
user: A user (instance of timesketch.models.user.User)
Returns:
A story (instance of timesketch.models.story.Story)
"""
story = Story.get_or_create(
title='Test', content='Test', sketch=sketch, user=user)
self._commit_to_database(story)
return story
def _create_story(self, sketch, user):
"""Create a story in the database.
Args:
sketch: A sketch (instance of timesketch.models.sketch.Sketch)
user: A user (instance of timesketch.models.user.User)
Returns:
A story (instance of timesketch.models.story.Story)
"""
story = Story.get_or_create(
title='Test', content='Test', sketch=sketch, user=user)
self._commit_to_database(story)
return story