mirrors/bookwyrm
1
0
Fork 1
mirror of https://github.com/bookwyrm-social/bookwyrm.git synced 2025-01-01 12:58:45 +00:00
Code Issues Projects Releases Wiki Activity

Create 0134_alter_stopped_reading.py

Browse source
This commit is contained in:
Mouse Reeve 2022-02-25 18:08:30 -08:00 committed by GitHub
parent 8deee2220e
commit b511928400
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
bookwyrm/migrations/0134_alter_stopped_reading.py
View file

@ -3,6 +3,22 @@
import bookwyrm.models.fields
from django.db import migrations
def add_shelves(apps, schema_editor):
"""add any superusers to the "admin" group"""
db_alias = schema_editor.connection.alias
shelf_model = apps.get_model("bookwyrm", "Shelf")
users = apps.get_model("bookwyrm", "User")
local_users = users.objects.using(db_alias).filter(local=True)
for user in local_users:
shelf_model.using(db_alias)(
name="Stopped reading",
identifier=Shelf.STOPPED_READING,
user=user,
editable=False,
).save(broadcast=False)
class Migration(migrations.Migration):
@ -56,4 +72,5 @@ class Migration(migrations.Migration):
null=True,
),
),
migrations.RunPython(add_shelves, reverse_code=migrations.RunPython.noop),
]