# -*- coding: utf-8 -*-
import datetime
from time import sleep
from lettuce import *
from django.utils.datastructures import SortedDict
from rapidsms.contrib.locations.models import *

from survey.features.page_objects.aggregates import AggregateStatusPage, DownloadExcelPage, InvestigatorReportPage
from survey.features.page_objects.survey_completion_rates import SurveyCompletionRatesPage
from survey.models import Survey, EnumerationArea, HouseholdMemberGroup
from survey.models.batch import Batch
from survey.models.households import Household, HouseholdMember
from survey.models.investigator import Investigator
from survey import investigator_configs


@step(u'And I have 2 batches with one open')
def and_i_have_2_batches_with_one_open(step):
    world.batch_1 = Batch.objects.create(order=1, name="Batch A", survey=world.survey_1)
    world.batch_2 = Batch.objects.create(order=2, name="Batch B", survey=world.survey_2)