Data inserted.

Author: mikeckennedy

apps/py/ch08_db/guitary/app.py

@@ -1,6 +1,7 @@
 import flask
 from guitary.services import catalog_service
 from guitary.data import session_factory
+from guitary.data import data_loader
 
 app = flask.Flask(__name__)
 
@@ -20,6 +21,7 @@ def guitars(style: str = None):
 def main():
     session_factory.global_init('guitary.sqlite')
     session_factory.create_tables()
+    data_loader.load_guitars_if_empty()
 
     app.run(debug=True)
 

apps/py/ch08_db/guitary/data/context_session.py

@@ -3,7 +3,7 @@
 
 class ContextSession:
     def __init__(self, session: Session):
-        self.session = session
+        self.session: Session = session
 
     def __enter__(self):
         return self

apps/py/ch08_db/guitary/data/data_loader.py

@@ -0,0 +1,32 @@
+from guitary.data.guitar import Guitar
+from guitary.data import session_factory
+
+
+def load_guitars_if_empty():
+
+    with session_factory.create_session() as ctx:
+        count = ctx.session.query(Guitar).count()
+        if count > 0:
+            print(f"Not adding new data, there are {count} guitars already.")
+            return
+
+        guitars = [
+            Guitar(name='AX Black', price=499, img='/static/img/guitars/ax-black.jpg', style='electric'),
+            Guitar(name='Jet Black Electric', price=599, img='/static/img/guitars/jet-black-electric.jpg',
+                   style='electric'),
+            Guitar(name='Weezer Classic', price=1499, img='/static/img/guitars/weezer-classic.jpg', style='electric'),
+            Guitar(name='Acoustic Black', price=1298, img='/static/img/guitars/black-acoustic.jpg', style='acoustic'),
+            Guitar(name='Mellow Yellow', price=799, img='/static/img/guitars/mellow-yellow.jpg', style='electric'),
+            Guitar(name='White Vibes', price=699, img='/static/img/guitars/white-vibes.jpg', style='electric'),
+            Guitar(name='Brush Riffs', price=599, img='/static/img/guitars/brushed-black-electric.jpg',
+                   style='electric'),
+            Guitar(name="Nature's Song", price=799, img='/static/img/guitars/natures-song.jpg', style='electric'),
+            Guitar(name='Electric Wood Grain', price=399, img='/static/img/guitars/woodgrain-electric.jpg',
+                   style='electric'),
+        ]
+
+        # Do work here...
+        for guitar in guitars:
+            ctx.session.add(guitar)
+
+        ctx.session.commit()