diff options
Diffstat (limited to 'database.py')
| -rw-r--r-- | database.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/database.py b/database.py index 369cf97..1a4bda0 100644 --- a/database.py +++ b/database.py @@ -5,7 +5,7 @@ from sqlalchemy import insert from sqlalchemy.orm import sessionmaker from sqlalchemy.exc import IntegrityError -#database_url = "sqlite:///jlm.db" +# database_url = "sqlite:///jlm.db" meta = MetaData() @@ -25,15 +25,17 @@ movies_table = Table( Column("votes", Integer), Column("comment_count", Integer), Column("tagline", String), - Column("embeddings", PickleType) - + Column("embeddings", PickleType), ) + def init_db_stuff(database_url: str): engine = create_engine(database_url) meta.create_all(engine) Session = sessionmaker(bind=engine) return engine, Session + + """ movie = { "title": movie["movie"]["title"], @@ -46,4 +48,4 @@ def init_db_stuff(database_url: str): "runtime": movie["movie"]["runtime"], "country": movie["movie"]["country"] } -"""
\ No newline at end of file +""" |