def test_add_two_movies(self):
arr = Movie(titleImdb='Arrival')
sic = Movie(titleImdb='Sicario')
db.session.add(arr)
db.session.add(sic)
db.session.commit()
movies = Movie.query.all()
c = len(movies)
assert c == 2
def test_add_two_movies_with_id(self):
arr = Movie(imdbId='2543164', titleLocal='Arrival')
sic = Movie(imdbId='3397884', titleLocal='Sicario')
db.session.add(arr)
db.session.add(sic)
db.session.commit()
movies = Movie.query.all()
c = len(movies)
assert c == 2
def test_add_two_movies(self):
an = Movie(titleImdb='Apocalypse Now')
sic = Movie(titleImdb='Sicario')
db.session.add(an)
db.session.add(sic)
db.session.commit()
movies = Movie.query.all()
c = len(movies)
assert c == 2
def setup(self):
print("setup method, class: TestMcIntoDb, fixture test method")
self.app = create_app(TestConfig)
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
sic = Movie(titleImdb='Sicario')
db.session.add(sic)
arr = Movie(titleImdb='Arrival')
db.session.add(arr)
db.session.commit()
def test_add_query_two_roles(self):
an = Movie(titleImdb='Apocalypse Now')
sic = Movie(titleImdb='Sicario')
kurz = Role(characterName='Kurz', film=an)
willard = Role(characterName='Willard', film=an)
db.session.add(an)
db.session.add(sic)
db.session.add(kurz)
db.session.add(willard)
db.session.commit()
m = Movie.query.get(1)
roles = m.roles.all()
c = len(roles)
assert c == 2
def test_add_first_movie(self):
an = Movie(titleImdb='Apocalypse Now')
db.session.add(an)
db.session.commit()
movies = Movie.query.all()
c = len(movies)
assert c == 1
def test_add_two_roles_peoples(self):
an = Movie(titleImdb='Apocalypse Now')
sic = Movie(titleImdb='Sicario')
marlon = People(name='Marlon')
sheen = People(name='Scheen')
kurz = Role(characterName='Kurz', film=an, actor=marlon)
willard = Role(characterName='Willard', film=an, actor=sheen)
db.session.add(marlon)
db.session.add(sheen)
db.session.add(an)
db.session.add(sic)
db.session.add(kurz)
db.session.add(willard)
db.session.commit()
m = People.query.get(1)
roles = m.roles.all()
c = len(roles)
assert c == 1
name = roles[0].characterName
assert name == 'Kurz'
def addManMovieWithoutIdToDb(inputTitle='', medium='-', source='', place=''):
'''
'''
newMovie = Movie(
imdbId='0000000',
titleLocal=inputTitle,
medium = medium,
place=place,
source = source)
db.session.add(newMovie)
db.session.commit()
def test_own_critics(self):
an = Movie(titleImdb='Apocalypse Now')
sic = Movie(titleImdb='Sicario')
marlon = People(name='Marlon')
sheen = People(name='Scheen')
coppola = Director(name='Coppola')
kurz = Role(characterName='Kurz')
kurz.film = an
kurz.actor = marlon
jd = Critic(name='JD', maxVal=5.0)
rat = Rating(value=5.0)
rat.film = an
rat.critic = jd
rats = Rating(value=4.5)
rats.film = sic
rats.critic = jd
willard = Role(characterName='Willard', film=an, actor=sheen)
an.director = coppola
db.session.add(marlon)
db.session.add(sheen)
db.session.add(coppola)
db.session.add(an)
db.session.add(sic)
db.session.add(kurz)
db.session.add(willard)
db.session.add(rat)
db.session.add(rats)
db.session.commit()
# TODO check ratings
f = Movie.query.get(1)
print(type(f))
dir = f.director.name
print(dir)
rq = Rating.query
# rq = Rating.query.filter_by(critic_id='JD')
r = rq.all()
print(type(r))
print(len(r))
print(r)
def test_add_director(self):
an = Movie(titleImdb='Apocalypse Now')
sic = Movie(titleImdb='Sicario')
marlon = People(name='Marlon')
sheen = People(name='Scheen')
coppola = Director(name='Coppola')
kurz = Role(characterName='Kurz')
kurz.film = an
kurz.actor = marlon
willard = Role(characterName='Willard', film=an, actor=sheen)
an.director = coppola
db.session.add(marlon)
db.session.add(sheen)
db.session.add(coppola)
db.session.add(an)
db.session.add(sic)
db.session.add(kurz)
db.session.add(willard)
db.session.commit()
f = Movie.query.get(1)
dir = f.director.name
assert dir == 'Coppola'
def addManMovieToDb(inputMovieId,
inputTitle='', medium='',
source='', place='',
ownrating='',
amgrating=''):
'''
first search local imdb data for item, extract infos about
movie, director, rating
then insert new data set to db
'''
# get data from imdb tsv
imdbData = tsv.getMovieData(inputMovieId)
titleImdb=''
titleOrig=''
year = ''
director = ''
# prepare data for
length = 0
if imdbData != None:
length = imdbData[0]
titleImdb = imdbData[1]
if len(imdbData) > 2:
titleOrig = imdbData[2]
if len(imdbData) > 3:
year = imdbData[3]
newMovie = Movie(imdbId=inputMovieId,
titleImdb=titleImdb,
titleOrig=titleOrig,
titleLocal=inputTitle,
medium = medium,
year = year,
linelength = length,
place = place,
source = source)
db.session.add(newMovie)
dirData = tsv.getMovieDirector(inputMovieId)
if dirData != None:
directorId = dirData
dirInDb = searchDirectorDb(directorId)
if dirInDb == None:
dirName = tsv.getNameData(directorId)
director = Director(peopleImdbId=directorId, name=dirName)
db.session.add(director)
else:
director = dirInDb
newMovie.director = director
ratData = tsv.getMovieRating(inputMovieId)
if ratData != None:
try:
critic = Critic.query.filter_by(name='Imdb').first()
rat = Rating(movie_id=newMovie.id, critic_id=critic.id, value=ratData)
db.session.add(rat)
except:
current_app.logger.error('Unhandled exception', exc_info=sys.exc_info())
pass
if ownrating != '':
try:
critic = Critic.query.filter_by(name='JD').first()
rat = Rating(movie_id=newMovie.id, critic_id=critic.id, value=ownrating)
db.session.add(rat)
except:
current_app.logger.error('Unhandled exception', exc_info=sys.exc_info())
pass
if amgrating != '':
try:
critic = Critic.query.filter_by(name='AMG').first()
rat = Rating(movie_id=newMovie.id, critic_id=critic.id, value=amgrating)
db.session.add(rat)
except:
current_app.logger.error('Unhandled exception', exc_info=sys.exc_info())
pass
db.session.commit()