def initialize(self): self.clean_data() film1 = Film() film1.id = 1 film1.title = "Battlefield Earth" film1.type = Object.TYPE_FILM film1.permalink = "battlefield-earth-i" film1.release_year = 2010 film1.save() self.film1 = film1 film2 = Film() film2.id = 2 film2.title = "Battlefield Earth II" film2.type = Object.TYPE_FILM film2.permalink = "battlefield-earth-ii" film2.release_year = 2001 film2.save() self.film2 = film2 item = Item() item.film = film1 item.external_id = '232323' item.url_product = 'http://www.amazonka.pl/2046_wong-kar-wai,99900073479.bhtml' item.save() self.item1 = item
def initialize(self): self.clean_data() # set up users self.u1 = User.objects.create_user('michuk', '*****@*****.**', 'secret') self.u1.save() # set up film self.film = Film() self.film.id = 233 self.film.title = "Battlefield Earth II" self.film.type = Object.TYPE_FILM self.film.permalink = "battlefirld-earth-ii" self.film.release_year = 2010 self.film.save(saved_by=2) self.film.save_tags("tag1", LANG="pl", saved_by=2) # set up film self.film1 = Film() self.film1.id = 236 self.film1.title = "Battlefield Earth III" self.film1.type = Object.TYPE_FILM self.film1.permalink = "battlefirld-earth-iii" self.film1.release_year = 2011 self.film1.save(saved_by=2) self.film1.save_tags("tag2", LANG="pl", saved_by=2) # set up activities sr = UserActivity(user=self.u1, film=self.film, object=self.film, content="Lorem", activity_type=UserActivity.TYPE_SHORT_REVIEW, featured=True) sr.save() sr1 = UserActivity() sr1.user = self.u1 sr1.film = self.film1 sr1.object = self.film1 sr1.content = "ipsum" sr1.activity_type = UserActivity.TYPE_SHORT_REVIEW sr1.featured = True sr1.save() rating1 = UserActivity() rating1.activity_type = UserActivity.TYPE_RATING rating1.user = self.u1 rating1.object = self.film rating1.featured = True rating1.save() rating2 = UserActivity() rating2.activity_type = UserActivity.TYPE_RATING rating2.user = self.u1 rating2.object = self.film rating2.featured = True rating2.save()
def setUp(self): self.u1 = User.objects.create_user("user", "*****@*****.**", "user") self.u1.is_superuser = True self.u1.save() self.u2 = User.objects.create_user("user2", "*****@*****.**", "user2") self.u3 = User.objects.create_user("user3", "*****@*****.**", "user3") self.u4 = User.objects.create_user("user4", "*****@*****.**", "user4") self.f1 = Film() self.f1.title = "Battlefield Earth II" self.f1.type = Object.TYPE_FILM self.f1.permalink = "battlefirld-earth-ii" self.f1.release_year = 2010 self.f1.production_country_list = "USA" self.f1.save() self.f1.save_tags("sciencie-fiction komedia test1 test2", LANG="pl", saved_by=2) self.f2 = Film() self.f2.title = "Battlefield Earth II 2" self.f2.type = Object.TYPE_FILM self.f2.permalink = "battlefirld-earth-ii-2" self.f2.release_year = 2010 self.f2.production_country_list = "USA" self.f2.save() self.f2.save_tags("sciencie komedia test1", LANG="pl", saved_by=2) rating_helper.rate(self.u1, 1, self.f1.id) rating_helper.rate(self.u1, 2, self.f2.id) rating_helper.rate(self.u2, 3, self.f1.id) rating_helper.rate(self.u3, 3, self.f1.id) rating_helper.rate(self.u4, 3, self.f2.id)
def setUp(self): # set up films self.film1 = Film() self.film1.title = "Battlefield Earth II" self.film1.type = Object.TYPE_FILM self.film1.permalink = "battlefirld-earth-ii" self.film1.release_year = 2010 self.film1.production_country_list = "USA" self.film1.save() self.film1.save_tags("sciencie-fiction komedia test1 test2", LANG="pl", saved_by=2) self.film2 = Film() self.film2.title = "Battlefield Earth III" self.film2.type = Object.TYPE_FILM self.film2.permalink = "battlefirld-earth-iii" self.film2.release_year = 2011 self.film2.production_country_list = "USA" self.film2.save() self.film2.save_tags("sciencie-fiction komedia test1", LANG="pl", saved_by=2) self.film3 = Film() self.film3.title = "Battlefield Earth IV" self.film3.type = Object.TYPE_FILM self.film3.permalink = "battlefirld-earth-iv" self.film3.release_year = 2012 self.film3.production_country_list = "Italy" self.film3.save() self.film3.save_tags("sciencie-fiction komedia test3 test5", LANG="pl", saved_by=2) self.film4 = Film() self.film4.title = "Battlefield Earth V" self.film4.type = Object.TYPE_FILM self.film4.permalink = "battlefirld-earth-v" self.film4.release_year = 2013 self.film4.production_country_list = "UK" self.film4.save() self.film4.save_tags("sciencie-fiction komedia test5", LANG="en", saved_by=2) self.film4.save_tags("sciencie-fiction komedia", LANG="en", saved_by=2) self.film5 = Film() self.film5.title = "Battlefield Earth VI" self.film5.type = Object.TYPE_FILM self.film5.permalink = "battlefirld-earth-vi" self.film5.release_year = 2013 self.film5.production_country_list = "UK" self.film5.save() self.film5.save_tags("sciencie-fiction comedy", LANG="en", saved_by=2)
def create_films(self): self.film1 = Film() self.film1.title = "The Terminator" self.film1.type = Object.TYPE_FILM self.film1.permalink = "the-terminator" self.film1.release_year = 1984 self.film1.save() self.film2 = Film() self.film2.title = "Rambo" self.film2.type = Object.TYPE_FILM self.film2.permalink = "rambo" self.film2.release_year = 1982 self.film2.save()
def setUp( self ): # test user with no permissions self.u1 = User.objects.create_user( "user", "*****@*****.**", "user" ) # test user with moderation_permission self.u2 = User.objects.create_user( "root", "*****@*****.**", "root" ) self.u2.user_permissions.add( Permission.objects.get( codename="can_add_link" ) ) self.u2.save() # set up film self.film = Film() self.film.title = "Battlefield Earth II" self.film.type = Object.TYPE_FILM self.film.permalink = "battlefirld-earth-ii" self.film.release_year = 2010 self.film.save() # set up links self.link1 = ExternalLink( user=self.u2, film=self.film, permalink='LINK', version=1, type=ExternalLink.TYPE_LINK, status=ExternalLink.PUBLIC_STATUS, url_kind = ExternalLink.TRAILER, url="http://yt.com" ) self.link2 = ExternalLink( user=self.u2, film=self.film, permalink='LINK', version=1, type=ExternalLink.TYPE_LINK, status=ExternalLink.PUBLIC_STATUS, url_kind = ExternalLink.TRAILER, url="http://yt.com/v=1" ) self.link1.save() self.link2.save()
def test_filmindb(self): """ Test failed fetch movie in db """ self.initialize() self.client.login(username=self.user.username, password='******') film_to_import = FilmToImport(user = self.user, title = 'Elektroniczny Morderca', imdb_url = 'http://www.imdb.com/title/tt0088247/', imdb_id='0088247', status=FilmToImport.ACCEPTED) film_to_import.save() film = Film() film.title = "The Terminator" film.type = Object.TYPE_FILM film.imdb_code = "0088247" film.permalink = "the-terminator" film.release_year = 1984 film.save() run(False, False, False, False, False, True, "http") film_to_import = FilmToImport.objects.get(title = 'Elektroniczny Morderca') self.assertEqual(film_to_import.attempts, 1) self.assertEqual(film_to_import.status, FilmToImport.ALREADY_IN_DB)
def testQueuedItem(self): self.assertEqual(QueuedItem.objects.count(), 1) self.assertTrue( isinstance(QueuedItem.objects.all()[0].content_object, User)) f = Film( type=1, permalink='przypadek', imdb_code=111, status=1, version=1, \ release_year=1999, title='Przypadek' ) f.save() self.assertEqual(QueuedItem.objects.count(), 3) self.assertTrue( isinstance(QueuedItem.objects.all()[1].content_object, Film)) modified_at = QueuedItem.objects.all()[1].modified_at f.title = 'Przypadek - test' f.save() self.assertEqual(QueuedItem.objects.count(), 3) self.assertTrue(QueuedItem.objects.all()[1].modified_at > modified_at) f.delete() self.assertEqual(QueuedItem.objects.all()[1].action_type, QueuedItem.ACTION_REMOVED) self.assertEqual(QueuedItem.objects.count(), 3)
def film_from_film_on_channel(f): film = Film( title=f.title, permalink="unmatched-film-%d" % f.pk, image=None, production_country_list='', ) film._directors = f.directors return film
def initialize(self): self.clean_data() # set up users self.u1 = User.objects.create_user('michuk', '*****@*****.**', 'secret') self.u1.save() self.u2 = User(username='******', email='*****@*****.**') self.u2.save() self.u3 = User(username='******', email='*****@*****.**') self.u3.save() self.u4 = User(username='******', email='*****@*****.**') self.u4.save() # set up film self.film = Film() self.film.title = "Battlefield Earth II" self.film.type = Object.TYPE_FILM self.film.permalink = "battlefirld-earth-ii" self.film.release_year = 2010 self.film.save() self.film.save_tags('comedy') self.film1 = Film() self.film1.title = "Battlefield Earth III" self.film1.type = Object.TYPE_FILM self.film1.permalink = "battlefirld-earth-iii" self.film1.release_year = 2010 self.film1.save() self.film1.save_tags('sci-fi') self.film2 = Film() self.film2.title = "Battlefield Earth IV" self.film2.type = Object.TYPE_FILM self.film2.permalink = "battlefirld-earth-iv" self.film2.release_year = 2010 self.film2.save() self.film2.save_tags('horror') rating_helper.rate(self.u1, 8, self.film.id) rating_helper.rate(self.u1, 8, self.film1.id) rating_helper.rate(self.u1, 8, self.film2.id) rating_helper.rate(self.u2, 8, self.film2.id, _skip_activity=True)
def testFixCountries(self): from film20.import_films.update_countries import update_all_films f1 = Film(type=1, imdb_code="123", release_year=2002, title="Fake 1", production_country_list="FrancePolandGermanyUK,") f1.save() f2 = Film(type=1, imdb_code="456", release_year=2010, title="Fake 2", production_country_list="USAUKUnited Arab Emirates,") f2.save() f3 = Film(type=1, imdb_code="789", release_year=2010, title="Fake 3", production_country_list="USA") f3.save() f4 = Film(type=1, imdb_code="111", release_year=2010, title="Fake 4", production_country_list="UK, Aasdasd asdas, ,") f4.save() update_all_films() self.failUnlessEqual( Film.objects.get(pk=f1.pk).production_country_list, "France,Poland,Germany,UK") self.failUnlessEqual( Film.objects.get(pk=f2.pk).production_country_list, "USA,UK,United Arab Emirates") self.failUnlessEqual( Film.objects.get(pk=f3.pk).production_country_list, "USA") self.failUnlessEqual( Film.objects.get(pk=f4.pk).production_country_list, "UK")
def testQueueCommand(self): f = Film( type=1, permalink='przypadek', imdb_code=111, status=1, version=1, \ release_year=1999, title='Przypadek' ) f.save() f.delete() self.assertEqual(QueuedItem.objects.count(), 3) call_command('update_index_queue') self.assertEqual(QueuedItem.objects.count(), 1)
def setUp(self): self.u1 = User.objects.create_user("user1", "*****@*****.**", "user") self.u2 = User.objects.create_user("user2", "*****@*****.**", "user") self.u3 = User.objects.create_user("user3", "*****@*****.**", "user") self.f1 = Film() self.f1.title = "Battlefield Earth" self.f1.type = Film.TYPE_FILM self.f1.permalink = "battlefirld-earth" self.f1.release_year = 208 self.f1.production_country_list = "USA" self.f1.save() self.f2 = Film() self.f2.title = "Battlefield Earth II" self.f2.type = Film.TYPE_FILM self.f2.permalink = "battlefirld-earth-ii" self.f2.release_year = 2010 self.f2.production_country_list = "USA" self.f2.save()
def setUp(self): # set up films self.film1 = Film() self.film1.title = "Film 1" self.film1.type = Object.TYPE_FILM self.film1.permalink = "film-1" self.film1.release_year = 2010 self.film1.production_country_list = "USA" self.film1.save() self.film1.save_tags("dramat komedia ", LANG="pl", saved_by=2) self.film1.save_tags("drama comedy wizards ", LANG="en", saved_by=2)
def initialize(self): self.clean_data() film = Film() film.title = "The Terminator" film.type = Object.TYPE_FILM film.imdb_code = "0088247" film.permalink = "the-terminator" film.release_year = 1984 film.save() self.film = film
def test_watching_notification(self): """ Testing notifications for different types of activities (posts, short reviews and threads) """ self.initialize() activity = UserActivity.objects.get_for_object(self.post1) # create two comments for blog post by two different users t = ThreadedComment(user=self.u2, comment="This is a comment", content_object=activity, type=Object.TYPE_COMMENT) t.save() t = ThreadedComment(user=self.u3, comment="This is also a comment", content_object=activity, type=Object.TYPE_COMMENT) t.save() # create film and a short review and a comment to it f1 = Film(type=1, permalink='przypadek', imdb_code=111, status=1, version=1, release_year=1999, title='Przypadek', popularity=1, popularity_month=1) f1.save() sr = ShortReview(user=self.u1, review_text="This is a short review", object=f1, type=Object.TYPE_SHORT_REVIEW) sr.save() activity2 = UserActivity.objects.get_for_object(sr) t = ThreadedComment(user=self.u2, comment="This is a comment to a short review", content_object=activity2, type=Object.TYPE_COMMENT) t.save() # assert all is fine watching_objects = Watching.objects.all() self.assertEquals( len(watching_objects), 5 ) # number of unique comments in threads should be equal to watching objects reply_notices = notification.Notice.objects.filter( notice_type__label='reply') # TODO: uncommend and fix in http://jira.filmaster.org/browse/FLM-1116 self.assertEquals(len(reply_notices), 4) # number of notices
def setUp( self ): super( TaggingToolsTestCase, self ).setUp() # .. self.admin = User.objects.create_user( "admin", "*****@*****.**", "admin" ) self.admin.is_superuser = True self.admin.save() self.client = Client( follow=True ) self.rename_tag = registry.get_by_name( 'rename-tag' ) self.alias_tag = registry.get_by_name( 'alias-tag' ) self.film1 = Film() self.film1.title = "Battlefield Earth II" self.film1.type = Object.TYPE_FILM self.film1.permalink = "battlefirld-earth-ii" self.film1.release_year = 2010 self.film1.production_country_list = "USA" self.film1.save() self.film1.save_tags( "komedia, kosmos, test" ) self.film2 = Film() self.film2.title = "Battlefield Earth III" self.film2.type = Object.TYPE_FILM self.film2.permalink = "battlefirld-earth-iii" self.film2.release_year = 2011 self.film2.production_country_list = "USA" self.film2.save() self.film2.save_tags( "komedia, kosmos, test" ) self.film3 = Film() self.film3.title = "Battlefield Earth IV" self.film3.type = Object.TYPE_FILM self.film3.permalink = "battlefirld-earth-iv" self.film3.release_year = 2012 self.film3.production_country_list = "Italy" self.film3.save() self.film3.save_tags( "komedia romantyczna, kosmos, test" ) self.tag1 = Tag.objects.create( name='ciekawy' )
def initialize(self): self.clean_data() # set up users self.u1= User.objects.create_user('michuk', '*****@*****.**', 'secret') self.u1.save() # set up film self.film = Film() self.film.title = "Battlefield Earth II" self.film.type = Object.TYPE_FILM self.film.permalink = "battlefirld-earth-ii" self.film.release_year = 2010 self.film.save()
def initialize(self): self.clean_data() # set up film self.film = Film() self.film.title = "Battlefield Earth II" self.film.type = Film.TYPE_FILM self.film.permalink = "battlefirld-earth-ii" self.film.release_year = 2010 self.film.save() filename = "First_blood_poster.jpg" myfile = MEDIA_ROOT + "test/First_blood_poster.jpg" f = open(myfile, 'rb') myfile = File(f) self.film.image.save(filename, myfile)
def test_save_searchkeys(self): """ Test saving searchkeys for localized film """ self.film = Film() self.film.title = u"Olympia - Fest der Schönheit" self.film.type = Object.TYPE_FILM self.film.permalink = "olympia-2-teil-fest-der-schoenheit" self.film.release_year = 1938 self.film.save() fetch_synopses_for_films([ self.film, ], fetcher="tmdb") loc_film = FilmLocalized.objects.all()[0] self.assertEquals(loc_film.description is None, True) self.assertEquals(loc_film.fetched_description is None, False)
def setUp(self): self.user = User.objects.get(username='******') self.user.set_password('bob') self.user.save() self.film = Film(type=1, permalink='przypadek', imdb_code=111, status=1, version=1, release_year=1999, title='Przypadek', popularity=1, popularity_month=1) self.film.save() self.person = Person(type=2, permalink='jerzy-stuhr', imdb_code=112, status=1, version=1, name='Jerzy', surname='Stuhr') self.person.save()
def setUp(self): f1 = Film(type=1, permalink='the-alien', imdb_code=111, status=1, version=1, release_year=1979, title='The Alien', popularity=1, popularity_month=1) f1.save() f2 = Film(type=1, permalink='the-shawshank-redemption', imdb_code=112, status=1, version=1, release_year=1994, title='The Shawshank Redemption', popularity=1, popularity_month=1) f2.save() f3 = Film(type=1, permalink='terminator-2-judgment-day', imdb_code=113, status=1, version=1, release_year=1991, title='Terminator 2: Judgment Day', popularity=1, popularity_month=1) f3.save() f4 = Film(type=1, permalink='american-history-x', imdb_code=114, status=1, version=1, release_year=1998, title='American History X', popularity=1, popularity_month=1) f4.save() f5 = Film(type=1, permalink='the-big-lebowski', imdb_code=115, status=1, version=1, release_year=1998, title='The Big Lebowski', popularity=1, popularity_month=1) f5.save() f6 = Film(type=1, permalink='the-goonies', imdb_code=116, status=1, version=1, release_year=1985, title='The Goonies', popularity=1, popularity_month=1) f6.save() f7 = Film(type=1, permalink='the-lord-of-the-rings-the-fellowship-of-the-ring', imdb_code=117, status=1, version=1, release_year=2001, title='The Lord of the Rings: Fellowship of the Ring', popularity=1, popularity_month=1) f7.save() f8 = Film(type=1, permalink='nine', imdb_code=118, status=1, version=1, release_year=2009, title='Nine', popularity=1, popularity_month=1) f8.save() f10 = Film(type=1, permalink='xiao-cai-feng', imdb_code=119, status=1, version=1, release_year=2002, title='Xiao cai feng', popularity=1, popularity_month=1) f10.save() f11 = Film(type=1, permalink='bebes', imdb_code=120, status=1, version=1, release_year=2010, title='Bebe(s)', popularity=1, popularity_month=1) f11.save() f12 = Film(type=1, permalink='rope', imdb_code=121, status=1, version=1, release_year=1948, title='Rope', popularity=1, popularity_month=1) f12.save() f13 = Film(type=1, permalink='larry-crowne', imdb_code=122, status=1, version=1, release_year=2011, title='Larry Crowne', popularity=1, popularity_month=1) f13.save() f14 = Film(type=1, permalink='we-live-in-public', imdb_code=123, status=1, version=1, release_year=2009, title='We live in public', popularity=1, popularity_month=1) f14.save() f9 = Film(type=1, permalink='krotki-film-o-zabijaniu', imdb_code=124, status=1, version=1, release_year=1980, title='Krotki film o zabijaniu', popularity=1, popularity_month=1) f9.save() self.u1 = User(username='******', email='*****@*****.**') self.u1.save()
def setUp(self): User.objects.all().delete() Film.objects.all().delete() Rating.objects.all().delete() ShortReview.objects.all().delete() #create sample user user1 = User.objects.create_user('michuk', '*****@*****.**', 'secret') user1.save() user2 = User.objects.create_user('john', '*****@*****.**', 'secret') user2.save() tags = "science-fiction comedy" #create sample user rating film1 = Film() film1.title = "Battlefield Earth II" film1.type = Object.TYPE_FILM film1.permalink = "battlefirld-earth-ii" film1.release_year = 2010 film1.production_country_list = "USA" film1.save() film1.save_tags(tags, LANG="pl", saved_by=2) film2 = Film() film2.title = "Battlefield Earth III" film2.type = Object.TYPE_FILM film2.permalink = "battlefirld-earth-iii" film2.release_year = 2011 film2.production_country_list = "USA" film2.save() film2.save_tags(tags, LANG="pl", saved_by=2) film3 = Film() film3.title = "Battlefield Earth IV" film3.type = Object.TYPE_FILM film3.permalink = "battlefirld-earth-iv" film3.release_year = 2012 film3.production_country_list = "Italy" film3.save() film3.save_tags(tags, LANG="pl", saved_by=2) film4 = Film() film4.title = "Battlefield Earth V" film4.type = Object.TYPE_FILM film4.permalink = "battlefirld-earth-v" film4.release_year = 2013 film4.production_country_list = "UK" film4.save() film4.save_tags(tags, LANG="pl", saved_by=2) # set up test user ratings save_rating(film1, user1, 1) save_rating(film2, user1, 3) save_rating(film3, user1, 4) save_rating(film4, user1, 6) save_rating(film2, user2, 2) save_rating(film3, user2, 3) save_rating(film4, user2, 5) users = User.objects.all() self.assertEquals(len(users), 2) films = Film.objects.all() self.assertEquals(len(films), 4) user1 = users[0].id ratings1 = Rating.objects.filter(Q(user=user1)) self.assertEquals(len(ratings1), 4) user2 = users[1].id ratings2 = Rating.objects.filter(Q(user=user2)) self.assertEquals(len(ratings2), 3) reviews1 = ShortReview.objects.filter(Q(user=user1)) self.assertEquals(len(reviews1), 0) reviews2 = ShortReview.objects.filter(Q(user=user2)) self.assertEquals(len(reviews2), 0)
def testImdbCodes(self): from film20.import_films.update_imdb_codes import update_all_imdb_codes f1 = Film(type=1, imdb_code="199123", release_year=2008, title="The Incredible Hulk") f1.save() f2 = Film(type=1, imdb_code="333332", release_year=1992, title="The Godfather Trilogy: 1901-1980") f2.save() f3 = Film(type=1, imdb_code="343454", release_year=2007, title="Youth Without Youth") f3.save() f4 = Film(type=1, imdb_code="343sd4", release_year=1007, title="Fake Movie") f4.save() f5 = Film(type=1, imdb_code="0110413", release_year=1994, title=u"Léon") f5.save() f6 = Film(type=1, imdb_code="0289765", release_year=2002, title="Red Dragon") f6.save() call_command('fix_films_imdb_codes') self.failUnlessEqual(Film.objects.get(pk=f1.pk).imdb_code, "0800080") self.assertTrue(Film.objects.get(pk=f1.pk).verified_imdb_code) self.failUnlessEqual(Film.objects.get(pk=f2.pk).imdb_code, "0150742") self.assertTrue(Film.objects.get(pk=f2.pk).verified_imdb_code) self.failUnlessEqual(Film.objects.get(pk=f3.pk).imdb_code, "0481797") self.assertTrue(Film.objects.get(pk=f3.pk).verified_imdb_code) self.failUnlessEqual(Film.objects.get(pk=f4.pk).imdb_code, "343sd4") self.assertFalse(Film.objects.get(pk=f4.pk).verified_imdb_code) self.assertEqual( Film.objects.get(pk=f4.pk).import_comment, "nothing matches") self.failUnlessEqual(Film.objects.get(pk=f5.pk).imdb_code, "0110413") self.assertTrue(Film.objects.get(pk=f5.pk).verified_imdb_code) self.failUnlessEqual(Film.objects.get(pk=f6.pk).imdb_code, "0289765") self.assertFalse(Film.objects.get(pk=f6.pk).verified_imdb_code) self.assertEqual( Film.objects.get(pk=f6.pk).import_comment, "too many matching movies: '0289765,0379467'")
def init_films(self): self.f1 = Film(type=1, permalink='przypadek', imdb_code=111, status=1, version=1, release_year=1999, title='Przypadek', popularity=20, popularity_month=10) self.f1.save() self.fc1 = FilmOnChannel(title='fc1', film=self.f1) self.fc1.save() self.fr1 = FilmRanking(film=self.f1, type=Rating.TYPE_FILM, average_score=Decimal('1.0'), number_of_votes=100) self.fr1.save() self.f2 = Film(type=1, permalink='wrestler', imdb_code=112, status=1, version=1, release_year=2008, title='Wrestler', popularity=10, popularity_month=1) self.f2.save() self.fc2 = FilmOnChannel(title='fc2', film=self.f2) self.fc2.save() self.fr2 = FilmRanking(film=self.f2, type=Rating.TYPE_FILM, average_score=Decimal('2.0'), number_of_votes=100) self.fr2.save() self.f3 = Film(type=1, permalink='american-history-x', imdb_code=113, status=1, version=1, release_year=1998, title='American History X', popularity=1, popularity_month=1) self.f3.save() self.fc3 = FilmOnChannel(title='fc3', film=self.f3) self.fc3.save() self.fr3 = FilmRanking(film=self.f3, type=Rating.TYPE_FILM, average_score=Decimal('3.0'), number_of_votes=100) self.fr3.save() self.f4 = Film(type=1, permalink='the-big-lebowski', imdb_code=114, status=1, version=1, release_year=1998, title='The Big Lebowski', popularity=1, popularity_month=1) self.f4.save() self.fc4 = FilmOnChannel(title='fc4', film=self.f4) self.fc4.save() self.fr4 = FilmRanking(film=self.f4, type=Rating.TYPE_FILM, average_score=Decimal('4.0'), number_of_votes=100) self.fr4.save() self.f5 = Film( type=1, permalink='the-lord-of-the-rings-the-fellowship-of-the-ring', imdb_code=115, status=1, version=1, release_year=2001, title='The Lord of the Rings: The Fellowship of the Ring', popularity=1, popularity_month=1) self.f5.save() self.fc5 = FilmOnChannel(title='fc5', film=self.f5) self.fc5.save() self.fr5 = FilmRanking(film=self.f5, type=Rating.TYPE_FILM, average_score=Decimal('5.0'), number_of_votes=100) self.fr5.save() self.f6 = Film(type=1, permalink='raiders-of-the-lost-ark', imdb_code=116, status=1, version=1, release_year=1981, title='Raiders of the Lost Ark', popularity=1, popularity_month=1) self.f6.save() self.fc6 = FilmOnChannel(title='fc6', film=self.f6) self.fc6.save() self.fr6 = FilmRanking(film=self.f6, type=Rating.TYPE_FILM, average_score=Decimal('6.0'), number_of_votes=100) self.fr6.save() self.f7 = Film(type=1, permalink='the-alien', imdb_code=117, status=1, version=1, release_year=1979, title='The Alien', popularity=1, popularity_month=1) self.f7.save() self.fc7 = FilmOnChannel(title='fc7', film=self.f7) self.fc7.save() self.fr7 = FilmRanking(film=self.f7, type=Rating.TYPE_FILM, average_score=Decimal('7.0'), number_of_votes=100) self.fr7.save() self.f8 = Film(type=1, permalink='terminator-2-judgment-day', imdb_code=118, status=1, version=1, release_year=1991, title='Terminator 2: Judgment Day', popularity=1, popularity_month=1) self.f8.save() self.fc8 = FilmOnChannel(title='fc8', film=self.f8) self.fc8.save() self.fr8 = FilmRanking(film=self.f8, type=Rating.TYPE_FILM, average_score=Decimal('8.0'), number_of_votes=100) self.fr8.save() self.f9 = Film(type=1, permalink='pulp-fiction', imdb_code=119, status=1, version=1, release_year=1991, title='Pulp Fiction', popularity=1, popularity_month=1) self.f9.save() self.f9.save_tags('pulp')
def initialize(self): self.clean_data() tags = "sciencie-fiction comedy" # set up test user self.user1 = User.objects.create_user("user1", "*****@*****.**", "secret") self.user1.save() # set up films self.film1 = Film() self.film1.title = "Battlefield Earth II" self.film1.type = Object.TYPE_FILM self.film1.permalink = "battlefirld-earth-ii" self.film1.release_year = 2010 self.film1.production_country_list = "USA" self.film1.save() self.film1.save_tags(tags, LANG="pl", saved_by=2) self.film2 = Film() self.film2.title = "Battlefield Earth III" self.film2.type = Object.TYPE_FILM self.film2.permalink = "battlefirld-earth-iii" self.film2.release_year = 2011 self.film2.production_country_list = "USA" self.film2.save() self.film2.save_tags(tags, LANG="pl", saved_by=2) self.film3 = Film() self.film3.title = "Battlefield Earth IV" self.film3.type = Object.TYPE_FILM self.film3.permalink = "battlefirld-earth-iv" self.film3.release_year = 2012 self.film3.production_country_list = "Italy" self.film3.save() self.film3.save_tags(tags, LANG="pl", saved_by=2) self.film4 = Film() self.film4.title = "Battlefield Earth V" self.film4.type = Object.TYPE_FILM self.film4.permalink = "battlefirld-earth-v" self.film4.release_year = 2013 self.film4.production_country_list = "UK" self.film4.save() self.film4.save_tags(tags, LANG="pl", saved_by=2) # set up filmrankings self.filmranking1 = FilmRanking() self.filmranking1.film = self.film1 self.filmranking1.type = Rating.TYPE_FILM self.filmranking1.average_score = Decimal('8.0') self.filmranking1.number_of_votes = 80 self.filmranking1.save() self.filmranking2 = FilmRanking() self.filmranking2.film = self.film2 self.filmranking2.type = Rating.TYPE_FILM self.filmranking2.average_score = Decimal('7.0') self.filmranking2.number_of_votes = 70 self.filmranking2.save() self.filmranking3 = FilmRanking() self.filmranking3.film = self.film3 self.filmranking3.type = Rating.TYPE_FILM self.filmranking3.average_score = Decimal('6.0') self.filmranking3.number_of_votes = 60 self.filmranking3.save() self.filmranking4 = FilmRanking() self.filmranking4.film = self.film4 self.filmranking4.type = Rating.TYPE_FILM self.filmranking4.average_score = Decimal('5.0') self.filmranking4.number_of_votes = 2 self.filmranking4.save() # save compared objects in database self.compared_film1 = FilmComparator() self.compared_film1.main_film = self.film1 self.compared_film1.compared_film = self.film1 self.compared_film1.score = 10 self.compared_film1.save() self.compared_film2 = FilmComparator() self.compared_film2.main_film = self.film1 self.compared_film2.compared_film = self.film2 self.compared_film2.score = 9 self.compared_film2.save() self.compared_film3 = FilmComparator() self.compared_film3.main_film = self.film1 self.compared_film3.compared_film = self.film3 self.compared_film3.score = 8 self.compared_film3.save() self.compared_film4 = FilmComparator() self.compared_film4.main_film = self.film1 self.compared_film4.compared_film = self.film4 self.compared_film4.score = 7 self.compared_film4.save() # set up test user ratings self.user1rating1 = Rating() self.user1rating1.type = Rating.TYPE_FILM self.user1rating1.user = self.user1 self.user1rating1.film = self.film1 self.user1rating1.parent = self.film1 self.user1rating1.rating = 10 self.user1rating1.save() self.user1rating2 = Rating() self.user1rating2.type = Rating.TYPE_FILM self.user1rating2.user = self.user1 self.user1rating2.film = self.film2 self.user1rating2.parent = self.film2 self.user1rating2.rating = 9 self.user1rating2.save() self.user1rating3 = Rating() self.user1rating3.type = Rating.TYPE_FILM self.user1rating3.user = self.user1 self.user1rating3.film = self.film3 self.user1rating3.parent = self.film3 self.user1rating3.rating = 8 self.user1rating3.save() self.user1rating4 = Rating() self.user1rating4.type = Rating.TYPE_FILM self.user1rating4.user = self.user1 self.user1rating4.film = self.film4 self.user1rating4.parent = self.film4 self.user1rating4.rating = 7 self.user1rating4.save()
def setUp(self): User.objects.all().delete() Object.objects.all().delete() Film.objects.all().delete() FilmLog.objects.all().delete() Rating.objects.all().delete() Recommendation.objects.all().delete() Person.objects.all().delete() AddedCharacter.objects.all().delete() AddedFilm.objects.all().delete() # set up users self.u1 = User(username='******', email='*****@*****.**') self.u1.save() self.u2 = User.objects.create_user("root", "*****@*****.**", "root") self.u2.user_permissions.add( Permission.objects.get(codename="can_accept_added_films")) self.u2.save() self.u3 = User(username='******', email='*****@*****.**') self.u3.save() self.u4 = User(username='******', email='*****@*****.**') self.u4.save() self.u5 = User(username='******', email='*****@*****.**') self.u5.save() self.u6 = User(username='******', email='*****@*****.**') self.u6.save() # some persons self.p1 = Person(name="Clint", surname="Eastwood", imdb_code=None, type=Person.TYPE_PERSON) self.p1.save() self.p2 = Person(name="Sylvester", surname="Stallone", imdb_code=None, type=Person.TYPE_PERSON) self.p2.save() self.p3 = Person(name="Jack", surname="Mort", imdb_code=None, type=Person.TYPE_PERSON) self.p3.save() # set up films self.f1 = Film(type=1, permalink='przypadek', imdb_code=111, status=1, version=1, release_year=1999, title='Przypadek', popularity=20, popularity_month=10) self.f1.save() self.f2 = Film(type=1, permalink='wrestler', imdb_code=112, status=1, version=1, release_year=2008, title='Wrestler', popularity=10, popularity_month=1) self.f2.save() self.f3 = Film(type=1, permalink='american-history-x', imdb_code=113, status=1, version=1, release_year=1998, title='American History X', popularity=1, popularity_month=1) self.f3.save() self.f4 = Film(type=1, permalink='the-big-lebowski', imdb_code=114, status=1, version=1, release_year=1998, title='The Big Lebowski', popularity=1, popularity_month=1) self.f4.save() self.f5 = Film( type=1, permalink='the-lord-of-the-rings-the-fellowship-of-the-ring', imdb_code=115, status=1, version=1, release_year=2001, title='The Lord of the Rings: The Fellowship of the Ring', popularity=1, popularity_month=1) self.f5.save() self.f6 = Film(type=1, permalink='raiders-of-the-lost-ark', imdb_code=116, status=1, version=1, release_year=1981, title='Raiders of the Lost Ark', popularity=1, popularity_month=1) self.f6.save() self.f7 = Film(type=1, permalink='the-alien', imdb_code=117, status=1, version=1, release_year=1979, title='The Alien', popularity=1, popularity_month=1) self.f7.save() self.f8 = Film(type=1, permalink='terminator-2-judgment-day', imdb_code=118, status=1, version=1, release_year=1991, title='Terminator 2: Judgment Day', popularity=1, popularity_month=1) self.f8.save() self.f9 = Film(type=1, permalink='przypadek', imdb_code=119, status=1, version=1, release_year=1999, title='Przypadek', popularity=20, popularity_month=10) self.f9.save() self.f10 = Film(type=1, permalink='wrestler', imdb_code=120, status=1, version=1, release_year=2008, title='Wrestler', popularity=10, popularity_month=1) self.f10.save() self.f11 = Film(type=1, permalink='american-history-x', imdb_code=121, status=1, version=1, release_year=1998, title='American History X', popularity=1, popularity_month=1) self.f11.save() self.f12 = Film(type=1, permalink='the-big-lebowski', imdb_code=122, status=1, version=1, release_year=1998, title='The Big Lebowski', popularity=1, popularity_month=1) self.f12.save() self.f13 = Film( type=1, permalink='the-lord-of-the-rings-the-fellowship-of-the-ring', imdb_code=123, status=1, version=1, release_year=2001, title='The Lord of the Rings: The Fellowship of the Ring', popularity=1, popularity_month=1) self.f13.save() self.f14 = Film(type=1, permalink='raiders-of-the-lost-ark', imdb_code=124, status=1, version=1, release_year=1981, title='Raiders of the Lost Ark', popularity=1, popularity_month=1) self.f14.save() self.f15 = Film(type=1, permalink='the-alien', imdb_code=125, status=1, version=1, release_year=1979, title='The Alien', popularity=1, popularity_month=1) self.f15.save() self.f16 = Film(type=1, permalink='przypadek', imdb_code=126, status=1, version=1, release_year=1999, title='Przypadek', popularity=20, popularity_month=10) self.f16.save() self.f17 = Film(type=1, permalink='wrestler', imdb_code=127, status=1, version=1, release_year=2008, title='Wrestler', popularity=10, popularity_month=1) self.f17.save() self.f18 = Film(type=1, permalink='american-history-x', imdb_code=128, status=1, version=1, release_year=1998, title='American History X', popularity=1, popularity_month=1) self.f18.save() self.f19 = Film(type=1, permalink='the-big-lebowski', imdb_code=129, status=1, version=1, release_year=1998, title='The Big Lebowski', popularity=1, popularity_month=1) self.f19.save() self.f20 = Film( type=1, permalink='the-lord-of-the-rings-the-fellowship-of-the-ring', imdb_code=130, status=1, version=1, release_year=2001, title='The Lord of the Rings: The Fellowship of the Ring', popularity=1, popularity_month=1) self.f20.save() self.f21 = Film(type=1, permalink='raiders-of-the-lost-ark', imdb_code=131, status=1, version=1, release_year=1981, title='Raiders of the Lost Ark', popularity=1, popularity_month=1) self.f21.save() self.f22 = Film(type=1, permalink='the-alien', imdb_code=132, status=1, version=1, release_year=1979, title='The Alien', popularity=1, popularity_month=1) self.f22.save() self.f23 = Film(type=1, permalink='terminator-2-judgment-day', imdb_code=133, status=1, version=1, release_year=1991, title='Terminator 2: Judgment Day', popularity=1, popularity_month=1) self.f23.save() self.f24 = Film(type=1, permalink='przypadek', imdb_code=134, status=1, version=1, release_year=1999, title='Przypadek', popularity=20, popularity_month=10) self.f24.save() self.f25 = Film(type=1, permalink='wrestler', imdb_code=135, status=1, version=1, release_year=2008, title='Wrestler', popularity=10, popularity_month=1) self.f25.save() self.f26 = Film(type=1, permalink='american-history-x', imdb_code=136, status=1, version=1, release_year=1998, title='American History X', popularity=1, popularity_month=1) self.f26.save() self.f27 = Film(type=1, permalink='the-big-lebowski', imdb_code=137, status=1, version=1, release_year=1998, title='The Big Lebowski', popularity=1, popularity_month=1) self.f27.save() self.f28 = Film( type=1, permalink='the-lord-of-the-rings-the-fellowship-of-the-ring', imdb_code=138, status=1, version=1, release_year=2001, title='The Lord of the Rings: The Fellowship of the Ring', popularity=1, popularity_month=1) self.f28.save() self.f29 = Film(type=1, permalink='raiders-of-the-lost-ark', imdb_code=139, status=1, version=1, release_year=1981, title='Raiders of the Lost Ark', popularity=1, popularity_month=1) self.f29.save() self.f30 = Film(type=1, permalink='the-alien', imdb_code=140, status=1, version=1, release_year=1979, title='The Alien', popularity=1, popularity_month=1) self.f30.save() self.f31 = AddedFilm(title='Added film part 1', localized_title='Added film PL', release_year=2010, user=self.u1) self.f31.save() self.f31.directors.add(self.p1) self.f31.save() AddedCharacter.objects.create(added_film=self.f31, person=self.p3, character="Jack Mort") self.f31.accept(self.u2) self.f32 = AddedFilm(title='Added film part 2', localized_title='Added film PL', release_year=2010, user=self.u1) self.f32.save() self.f32.directors.add(self.p2) self.f32.save() AddedCharacter.objects.create(added_film=self.f32, person=self.p3, character="Jack Mort") self.f32.accept(self.u2) #tagging self.f1.save_tags('comedy drama horror animation mystery') self.f2.save_tags('comedy sci-fi thriller series') self.f3.save_tags('comedy') self.f4.save_tags('sci-fi') self.f5.save_tags('horror') self.f6.save_tags('thriller') self.f7.save_tags('drama') self.f8.save_tags('series') self.f9.save_tags('sci-fi') self.f10.save_tags('fantasy') self.f11.save_tags('mystery') self.f12.save_tags('action') self.f13.save_tags('adventure') self.f14.save_tags('animation') self.f15.save_tags('family') self.f16.save_tags('thriller') self.f17.save_tags('drama') self.f18.save_tags('series') self.f19.save_tags('sci-fi') self.f20.save_tags('fantasy') self.f21.save_tags('comedy') self.f22.save_tags('animation') self.f23.save_tags('comedy') self.f24.save_tags('sci-fi') self.f25.save_tags('horror') self.f26.save_tags('thriller') self.f27.save_tags('drama') self.f28.save_tags('series') self.f29.save_tags('sci-fi') self.f30.save_tags('fantasy') #First user ratings save_rating(self.f1, self.u1, 1) save_rating(self.f2, self.u1, 2) save_rating(self.f3, self.u1, 3) save_rating(self.f4, self.u1, 4) save_rating(self.f5, self.u1, 5) save_rating(self.f6, self.u1, 6) save_rating(self.f7, self.u1, 7) save_rating(self.f8, self.u1, 8) save_rating(self.f9, self.u1, 9) save_rating(self.f10, self.u1, 10) save_rating(self.f11, self.u1, 1) save_rating(self.f12, self.u1, 2) save_rating(self.f13, self.u1, 3) save_rating(self.f14, self.u1, 4) save_rating(self.f15, self.u1, 5) save_rating(self.f16, self.u1, 6) save_rating(self.f17, self.u1, 7) save_rating(self.f18, self.u1, 8) save_rating(self.f19, self.u1, 9) save_rating(self.f20, self.u1, 10) save_rating(self.f21, self.u1, 1) save_rating(self.f22, self.u1, 2) save_rating(self.f23, self.u1, 3) save_rating(self.f24, self.u1, 4) save_rating(self.f25, self.u1, 5) save_rating(self.f26, self.u1, 6) save_rating(self.f27, self.u1, 7) save_rating(self.f28, self.u1, 8) save_rating(self.f29, self.u1, 9) save_rating(self.f30, self.u1, 10) #Second user ratings - more than minimal common films requirement (15) save_rating(self.f1, self.u2, 1) save_rating(self.f2, self.u2, 2) save_rating(self.f3, self.u2, 3) save_rating(self.f4, self.u2, 4) save_rating(self.f5, self.u2, 5) save_rating(self.f6, self.u2, 6) save_rating(self.f7, self.u2, 7) save_rating(self.f8, self.u2, 8) save_rating(self.f9, self.u2, 9) save_rating(self.f10, self.u2, 10) save_rating(self.f11, self.u2, 1) #save_rating(self.f12, self.u2, 2) save_rating(self.f13, self.u2, 3) #save_rating(self.f14, self.u2, 4) save_rating(self.f15, self.u2, 5) #save_rating(self.f16, self.u2, 6) save_rating(self.f17, self.u2, 7) #save_rating(self.f18, self.u2, 8) save_rating(self.f19, self.u2, 9) #save_rating(self.f20, self.u2, 10) save_rating(self.f21, self.u2, 1) #save_rating(self.f22, self.u2, 2) save_rating(self.f23, self.u2, 3) #save_rating(self.f24, self.u2, 4) save_rating(self.f25, self.u2, 5) #save_rating(self.f26, self.u2, 6) save_rating(self.f27, self.u2, 7) #save_rating(self.f28, self.u2, 8) save_rating(self.f29, self.u2, 9) #save_rating(self.f30, self.u2, 10) # Third user ratings - equal with minimal common films requirement (15) #save_rating(self.f1, self.u3, 1) #save_rating(self.f2, self.u3, 2) #save_rating(self.f3, self.u3, 3) #save_rating(self.f4, self.u3, 4) #save_rating(self.f5, self.u3, 5) #save_rating(self.f6, self.u3, 6) #save_rating(self.f7, self.u3, 7) #save_rating(self.f8, self.u3, 8) #save_rating(self.f9, self.u3, 9) #save_rating(self.f10, self.u3, 10) #save_rating(self.f11, self.u3, 1) #save_rating(self.f12, self.u3, 2) #save_rating(self.f13, self.u3, 3) #save_rating(self.f14, self.u3, 4) #save_rating(self.f15, self.u3, 5) save_rating(self.f16, self.u3, 6) save_rating(self.f17, self.u3, 7) save_rating(self.f18, self.u3, 8) save_rating(self.f19, self.u3, 9) save_rating(self.f20, self.u3, 10) save_rating(self.f21, self.u3, 1) save_rating(self.f22, self.u3, 2) save_rating(self.f23, self.u3, 3) save_rating(self.f24, self.u3, 4) save_rating(self.f25, self.u3, 5) save_rating(self.f26, self.u3, 6) save_rating(self.f27, self.u3, 7) save_rating(self.f28, self.u3, 8) save_rating(self.f29, self.u3, 9) save_rating(self.f30, self.u3, 10) # Fourth user ratings - less than minimal films requirement #save_rating(self.f1, self.u4, 1) #save_rating(self.f2, self.u4, 2) #save_rating(self.f3, self.u4, 3) #save_rating(self.f4, self.u4, 4) #save_rating(self.f5, self.u4, 5) #save_rating(self.f6, self.u4, 6) #save_rating(self.f7, self.u4, 7) #save_rating(self.f8, self.u4, 8) #save_rating(self.f9, self.u4, 9) #save_rating(self.f10, self.u4, 10) #save_rating(self.f11, self.u4, 1) #save_rating(self.f12, self.u4, 2) #save_rating(self.f13, self.u4, 3) #save_rating(self.f14, self.u4, 4) #save_rating(self.f15, self.u4, 5) #save_rating(self.f16, self.u4, 6) #save_rating(self.f17, self.u4, 7) #save_rating(self.f18, self.u4, 8) #save_rating(self.f19, self.u4, 9) #save_rating(self.f20, self.u4, 10) save_rating(self.f21, self.u4, 1) save_rating(self.f22, self.u4, 2) save_rating(self.f23, self.u4, 3) save_rating(self.f24, self.u4, 4) save_rating(self.f25, self.u4, 5) save_rating(self.f26, self.u4, 6) save_rating(self.f27, self.u4, 7) save_rating(self.f28, self.u4, 8) save_rating(self.f29, self.u4, 9) save_rating(self.f30, self.u4, 10) # Fifth user ratings - the same films rated (1st and 5th user) but different # ratings #save_rating(self.f1, self.u5, 10) #save_rating(self.f2, self.u5, 9) #save_rating(self.f3, self.u5, 8) #save_rating(self.f4, self.u5, 7) #save_rating(self.f5, self.u5, 6) #save_rating(self.f6, self.u5, 5) #save_rating(self.f7, self.u5, 4) #save_rating(self.f8, self.u5, 3) #save_rating(self.f9, self.u5, 2) #save_rating(self.f10, self.u5, 1) #save_rating(self.f11, self.u5, 10) #save_rating(self.f12, self.u5, 9) #save_rating(self.f13, self.u5, 8) #save_rating(self.f14, self.u5, 7) #save_rating(self.f15, self.u5, 6) #save_rating(self.f16, self.u5, 5) #save_rating(self.f17, self.u5, 4) #save_rating(self.f18, self.u5, 3) #save_rating(self.f19, self.u5, 2) #save_rating(self.f20, self.u5, 1) #save_rating(self.f21, self.u5, 10) #save_rating(self.f22, self.u5, 9) #save_rating(self.f23, self.u5, 8) #save_rating(self.f24, self.u5, 7) #save_rating(self.f25, self.u5, 6) #save_rating(self.f26, self.u5, 5) #save_rating(self.f27, self.u5, 4) #save_rating(self.f28, self.u5, 3) #save_rating(self.f29, self.u5, 2) #save_rating(self.f30, self.u5, 1) # Sixth user ratings - the same films rated (1st and 5th user) but different # ratings save_rating(self.f1, self.u6, 10) save_rating(self.f2, self.u6, 10) save_rating(self.f3, self.u6, 10) save_rating(self.f4, self.u6, 10) save_rating(self.f5, self.u6, 10) save_rating(self.f6, self.u6, 10) save_rating(self.f7, self.u6, 10) save_rating(self.f8, self.u6, 10) save_rating(self.f9, self.u6, 10) save_rating(self.f10, self.u6, 10) save_rating(self.f11, self.u6, 10) save_rating(self.f12, self.u6, 10) save_rating(self.f13, self.u6, 10) save_rating(self.f14, self.u6, 10) save_rating(self.f15, self.u6, 10) save_rating(self.f16, self.u6, 10) save_rating(self.f17, self.u6, 10) save_rating(self.f18, self.u6, 10) save_rating(self.f19, self.u6, 10) save_rating(self.f20, self.u6, 10) save_rating(self.f21, self.u6, 10) save_rating(self.f22, self.u6, 10) save_rating(self.f23, self.u6, 10) save_rating(self.f24, self.u6, 10) save_rating(self.f25, self.u6, 10) save_rating(self.f26, self.u6, 10) save_rating(self.f27, self.u6, 10) save_rating(self.f28, self.u6, 10) save_rating(self.f29, self.u6, 10) save_rating(self.f30, self.u6, 10)
def accept(self, user): # generate permalink permalink = get_film_permalink(self) film = Film(title=self.title, release_year=self.release_year, image=self.image, version=1, is_tv_series=(self.type == self.TYPE_SERIES), completion_year=self.completion_year if (self.type == self.TYPE_SERIES) else None, type=Film.TYPE_FILM, permalink=permalink, status=1, production_country_list=','.join( [c.country for c in self.production_country.all()])) film.save() # add localized title if self.localized_title: film.save_localized_title(self.localized_title) # add directors for director in self.directors.all(): film.directors.add(director) # ... and set director flag to true director.is_director = True director.save() # set countries for country in self.production_country.all(): film.production_country.add(country) other_languages = get_other_languages() # set actors for actor in self.get_actors(): # save charater in default LANG Character(film=film, person=actor.person, importance=actor.importance, character=actor.character).save() # mark this person as actor actor.person.is_actor = True actor.save() # ... and in other languages for lang in other_languages: Character(film=film, person=actor.person, importance=actor.importance, LANG=lang).save() # set reference self.film = film super(AddedFilm, self).accept(user)
def initialize(self): self.clean_data() # set up users self.u1 = User.objects.create_user('michuk', '*****@*****.**', 'secret') self.u1.save() self.u2 = User(username='******', email='*****@*****.**') self.u2.save() self.u3 = User(username='******', email='*****@*****.**') self.u3.save() self.u4 = User(username='******', email='*****@*****.**') self.u4.save() self.contest = Contest() self.contest.name = "Plebiscyt" self.contest.start_date = datetime.now() self.contest.end_date = datetime.now() self.contest.event_status = Contest.STATUS_OPEN self.contest.type = Object.TYPE_CONTEST self.contest.permalink = "plebiscyt" self.contest.save() film1 = Film() film1.title = "Battlefield Earth II" film1.type = Object.TYPE_FILM film1.permalink = "/battlefirld-earth-ii/" film1.release_year = 2010 film1.save() person1 = Person() person1.name = "John" person1.surname = "Travolta" person1.type = Object.TYPE_PERSON person1.permalink = "/john-travolta/" person1.save() person2 = Person() person2.name = "Tom" person2.surname = "Cruise" person2.type = Object.TYPE_PERSON person2.permalink = "/tom-cruise/" person2.save() person3 = Person() person3.name = "Forrest" person3.surname = "Whitaker" person3.type = Object.TYPE_PERSON person3.permalink = "/forrest-whitaker/" person3.save() character1 = Character() character1.character = "Scientologist 1" character1.person = person1 character1.film = film1 character1.save() self.character1 = character1 character2 = Character() character2.character = "Scientologist 2" character2.person = person2 character2.film = film1 character2.save() self.character2 = character2 character3 = Character() character3.character = "A non-scientologist" character3.person = person3 character3.film = film1 character3.save() self.character3 = character3 today = datetime.today() tomorrow = today + timedelta(1) self.game1 = Game() self.game1.contest = self.contest self.game1.type = Object.TYPE_GAME self.game1.permalink = "scientologist-1-vs-scientologist-2" self.game1.level = Game.LEVEL_32 self.game1.character1 = character1 self.game1.character2 = character2 self.game1.start_date = today self.game1.end_date = today self.game1.save() self.game2 = Game() self.game2.contest = self.contest self.game2.type = Object.TYPE_GAME self.game1.permalink = "scientologist-1-vs-scientologist-2" self.game2.level = Game.LEVEL_32 self.game2.character1 = character1 self.game2.character2 = character2 self.game2.start_date = tomorrow self.game2.end_date = tomorrow self.game2.save()
def setUp(self): self.user = User.objects.create_user("user", "*****@*****.**", "user") self.film = Film( type=1, permalink='przypadek', imdb_code=111, status=1, version=1, \ release_year=1999, title='Przypadek' ) self.film.save()