altf4arnold/datatracker
1
0
Fork 0
Code Issues Pull requests Actions 2 Packages Projects Releases Wiki Activity

chore: Migrate SearchRules for the rfc state

Browse source
This commit is contained in:
Jennifer Richards 2023-07-11 16:14:49 -03:00
parent 71270621b7
commit 5385760f2a
No known key found for this signature in database
GPG key ID: 9B2BF5C5ADDA6A6E
1 changed files with 22 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
ietf/community/migrations/0003_track_rfcs.py
View file

@ -8,6 +8,7 @@ def forward(apps, schema_editor):
CommunityList = apps.get_model("community", "CommunityList")
RelatedDocument = apps.get_model("doc", "RelatedDocument")
# Handle individually tracked documents
for cl in CommunityList.objects.all():
for rfc in set(
RelatedDocument.objects.filter(
@ -17,11 +18,30 @@ def forward(apps, schema_editor):
):
cl.added_docs.add(rfc)
# Handle rules
SearchRule = apps.get_model("community", "SearchRule")
State = apps.get_model("doc", "State")
draft_rfc_state = State.objects.get(type_id="draft", slug="rfc")
rfc_published_state = State.objects.get(type_id="rfc", slug="published")
SearchRule.objects.filter(state=draft_rfc_state).update(state=rfc_published_state)
def reverse(apps, schema_editor):
Document = apps.get_model("doc", "Document")
for rfc in Document.objects.filter(type__slug="rfc"):
rfc.communitylist_set.clear()
SearchRule = apps.get_model("community", "SearchRule")
State = apps.get_model("doc", "State")
draft_rfc_state = State.objects.get(type_id="draft", slug="rfc")
rfc_published_state = State.objects.get(type_id="rfc", slug="published")
SearchRule.objects.filter(state=rfc_published_state).update(state=draft_rfc_state)
class Migration(migrations.Migration):
dependencies = [
("community", "0002_auto_20230320_1222"),
("doc", "0010_move_rfc_docaliases"),
]
operations = [migrations.RunPython(forward)]
operations = [migrations.RunPython(forward, reverse)]