def create_test_models(self, iterations):
"""Creates models directly for testing.
Args:
iterations: int, The number of Answers and Questions to create.
Returns:
Two ndb.Key lists for the manually created Answers and Questions.
"""
question_keys = []
answers = []
for i in xrange(iterations):
answers.append(
survey_models.Answer.create(
text='Answer {num}'.format(num=i),
more_info_enabled=True,
placeholder_text='PlaceHolder {num}'.format(num=i)))
question_keys.append(
survey_models.Question(
question_type=survey_models.QuestionType.ASSIGNMENT,
question_text=_QUESTION.format(num=i),
enabled=True,
rand_weight=1,
answers=answers).put())
return answers, question_keys
def initialize_tables(self):
"""Performs first-time setup by creating dataset/tables."""
if constants.ON_LOCAL:
logging.debug('On local, not connecting to BQ.')
return
logging.info('Beginning BigQuery initialization.')
try:
self._dataset.create()
except cloud.exceptions.Conflict:
logging.warning(
'Dataset %s already exists, not creating.', self._dataset.name)
else:
logging.info('Dataset %s successfully created.', self._dataset.name)
self._create_table(constants.BIGQUERY_DEVICE_TABLE, device_model.Device())
self._create_table(constants.BIGQUERY_SHELF_TABLE, shelf_model.Shelf())
self._create_table(
constants.BIGQUERY_SURVEY_TABLE, survey_models.Question())
logging.info('BigQuery successfully initialized.')