Skip to content

Commit e601a2c

Browse files
authored
Merge pull request #103 from tecladocode/develop
2 parents 77fe34e + 25684cb commit e601a2c

File tree

65 files changed

+527
-41
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

65 files changed

+527
-41
lines changed
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
1-
from models.user import UserModel
21
from models.item import ItemModel
3-
from models.tag import TagModel
42
from models.store import StoreModel
5-
from models.item_tags import ItemsTags
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
flask
2-
flask-smorest
32
flask-sqlalchemy
3+
flask-smorest
44
python-dotenv
55
marshmallow
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
flask
2+
flask-sqlalchemy
23
flask-smorest
34
python-dotenv
45
marshmallow
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
---
2+
title: Delete models with relationships
3+
description: Tell SQLAlchemy what to do with related models when you delete the parent.
4+
---
5+
6+
# Delete models with relationships using cascades
7+
8+
When you delete a model that has a relationship to other models that still exist, the default behavior in SQLAlchemy with PostgreSQL is to raise an error. This is because SQLAlchemy does not want to allow you to accidentally delete data that is still being used by other models.
9+
10+
Let's say you have a `Store 1` that has two items, `Item 1` and `Item 2`. If you try to delete Store 1 without first deleting Item 1 and Item 2, SQLAlchemy will raise an error because the items are still related to the store.
11+
12+
This means the items have a **Foreign Key** that references the store you're trying to delete. If the store actually was deleted, the items have a store ID that references something that doesn't exist.
13+
14+
To fix this, you can use a feature called "cascading deletes". Cascading deletes allow you to specify that when a model is deleted, any related models should also be deleted automatically.
15+
16+
SQLAlchemy makes it easy to add cascades to our models, here's how you might do that!
17+
18+
```python title="models/store.py"
19+
from db import db
20+
21+
22+
class StoreModel(db.Model):
23+
__tablename__ = "stores"
24+
25+
id = db.Column(db.Integer, primary_key=True)
26+
name = db.Column(db.String(80), unique=True, nullable=False)
27+
28+
# highlight-start
29+
items = db.relationship("ItemModel", back_populates="store", lazy="dynamic", cascade="all, delete")
30+
# highlight-end
31+
```
32+
33+
Remember that `StoreModel` and `ItemModel` have a one-to-many relationship, where each store can have multiple items, and each item belongs to a single store.
34+
35+
The `cascade="all,delete"` argument in the `relationship()` call for the `StoreModel.items` attribute specifies that when a store is deleted, all of its related items should also be deleted.
36+
37+
If you add a `cascade` on the relationship in the `ItemModel`, then when an item is deleted, its related store should also be deleted. This is not what we want, so we won't add a cascade to `ItemModel`.
38+
39+
With this code in place, if you try to delete a store that still has items, the items will be deleted automatically along with the store. This will allow you to delete the store without having to delete the items individually.
40+
41+
For more information, I strongly recommend reading [the official documentation](https://docs.sqlalchemy.org/en/20/orm/cascades.html#delete)! There are also other cascade options you can pass in depending on what you want to happen to related models when the parent changes or is deleted.
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
FLASK_APP=app
2+
FLASK_DEBUG=True
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
FROM python:3.10
2+
EXPOSE 5000
3+
WORKDIR /app
4+
COPY ./requirements.txt requirements.txt
5+
RUN pip install --no-cache-dir --upgrade -r requirements.txt
6+
COPY . .
7+
CMD ["flask", "run", "--host", "0.0.0.0"]
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
from flask import Flask
2+
from flask_smorest import Api
3+
4+
from db import db
5+
6+
import models
7+
8+
from resources.item import blp as ItemBlueprint
9+
from resources.store import blp as StoreBlueprint
10+
11+
12+
def create_app(db_url=None):
13+
app = Flask(__name__)
14+
app.config["API_TITLE"] = "Stores REST API"
15+
app.config["API_VERSION"] = "v1"
16+
app.config["OPENAPI_VERSION"] = "3.0.3"
17+
app.config["OPENAPI_URL_PREFIX"] = "/"
18+
app.config["OPENAPI_SWAGGER_UI_PATH"] = "/swagger-ui"
19+
app.config[
20+
"OPENAPI_SWAGGER_UI_URL"
21+
] = "https://cdn.jsdelivr.net/npm/swagger-ui-dist/"
22+
app.config["SQLALCHEMY_DATABASE_URI"] = db_url or "sqlite:///data.db"
23+
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
24+
app.config["PROPAGATE_EXCEPTIONS"] = True
25+
db.init_app(app)
26+
api = Api(app)
27+
28+
with app.app_context():
29+
db.create_all()
30+
31+
api.register_blueprint(ItemBlueprint)
32+
api.register_blueprint(StoreBlueprint)
33+
34+
return app
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
from flask_sqlalchemy import SQLAlchemy
2+
3+
db = SQLAlchemy()
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
from models.item import ItemModel
2+
from models.store import StoreModel
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
from db import db
2+
3+
4+
class ItemModel(db.Model):
5+
__tablename__ = "items"
6+
7+
id = db.Column(db.Integer, primary_key=True)
8+
name = db.Column(db.String(80), unique=False, nullable=False)
9+
price = db.Column(db.Float(precision=2), unique=False, nullable=False)
10+
11+
store_id = db.Column(
12+
db.Integer, db.ForeignKey("stores.id"), unique=False, nullable=False
13+
)
14+
store = db.relationship("StoreModel", back_populates="items")
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
from db import db
2+
3+
4+
class StoreModel(db.Model):
5+
__tablename__ = "stores"
6+
7+
id = db.Column(db.Integer, primary_key=True)
8+
name = db.Column(db.String(80), unique=True, nullable=False)
9+
10+
items = db.relationship(
11+
"ItemModel", back_populates="store", lazy="dynamic", cascade="all, delete"
12+
)
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
flask
2+
flask-sqlalchemy
3+
flask-smorest
4+
python-dotenv
5+
marshmallow

docs/docs/06_sql_storage_sqlalchemy/10_delete_related_models_sqlalchemy/end/resources/__init__.py

Whitespace-only changes.
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,59 @@
1+
from flask.views import MethodView
2+
from flask_smorest import Blueprint, abort
3+
from sqlalchemy.exc import SQLAlchemyError
4+
5+
from db import db
6+
from models import ItemModel
7+
from schemas import ItemSchema, ItemUpdateSchema
8+
9+
blp = Blueprint("Items", "items", description="Operations on items")
10+
11+
12+
@blp.route("/item/<string:item_id>")
13+
class Item(MethodView):
14+
@blp.response(200, ItemSchema)
15+
def get(self, item_id):
16+
item = ItemModel.query.get_or_404(item_id)
17+
return item
18+
19+
def delete(self, item_id):
20+
item = ItemModel.query.get_or_404(item_id)
21+
db.session.delete(item)
22+
db.session.commit()
23+
return {"message": "Item deleted."}
24+
25+
@blp.arguments(ItemUpdateSchema)
26+
@blp.response(200, ItemSchema)
27+
def put(self, item_data, item_id):
28+
item = ItemModel.query.get(item_id)
29+
30+
if item:
31+
item.price = item_data["price"]
32+
item.name = item_data["name"]
33+
else:
34+
item = ItemModel(id=item_id, **item_data)
35+
36+
db.session.add(item)
37+
db.session.commit()
38+
39+
return item
40+
41+
42+
@blp.route("/item")
43+
class ItemList(MethodView):
44+
@blp.response(200, ItemSchema(many=True))
45+
def get(self):
46+
return ItemModel.query.all()
47+
48+
@blp.arguments(ItemSchema)
49+
@blp.response(201, ItemSchema)
50+
def post(self, item_data):
51+
item = ItemModel(**item_data)
52+
53+
try:
54+
db.session.add(item)
55+
db.session.commit()
56+
except SQLAlchemyError:
57+
abort(500, message="An error occurred while inserting the item.")
58+
59+
return item

0 commit comments

Comments
 (0)