def loadMovies (catalog, sep=';'):
"""
Carga las películas del archivo. Por cada libro se toman sus autores y por
cada uno de ellos, se crea en la lista de autores, a dicho autor y una
referencia al libro que se esta procesando.
"""
t1_start = process_time() #tiempo inicial
moviefile = cf.data_dir + 'GoodReads/AllMoviesDetailsCleaned.csv'
dialect = csv.excel()
dialect.delimiter=sep
with open(moviefile, encoding="utf-8-sig") as csvfile:
spamreader = csv.DictReader(csvfile, dialect=dialect)
for row in spamreader:
# Se adiciona el libro a la lista de libros
model.addMovieList(catalog, row)
# Se adiciona el libro al mapa de libros (key=title)
model.addMovieMapTitle(catalog, row)
model.addMovieMapId(catalog, row)
# Se obtienen los autores del libro
#authors = row['director_name'].split(",")
# Cada autor, se crea en la lista de autores del catalogo, y se
# adiciona un libro en la lista de dicho autor (apuntador al libro)
model.addGenre(catalog, row["genres"], row)
t1_stop = process_time() #tiempo final
print("Tiempo de ejecución carga películas:",t1_stop-t1_start," segundos")
def addElementsmapsDetails(catalog, file1, file2):
"""
Carga los elementos de los mapas relacionados con los details.
"""
file1 = config.file_dir + file1
dialect = csv.excel()
dialect.delimiter = ";"
input_file1 = csv.DictReader(open(file1, encoding="utf-8"),
dialect=dialect)
file2 = config.file_dir + file2
dialect = csv.excel()
dialect.delimiter = ";"
input_file2 = csv.DictReader(open(file2, encoding="utf-8"),
dialect=dialect)
for movie1 in input_file1:
lt.addLast(catalog['Data']['details'], movie1)
md.addCompany(movie1, catalog)
md.addGenre(movie1, catalog)
iterator = it.newIterator(catalog['Data']['details'])
for movie2 in input_file2:
movie1 = it.next(iterator)
lt.addLast(catalog['Data']['casting'], movie2)
md.addDirector(movie2, movie1, catalog)
md.addActor(movie2, movie1, catalog)
md.addCountry(movie2, movie1, catalog)
def loadGenres(catalog):
genres = [("Reggae", 60, 90), ("Down-tempo", 70, 100),
("Chill-out", 90, 120), ("Hip-hop", 85, 115),
("Jazz and Funk", 120, 125), ("Pop", 100, 130), ("R&B", 60, 80),
("Rock", 110, 140), ("Metal", 100, 160)]
for genre in genres:
model.addGenre(catalog, genre[0], genre[1], genre[2])
genres.clear()
def addGenre(catalog, genre):
Genre = [['Reggae', 60, 90], ['Dowm-tempo', 70, 100],
['Chill-out', 90, 120], ['Hip-hop', 85, 115],
['Jazz and Funk', 120, 125], ['Pop', 100, 130], ['RGB', 60, 80],
['Rock', 110, 140], ['Metal', 100, 160]]
if genre is None:
for i in Genre:
model.addGenre(catalog['genre'], i)
else:
model.addGenre(catalog['genre'], genre)
def loadFeatures(cat, features):
ffile = cf.data_dir + "subsamples-small" + "\\" + features
input_file = csv.DictReader(open(ffile, encoding="utf-8"), delimiter=",")
a = True
for rep in input_file:
if a:
model.addCategories(cat, rep)
model.addGenre(cat)
a = False
add = model.addRep(cat, rep)
print("Se cargó el archivo de Context-Features")
return cat
def loadGenres(catalog):
"""
Carga todos los géneros
"""
t1_start = process_time() #tiempo inicial
moviesfile = cf.data_dir + 'themoviesdb/SmallMoviesDetailsCleaned.csv'
dialect = csv.excel()
dialect.delimiter = ";"
with open(moviesfile, encoding="utf-8") as csvfile:
spamreader = csv.DictReader(csvfile, dialect=dialect)
for row in spamreader:
model.addGenre(catalog, row)
t1_stop = process_time() #tiempo final
print("Tiempo de ejecución carga géneros", t1_stop - t1_start, " segundos")
def loadGenres (catalog):
'''
Carga el catálogo de géneros del archivo. Por cada género se cargan sus libros
'''
t1_start = process_time() #tiempo inicial
movies = catalog['movies']
for movie in movies:
genre_name = movie['genres']
pos = lt.isPresent(genre_name, catalog['genres'], equal)
if pos != 0:
model.updateGenre(catalog, movie, pos)
else:
model.addGenre(catalog, movie)
t1_stop = process_time() #tiempo final
print('Tiempo de ejecución carga peliculas',t1_stop-t1_start,' segundos')
endGenreslist_controller(catalog)
def loadMovies(catalog, sep=';'):
"""
Carga las películas del archivo.
"""
t1_start = process_time() #tiempo inicial
#moviesfile = cf.data_dir + 'themoviesdb/SmallMoviesDetailsCleaned.csv'
moviesfile = cf.data_dir + 'themoviesdb/AllMoviesDetailsCleaned.csv'
dialect = csv.excel()
dialect.delimiter = sep
with open(moviesfile, encoding="utf-8-sig") as csvfile:
spamreader = csv.DictReader(csvfile, dialect=dialect)
for row in spamreader:
# Se adiciona la pelicula a la lista de peliculas
#model.addMovieList(catalog, row)
# Se adiciona la pelicula al mapa de peliculas (key=title)
model.addMovieMap(catalog, row)
model.addTitlesMap(catalog, row)
model.addGenre(catalog, row)
t1_stop = process_time() #tiempo final
print("Tiempo de ejecución carga películas y géneros:", t1_stop - t1_start,
" segundos")
def loadMovies(catalog, sep=';'):
"""
Carga los libros del archivo. Por cada libro se toman sus autores y por
cada uno de ellos, se crea en la lista de autores, a dicho autor y una
referencia al libro que se esta procesando.
"""
t1_start = process_time() #tiempo inicial
moviesfile = cf.data_dir + 'SmallMoviesDetailsCleaned.csv'
dialect = csv.excel()
dialect.delimiter = sep
with open(moviesfile, encoding="utf-8-sig") as csvfile:
spamreader = csv.DictReader(csvfile, dialect=dialect)
for row in spamreader:
# Se adiciona la pelócula a la lista de películas
model.addMovieList(catalog, row)
# Se adiciona la película al mapa de películas (key=title)
model.addMovieMap(catalog, row)
# Se adiciona el id de la pelicula al mapa de ids (key= id)
model.addIdMap(catalog, row)
# Se adiciona el género al mapa de géneros (key= genre)
model.addGenre(catalog, row)
t1_stop = process_time() #tiempo final
print("Tiempo de ejecución carga películas:", t1_stop - t1_start,
" segundos")
def addGenre(mapa, genero, bajo, alto, tree):
return model.addGenre(mapa, genero, bajo, alto, tree)
def loadMovies(catalog, sep=';'):
"""
Carga las películas del archivo. Por cada película se toman sus directores y por
cada uno de ellos se crea una referencia a la película que se esta procesando.
"""
t1_start = process_time() #tiempo inicial
moviesfile = cf.data_dir + '/Movies/MoviesCastingRaw-small.csv'
dialect = csv.excel()
dialect.delimiter = sep
with open(moviesfile, encoding="utf-8-sig") as csvfile:
spamreader = csv.DictReader(csvfile, dialect=dialect)
for row in spamreader:
""" AQUI SE AGREGA ÚNICAMENTE LA INFORMACIÓN DE CASTING """
# Se obtienen los actores de la película
# Se crea en la lista de actores del catalogo, y se
# adiciona una película en la lista de dicho actor (apuntador a la película)
actors = [
"actor1_name", "actor2_name", "actor3_name", "actor4_name",
"actor5_name"
]
actors_movie = lt.newList("ARRAY_LIST")
for actor in actors:
if row[actor] != "none":
model.addActor(catalog, row[actor], row)
lt.addLast(actors_movie, row[actor])
#tipo_de_estructura = "lista"
tipo_de_estructura = "mapa"
# Se adiciona la película a la lista de películas
if tipo_de_estructura == "lista":
model.addMovieList(catalog, row, actors_movie)
# Se adiciona la película al mapa de películas (key=title)
if tipo_de_estructura == "mapa":
model.addMovieMap(catalog, row, actors_movie)
# Se obtiene el director de la película
model.addDirector(catalog, row["director_name"], row)
moviesfile = cf.data_dir + '/Movies/SmallMoviesDetailsCleaned.csv'
with open(moviesfile, encoding="utf-8-sig") as csvfile:
spamreader = csv.DictReader(csvfile, dialect=dialect)
actors = map.keySet(catalog["actors"])
directors = map.keySet(catalog["directors"])
""" AQUI SE DEBERÍA AGREGAR UNICAMENTE LA INFORMACIÓN DE VOTOS """
for row in spamreader:
"""
Con estructura de Lista
"""
if tipo_de_estructura == "lista":
addMovieListVoteData(catalog, row)
"""
Con estructura de Mapa
"""
if tipo_de_estructura == "mapa":
model.addMovieMapVoteData(catalog, row)
for actor in actors:
model.addActorVoteData(catalog, actor, row)
for director in directors:
model.addDirectorVoteData(catalog, director, row)
generos = row["genres"].split("|")
for genero in generos:
model.addGenre(catalog, genero, row)
t1_stop = process_time() #tiempo final
print("Tiempo de ejecución carga películas:", t1_stop - t1_start,
" segundos")