WIP: seems working

This commit is contained in:
Colin Goutte 2023-08-27 14:05:24 +02:00
parent 7624d264c8
commit 9788748d33
1 changed files with 21 additions and 1 deletions

View File

@ -19,6 +19,25 @@ def create_db():
models.Base.metadata.create_all(bind=engine)
def _json_quotes(s):
return s.replace("'", '"')
def adapt_movie_data(data_in: dict):
import copy, json
data_out = copy.deepcopy(data_in)
# adapt genresd to stringlist
breakpoint()
data_out["genres"] = [
genre["name"] for genre in json.loads(_json_quotes(data_in["genres"]))
]
return data_out
def fill_db(
db=SessionLocal(), movie_input_file: str = "input_data/movies_metadata_short.csv"
):
@ -28,7 +47,8 @@ def fill_db(
with open(movie_input_file) as csvfile:
for movie_data in csv.DictReader(csvfile):
crud.create_movie(db, **movie_data)
adapted_data = adapt_movie_data(movie_data)
crud.create_movie(db, **adapted_data)
if __name__ == "__main__":