added code

1 files changed, 49 insertions, 0 deletions

diff --git a/database.py b/database.py
new file mode 100644
index 0000000..369cf97
--- /dev/null
+++ b/database.py
@@ -0,0 +1,49 @@
+import sqlalchemy
+from sqlalchemy import create_engine
+from sqlalchemy import Table, Column, Integer, String, MetaData, ForeignKey, PickleType
+from sqlalchemy import insert
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.exc import IntegrityError
+
+#database_url = "sqlite:///jlm.db"
+
+meta = MetaData()
+
+movies_table = Table(
+    "movies",
+    meta,
+    Column("trakt_id", Integer, primary_key=True, autoincrement=False),
+    Column("title", String),
+    Column("overview", String),
+    Column("genres", String),
+    Column("year", Integer),
+    Column("released", String),
+    Column("runtime", Integer),
+    Column("country", String),
+    Column("language", String),
+    Column("rating", Integer),
+    Column("votes", Integer),
+    Column("comment_count", Integer),
+    Column("tagline", String),
+    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"],
+        "overview": movie["movie"]["overview"],
+        "genres": movie["movie"]["genres"],
+        "language": movie["movie"]["language"],
+        "year": movie["movie"]["year"],
+        "trakt_id": movie["movie"]["ids"]["trakt"],
+        "released": movie["movie"]["released"],
+        "runtime": movie["movie"]["runtime"],
+        "country": movie["movie"]["country"]
+    }
+"""
\ No newline at end of file