def initialize(self):
Film.objects.all().delete()
self.poland = Country.objects.get(code='PL')
self.init_films()
self.usa = Country(name='USA', code='US')
self.usa.save()
self.new_york = Town(country=self.usa,
name='New York',
has_cinemas=True)
self.new_york.save()
self.cinema_ny = Channel(type=TYPE_CINEMA,
town=self.new_york,
address='Some Street 3')
self.cinema_ny.save()
self.hbo = Channel(type=TYPE_TV_CHANNEL, country=self.usa)
self.hbo.save()
self.tvp = Channel(type=TYPE_TV_CHANNEL, country=self.poland)
self.tvp.save()
time_now = datetime.datetime.utcnow() + datetime.timedelta(0.1)
self.screen_ny1 = Screening(channel=self.cinema_ny,
film=self.fc1,
utc_time=time_now)
self.screen_ny1.save()
self.screen_ny2 = Screening(channel=self.cinema_ny,
film=self.fc2,
utc_time=time_now)
self.screen_ny2.save()
self.screen_ny3 = Screening(channel=self.cinema_ny,
film=self.fc3,
utc_time=time_now)
self.screen_ny3.save()
self.screen_ny4 = Screening(channel=self.cinema_ny,
film=self.fc4,
utc_time=time_now)
self.screen_ny4.save()
self.screen_ny5 = Screening(channel=self.cinema_ny,
film=self.fc5,
utc_time=time_now)
self.screen_ny5.save()
self.screen_hbo1 = Screening(channel=self.hbo,
film=self.fc1,
utc_time=time_now)
self.screen_hbo1.save()
self.screen_hbo2 = Screening(channel=self.hbo,
film=self.fc2,
utc_time=time_now)
self.screen_hbo2.save()
self.screen_hbo3 = Screening(channel=self.hbo,
film=self.fc3,
utc_time=time_now)
self.screen_hbo3.save()
self.screen_hbo4 = Screening(channel=self.hbo,
film=self.fc4,
utc_time=time_now)
self.screen_hbo4.save()
self.screen_hbo5 = Screening(channel=self.hbo,
film=self.fc5,
utc_time=time_now + datetime.timedelta(8))
self.screen_hbo5.save()
