SQLAlchemy tutorial part 2: ORM models.

Author: toddbirchard

sqlalchemy_tutorial/__init__.py

@@ -1,7 +1,9 @@
 """Script entry point."""
 import pprint
 
-from sqlalchemy_tutorial.database.engine import db
+from sqlalchemy_tutorial.database.connect import db
+from sqlalchemy_tutorial.database.models import create_tables
+from sqlalchemy_tutorial.orm import orm_create_user
 from sqlalchemy_tutorial.queries import fetch_job_listings, update_job_listing
 
 # Print formatter
@@ -10,7 +12,10 @@
 
 def init_script():
     """Demonstrate SELECT and UPDATE queries with SQLAlchemy."""
+    create_tables()
     rows_selected = fetch_job_listings(db)
     pp.pprint(rows_selected)
     rows_updated = update_job_listing(db)
     print(rows_updated)
+    user = orm_create_user()
+    print(user)

sqlalchemy_tutorial/database/__init__.py

@@ -1 +1 @@
-from .engine import db
+from .connect import db, session

sqlalchemy_tutorial/database/engine.py renamed to sqlalchemy_tutorial/database/connect.py

@@ -1,6 +1,6 @@
 """Create database connection."""
 from sqlalchemy import create_engine
-from sqlalchemy.orm import Session
+from sqlalchemy.orm import sessionmaker
 
 from config import SQLALCHEMY_DATABASE_PEM, SQLALCHEMY_DATABASE_URI
 
@@ -12,4 +12,5 @@
 )
 
 # Create database session
-session = Session(db)
+Session = sessionmaker(bind=db)
+session = Session()

sqlalchemy_tutorial/database/models.py

@@ -4,11 +4,32 @@
 from sqlalchemy.orm import relationship
 from sqlalchemy.sql import func
 
-from sqlalchemy_tutorial.database import engine
+from sqlalchemy_tutorial.database import db
 
 Base = declarative_base()
 
 
+class User(Base):
+    """User account."""
+
+    __tablename__ = "user"
+
+    id = Column(Integer, primary_key=True, autoincrement="auto")
+    username = Column(String(255), unique=True, nullable=False)
+    password = Column(Text, nullable=False)
+    email = Column(String(255), unique=True, nullable=False)
+    first_name = Column(String(255))
+    last_name = Column(String(255))
+    bio = Column(Text)
+    avatar_url = Column(Text)
+    last_seen = Column(DateTime)
+    created_at = Column(DateTime, server_default=func.now())
+    updated_at = Column(DateTime, onupdate=func.now())
+
+    def __repr__(self):
+        return "<User %r>" % self.username
+
+
 class Player(Base):
     """Individual player belonging to a team."""
 
@@ -20,7 +41,7 @@ class Player(Base):
     last_name = Column(String(255), nullable=False)
     position = Column(String(100), nullable=False)
     injured = Column(Boolean)
-    description = Column(Text, nullable=True)
+    description = Column(Text)
     created_at = Column(DateTime, server_default=func.now())
     updated_at = Column(DateTime, onupdate=func.now())
 
@@ -49,4 +70,4 @@ def __repr__(self):
 
 
 def create_tables():
-    return Base.metadata.create_all(engine)
+    return Base.metadata.create_all(db)

sqlalchemy_tutorial/orm.py

@@ -0,0 +1,19 @@
+"""Demonstrate the SQLAlchemy ORM"""
+from sqlalchemy_tutorial.database import session
+from sqlalchemy_tutorial.database.models import User
+
+
+def orm_create_user():
+    """Create a new instance of our `User` model."""
+    user = User(
+        username="admin",
+        password="Please don't set passwords like this",
+        email="[email protected]",
+        first_name="Todd",
+        last_name="Birchard",
+        bio="I write tutorials on the internet.",
+        avatar_url="https://storage.googleapis.com/hackersandslackers-cdn/authors/[email protected]",
+    )
+
+    session.add(user)  # Add the user
+    session.commit()  # Commit the change