!squash

cihai · Jun 4, 2023 · 0fb5ef3 · 0fb5ef3
1 parent 892c70c
commit 0fb5ef3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/examples/01_bootstrap.py b/examples/01_bootstrap.py
@@ -10,8 +10,10 @@
 
 bootstrap.bootstrap_unihan(session)
 
-random_row = session.query(Unhn).order_by(func.random()).limit(1).first()
+random_row_query = session.query(Unhn).order_by(func.random()).limit(1)
 
-pp = pprint.PrettyPrinter(indent=0)
+assert random_row_query is not None
 
-pp.pprint(random_row.to_dict())
+random_row = random_row_query.first()
+
+pprint.pprint(bootstrap.to_dict(random_row))
diff --git a/src/unihan_db/bootstrap.py b/src/unihan_db/bootstrap.py
@@ -252,8 +252,7 @@ def get_session(engine_url="sqlite:///{user_data_dir}/unihan_db.db"):
     engine = create_engine(engine_url)
 
     event.listen(mapper, "after_configured", add_to_dict(Base))
-    Base.metadata.bind = engine
-    Base.metadata.create_all()
+    Base.metadata.create_all(bind=engine)
     session_factory = sessionmaker(bind=engine)
     session = scoped_session(session_factory)