| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
1 parent 6457775 commit 9632980
13 files changed
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -21,56 +21,56 @@ def create_db_and_tables(): | |||
| 21 | 21 | ||
| 22 | 22 | ||
| 23 | 23 | def create_heroes(): | |
| 24 | - hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") # (1) | ||
| 25 | - hero_2 = Hero(name="Spider-Boy", secret_name="Pedro Parqueador") # (2) | ||
| 26 | - hero_3 = Hero(name="Rusty-Man", secret_name="Tommy Sharp", age=48) # (3) | ||
| 27 | - | ||
| 28 | - print("Before interacting with the database") # (4) | ||
| 29 | - print("Hero 1:", hero_1) # (5) | ||
| 30 | - print("Hero 2:", hero_2) # (6) | ||
| 31 | - print("Hero 3:", hero_3) # (7) | ||
| 32 | - | ||
| 33 | - with Session(engine) as session: # (8) | ||
| 34 | - session.add(hero_1) # (9) | ||
| 35 | - session.add(hero_2) # (10) | ||
| 36 | - session.add(hero_3) # (11) | ||
| 37 | - | ||
| 38 | - print("After adding to the session") # (12) | ||
| 39 | - print("Hero 1:", hero_1) # (13) | ||
| 40 | - print("Hero 2:", hero_2) # (14) | ||
| 41 | - print("Hero 3:", hero_3) # (15) | ||
| 42 | - | ||
| 43 | - session.commit() # (16) | ||
| 44 | - | ||
| 45 | - print("After committing the session") # (17) | ||
| 46 | - print("Hero 1:", hero_1) # (18) | ||
| 47 | - print("Hero 2:", hero_2) # (19) | ||
| 48 | - print("Hero 3:", hero_3) # (20) | ||
| 49 | - | ||
| 50 | - print("After committing the session, show IDs") # (21) | ||
| 51 | - print("Hero 1 ID:", hero_1.id) # (22) | ||
| 52 | - print("Hero 2 ID:", hero_2.id) # (23) | ||
| 53 | - print("Hero 3 ID:", hero_3.id) # (24) | ||
| 54 | - | ||
| 55 | - print("After committing the session, show names") # (25) | ||
| 56 | - print("Hero 1 name:", hero_1.name) # (26) | ||
| 57 | - print("Hero 2 name:", hero_2.name) # (27) | ||
| 58 | - print("Hero 3 name:", hero_3.name) # (28) | ||
| 59 | - | ||
| 60 | - session.refresh(hero_1) # (29) | ||
| 61 | - session.refresh(hero_2) # (30) | ||
| 62 | - session.refresh(hero_3) # (31) | ||
| 63 | - | ||
| 64 | - print("After refreshing the heroes") # (32) | ||
| 65 | - print("Hero 1:", hero_1) # (33) | ||
| 66 | - print("Hero 2:", hero_2) # (34) | ||
| 67 | - print("Hero 3:", hero_3) # (35) | ||
| 68 | - # (36) | ||
| 69 | - | ||
| 70 | - print("After the session closes") # (37) | ||
| 71 | - print("Hero 1:", hero_1) # (38) | ||
| 72 | - print("Hero 2:", hero_2) # (39) | ||
| 73 | - print("Hero 3:", hero_3) # (40) | ||
| 24 | + hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") # (1)! | ||
| 25 | + hero_2 = Hero(name="Spider-Boy", secret_name="Pedro Parqueador") # (2)! | ||
| 26 | + hero_3 = Hero(name="Rusty-Man", secret_name="Tommy Sharp", age=48) # (3)! | ||
| 27 | + | ||
| 28 | + print("Before interacting with the database") # (4)! | ||
| 29 | + print("Hero 1:", hero_1) # (5)! | ||
| 30 | + print("Hero 2:", hero_2) # (6)! | ||
| 31 | + print("Hero 3:", hero_3) # (7)! | ||
| 32 | + | ||
| 33 | + with Session(engine) as session: # (8)! | ||
| 34 | + session.add(hero_1) # (9)! | ||
| 35 | + session.add(hero_2) # (10)! | ||
| 36 | + session.add(hero_3) # (11)! | ||
| 37 | + | ||
| 38 | + print("After adding to the session") # (12)! | ||
| 39 | + print("Hero 1:", hero_1) # (13)! | ||
| 40 | + print("Hero 2:", hero_2) # (14)! | ||
| 41 | + print("Hero 3:", hero_3) # (15)! | ||
| 42 | + | ||
| 43 | + session.commit() # (16)! | ||
| 44 | + | ||
| 45 | + print("After committing the session") # (17)! | ||
| 46 | + print("Hero 1:", hero_1) # (18)! | ||
| 47 | + print("Hero 2:", hero_2) # (19)! | ||
| 48 | + print("Hero 3:", hero_3) # (20)! | ||
| 49 | + | ||
| 50 | + print("After committing the session, show IDs") # (21)! | ||
| 51 | + print("Hero 1 ID:", hero_1.id) # (22)! | ||
| 52 | + print("Hero 2 ID:", hero_2.id) # (23)! | ||
| 53 | + print("Hero 3 ID:", hero_3.id) # (24)! | ||
| 54 | + | ||
| 55 | + print("After committing the session, show names") # (25)! | ||
| 56 | + print("Hero 1 name:", hero_1.name) # (26)! | ||
| 57 | + print("Hero 2 name:", hero_2.name) # (27)! | ||
| 58 | + print("Hero 3 name:", hero_3.name) # (28)! | ||
| 59 | + | ||
| 60 | + session.refresh(hero_1) # (29)! | ||
| 61 | + session.refresh(hero_2) # (30)! | ||
| 62 | + session.refresh(hero_3) # (31)! | ||
| 63 | + | ||
| 64 | + print("After refreshing the heroes") # (32)! | ||
| 65 | + print("Hero 1:", hero_1) # (33)! | ||
| 66 | + print("Hero 2:", hero_2) # (34)! | ||
| 67 | + print("Hero 3:", hero_3) # (35)! | ||
| 68 | + # (36)! | ||
| 69 | + | ||
| 70 | + print("After the session closes") # (37)! | ||
| 71 | + print("Hero 1:", hero_1) # (38)! | ||
| 72 | + print("Hero 2:", hero_2) # (39)! | ||
| 73 | + print("Hero 3:", hero_3) # (40)! | ||
| 74 | 74 | ||
| 75 | 75 | ||
| 76 | 76 | def main(): | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,24 +1,24 @@ | |||
| 1 | - from typing import Optional # (1) | ||
| 1 | + from typing import Optional # (1)! | ||
| 2 | 2 | ||
| 3 | - from sqlmodel import Field, SQLModel, create_engine # (2) | ||
| 3 | + from sqlmodel import Field, SQLModel, create_engine # (2)! | ||
| 4 | 4 | ||
| 5 | 5 | ||
| 6 | - class Hero(SQLModel, table=True): # (3) | ||
| 7 | - id: Optional[int] = Field(default=None, primary_key=True) # (4) | ||
| 8 | - name: str # (5) | ||
| 9 | - secret_name: str # (6) | ||
| 10 | - age: Optional[int] = None # (7) | ||
| 6 | + class Hero(SQLModel, table=True): # (3)! | ||
| 7 | + id: Optional[int] = Field(default=None, primary_key=True) # (4)! | ||
| 8 | + name: str # (5)! | ||
| 9 | + secret_name: str # (6)! | ||
| 10 | + age: Optional[int] = None # (7)! | ||
| 11 | 11 | ||
| 12 | 12 | ||
| 13 | - sqlite_file_name = "database.db" # (8) | ||
| 14 | - sqlite_url = f"sqlite:///{sqlite_file_name}" # (9) | ||
| 13 | + sqlite_file_name = "database.db" # (8)! | ||
| 14 | + sqlite_url = f"sqlite:///{sqlite_file_name}" # (9)! | ||
| 15 | 15 | ||
| 16 | - engine = create_engine(sqlite_url, echo=True) # (10) | ||
| 16 | + engine = create_engine(sqlite_url, echo=True) # (10)! | ||
| 17 | 17 | ||
| 18 | 18 | ||
| 19 | - def create_db_and_tables(): # (11) | ||
| 20 | - SQLModel.metadata.create_all(engine) # (12) | ||
| 19 | + def create_db_and_tables(): # (11)! | ||
| 20 | + SQLModel.metadata.create_all(engine) # (12)! | ||
| 21 | 21 | ||
| 22 | 22 | ||
| 23 | - if __name__ == "__main__": # (13) | ||
| 24 | - create_db_and_tables() # (14) | ||
| 23 | + if __name__ == "__main__": # (13)! | ||
| 24 | + create_db_and_tables() # (14)! | ||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -71,23 +71,23 @@ def update_heroes(): | |||
| 71 | 71 | ||
| 72 | 72 | def delete_heroes(): | |
| 73 | 73 | with Session(engine) as session: | |
| 74 | - statement = select(Hero).where(Hero.name == "Spider-Youngster") # (1) | ||
| 75 | - results = session.exec(statement) # (2) | ||
| 76 | - hero = results.one() # (3) | ||
| 77 | - print("Hero: ", hero) # (4) | ||
| 74 | + statement = select(Hero).where(Hero.name == "Spider-Youngster") # (1)! | ||
| 75 | + results = session.exec(statement) # (2)! | ||
| 76 | + hero = results.one() # (3)! | ||
| 77 | + print("Hero: ", hero) # (4)! | ||
| 78 | 78 | ||
| 79 | - session.delete(hero) # (5) | ||
| 80 | - session.commit() # (6) | ||
| 79 | + session.delete(hero) # (5)! | ||
| 80 | + session.commit() # (6)! | ||
| 81 | 81 | ||
| 82 | - print("Deleted hero:", hero) # (7) | ||
| 82 | + print("Deleted hero:", hero) # (7)! | ||
| 83 | 83 | ||
| 84 | - statement = select(Hero).where(Hero.name == "Spider-Youngster") # (8) | ||
| 85 | - results = session.exec(statement) # (9) | ||
| 86 | - hero = results.first() # (10) | ||
| 84 | + statement = select(Hero).where(Hero.name == "Spider-Youngster") # (8)! | ||
| 85 | + results = session.exec(statement) # (9)! | ||
| 86 | + hero = results.first() # (10)! | ||
| 87 | 87 | ||
| 88 | - if hero is None: # (11) | ||
| 89 | - print("There's no hero named Spider-Youngster") # (12) | ||
| 90 | - # (13) | ||
| 88 | + if hero is None: # (11)! | ||
| 89 | + print("There's no hero named Spider-Youngster") # (12)! | ||
| 90 | + # (13)! | ||
| 91 | 91 | ||
| 92 | 92 | ||
| 93 | 93 | def main(): | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,7 +1,7 @@ | |||
| 1 | 1 | from fastapi.testclient import TestClient | |
| 2 | 2 | from sqlmodel import Session, SQLModel, create_engine | |
| 3 | 3 | ||
| 4 | - from .main import app, get_session # (1) | ||
| 4 | + from .main import app, get_session # (1)! | ||
| 5 | 5 | ||
| 6 | 6 | ||
| 7 | 7 | def test_create_hero(): | |
@@ -17,16 +17,16 @@ def get_session_override(): | |||
| 17 | 17 | ||
| 18 | 18 | app.dependency_overrides[get_session] = get_session_override | |
| 19 | 19 | ||
| 20 | - client = TestClient(app) # (2) | ||
| 20 | + client = TestClient(app) # (2)! | ||
| 21 | 21 | ||
| 22 | - response = client.post( # (3) | ||
| 22 | + response = client.post( # (3)! | ||
| 23 | 23 | "/heroes/", json={"name": "Deadpond", "secret_name": "Dive Wilson"} | |
| 24 | 24 | ) | |
| 25 | 25 | app.dependency_overrides.clear() | |
| 26 | - data = response.json() # (4) | ||
| 26 | + data = response.json() # (4)! | ||
| 27 | 27 | ||
| 28 | - assert response.status_code == 200 # (5) | ||
| 29 | - assert data["name"] == "Deadpond" # (6) | ||
| 30 | - assert data["secret_name"] == "Dive Wilson" # (7) | ||
| 31 | - assert data["age"] is None # (8) | ||
| 32 | - assert data["id"] is not None # (9) | ||
| 28 | + assert response.status_code == 200 # (5)! | ||
| 29 | + assert data["name"] == "Deadpond" # (6)! | ||
| 30 | + assert data["secret_name"] == "Dive Wilson" # (7)! | ||
| 31 | + assert data["age"] is None # (8)! | ||
| 32 | + assert data["id"] is not None # (9)! | ||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,7 +1,7 @@ | |||
| 1 | 1 | from fastapi.testclient import TestClient | |
| 2 | 2 | from sqlmodel import Session, SQLModel, create_engine | |
| 3 | 3 | ||
| 4 | - from .main import app, get_session # (1) | ||
| 4 | + from .main import app, get_session # (1)! | ||
| 5 | 5 | ||
| 6 | 6 | ||
| 7 | 7 | def test_create_hero(): | |
@@ -12,17 +12,17 @@ def test_create_hero(): | |||
| 12 | 12 | ||
| 13 | 13 | with Session(engine) as session: | |
| 14 | 14 | ||
| 15 | - def get_session_override(): # (2) | ||
| 16 | - return session # (3) | ||
| 15 | + def get_session_override(): # (2)! | ||
| 16 | + return session # (3)! | ||
| 17 | 17 | ||
| 18 | - app.dependency_overrides[get_session] = get_session_override # (4) | ||
| 18 | + app.dependency_overrides[get_session] = get_session_override # (4)! | ||
| 19 | 19 | ||
| 20 | 20 | client = TestClient(app) | |
| 21 | 21 | ||
| 22 | 22 | response = client.post( | |
| 23 | 23 | "/heroes/", json={"name": "Deadpond", "secret_name": "Dive Wilson"} | |
| 24 | 24 | ) | |
| 25 | - app.dependency_overrides.clear() # (5) | ||
| 25 | + app.dependency_overrides.clear() # (5)! | ||
| 26 | 26 | data = response.json() | |
| 27 | 27 | ||
| 28 | 28 | assert response.status_code == 200 | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,21 +1,21 @@ | |||
| 1 | 1 | from fastapi.testclient import TestClient | |
| 2 | 2 | from sqlmodel import Session, SQLModel, create_engine | |
| 3 | 3 | ||
| 4 | - from .main import app, get_session # (1) | ||
| 4 | + from .main import app, get_session # (1)! | ||
| 5 | 5 | ||
| 6 | 6 | ||
| 7 | 7 | def test_create_hero(): | |
| 8 | - engine = create_engine( # (2) | ||
| 8 | + engine = create_engine( # (2)! | ||
| 9 | 9 | "sqlite:///testing.db", connect_args={"check_same_thread": False} | |
| 10 | 10 | ) | |
| 11 | - SQLModel.metadata.create_all(engine) # (3) | ||
| 11 | + SQLModel.metadata.create_all(engine) # (3)! | ||
| 12 | 12 | ||
| 13 | - with Session(engine) as session: # (4) | ||
| 13 | + with Session(engine) as session: # (4)! | ||
| 14 | 14 | ||
| 15 | 15 | def get_session_override(): | |
| 16 | - return session # (5) | ||
| 16 | + return session # (5)! | ||
| 17 | 17 | ||
| 18 | - app.dependency_overrides[get_session] = get_session_override # (4) | ||
| 18 | + app.dependency_overrides[get_session] = get_session_override # (4)! | ||
| 19 | 19 | ||
| 20 | 20 | client = TestClient(app) | |
| 21 | 21 | ||
@@ -30,4 +30,4 @@ def get_session_override(): | |||
| 30 | 30 | assert data["secret_name"] == "Dive Wilson" | |
| 31 | 31 | assert data["age"] is None | |
| 32 | 32 | assert data["id"] is not None | |
| 33 | - # (6) | ||
| 33 | + # (6)! | ||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,15 +1,15 @@ | |||
| 1 | 1 | from fastapi.testclient import TestClient | |
| 2 | 2 | from sqlmodel import Session, SQLModel, create_engine | |
| 3 | - from sqlmodel.pool import StaticPool # (1) | ||
| 3 | + from sqlmodel.pool import StaticPool # (1)! | ||
| 4 | 4 | ||
| 5 | 5 | from .main import app, get_session | |
| 6 | 6 | ||
| 7 | 7 | ||
| 8 | 8 | def test_create_hero(): | |
| 9 | 9 | engine = create_engine( | |
| 10 | - "sqlite://", # (2) | ||
| 10 | + "sqlite://", # (2)! | ||
| 11 | 11 | connect_args={"check_same_thread": False}, | |
| 12 | - poolclass=StaticPool, # (3) | ||
| 12 | + poolclass=StaticPool, # (3)! | ||
| 13 | 13 | ) | |
| 14 | 14 | SQLModel.metadata.create_all(engine) | |
| 15 | 15 | ||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,24 +1,24 @@ | |||
| 1 | - import pytest # (1) | ||
| 1 | + import pytest # (1)! | ||
| 2 | 2 | from fastapi.testclient import TestClient | |
| 3 | 3 | from sqlmodel import Session, SQLModel, create_engine | |
| 4 | 4 | from sqlmodel.pool import StaticPool | |
| 5 | 5 | ||
| 6 | 6 | from .main import app, get_session | |
| 7 | 7 | ||
| 8 | 8 | ||
| 9 | - @pytest.fixture(name="session") # (2) | ||
| 10 | - def session_fixture(): # (3) | ||
| 9 | + @pytest.fixture(name="session") # (2)! | ||
| 10 | + def session_fixture(): # (3)! | ||
| 11 | 11 | engine = create_engine( | |
| 12 | 12 | "sqlite://", connect_args={"check_same_thread": False}, poolclass=StaticPool | |
| 13 | 13 | ) | |
| 14 | 14 | SQLModel.metadata.create_all(engine) | |
| 15 | 15 | with Session(engine) as session: | |
| 16 | - yield session # (4) | ||
| 16 | + yield session # (4)! | ||
| 17 | 17 | ||
| 18 | 18 | ||
| 19 | - def test_create_hero(session: Session): # (5) | ||
| 19 | + def test_create_hero(session: Session): # (5)! | ||
| 20 | 20 | def get_session_override(): | |
| 21 | - return session # (6) | ||
| 21 | + return session # (6)! | ||
| 22 | 22 | ||
| 23 | 23 | app.dependency_overrides[get_session] = get_session_override | |
| 24 | 24 | ||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -16,19 +16,19 @@ def session_fixture(): | |||
| 16 | 16 | yield session | |
| 17 | 17 | ||
| 18 | 18 | ||
| 19 | - @pytest.fixture(name="client") # (1) | ||
| 20 | - def client_fixture(session: Session): # (2) | ||
| 21 | - def get_session_override(): # (3) | ||
| 19 | + @pytest.fixture(name="client") # (1)! | ||
| 20 | + def client_fixture(session: Session): # (2)! | ||
| 21 | + def get_session_override(): # (3)! | ||
| 22 | 22 | return session | |
| 23 | 23 | ||
| 24 | - app.dependency_overrides[get_session] = get_session_override # (4) | ||
| 24 | + app.dependency_overrides[get_session] = get_session_override # (4)! | ||
| 25 | 25 | ||
| 26 | - client = TestClient(app) # (5) | ||
| 27 | - yield client # (6) | ||
| 28 | - app.dependency_overrides.clear() # (7) | ||
| 26 | + client = TestClient(app) # (5)! | ||
| 27 | + yield client # (6)! | ||
| 28 | + app.dependency_overrides.clear() # (7)! | ||
| 29 | 29 | ||
| 30 | 30 | ||
| 31 | - def test_create_hero(client: TestClient): # (8) | ||
| 31 | + def test_create_hero(client: TestClient): # (8)! | ||
| 32 | 32 | response = client.post( | |
| 33 | 33 | "/heroes/", json={"name": "Deadpond", "secret_name": "Dive Wilson"} | |
| 34 | 34 | ) | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -20,24 +20,24 @@ def create_db_and_tables(): | |||
| 20 | 20 | SQLModel.metadata.create_all(engine) | |
| 21 | 21 | ||
| 22 | 22 | ||
| 23 | - def create_heroes(): # (1) | ||
| 24 | - hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") # (2) | ||
| 23 | + def create_heroes(): # (1)! | ||
| 24 | + hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") # (2)! | ||
| 25 | 25 | hero_2 = Hero(name="Spider-Boy", secret_name="Pedro Parqueador") | |
| 26 | 26 | hero_3 = Hero(name="Rusty-Man", secret_name="Tommy Sharp", age=48) | |
| 27 | 27 | ||
| 28 | - with Session(engine) as session: # (3) | ||
| 29 | - session.add(hero_1) # (4) | ||
| 28 | + with Session(engine) as session: # (3)! | ||
| 29 | + session.add(hero_1) # (4)! | ||
| 30 | 30 | session.add(hero_2) | |
| 31 | 31 | session.add(hero_3) | |
| 32 | 32 | ||
| 33 | - session.commit() # (5) | ||
| 34 | - # (6) | ||
| 33 | + session.commit() # (5)! | ||
| 34 | + # (6)! | ||
| 35 | 35 | ||
| 36 | 36 | ||
| 37 | - def main(): # (7) | ||
| 38 | - create_db_and_tables() # (8) | ||
| 39 | - create_heroes() # (9) | ||
| 37 | + def main(): # (7)! | ||
| 38 | + create_db_and_tables() # (8)! | ||
| 39 | + create_heroes() # (9)! | ||
| 40 | 40 | ||
| 41 | 41 | ||
| 42 | - if __name__ == "__main__": # (10) | ||
| 43 | - main() # (11) | ||
| 42 | + if __name__ == "__main__": # (10)! | ||
| 43 | + main() # (11)! | ||
| Back | FazBrowse Home | New Git URL |
0 commit comments