NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-18 05:28:16 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Clean up migration

Browse Source
This commit is contained in:
Jeremy Stretch 2024-02-07 09:10:04 -05:00
parent c5916af1d0
commit 7a10b4ef89
2 changed files with 31 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
netbox/extras/migrations/0107_convert_reports_to_scripts.py Normal file
View File

@ -0,0 +1,31 @@
from django.db import migrations
def convert_reportmodule_jobs(apps, schema_editor):
ContentType = apps.get_model('contenttypes', 'ContentType')
Job = apps.get_model('core', 'Job')
# Convert all ReportModule jobs to ScriptModule jobs
if reportmodule_ct := ContentType.objects.filter(app_label='extras', model='reportmodule').first():
scriptmodule_ct = ContentType.objects.get(app_label='extras', model='scriptmodule')
Job.objects.filter(object_type_id=reportmodule_ct.id).update(object_type_id=scriptmodule_ct.id)
class Migration(migrations.Migration):
dependencies = [
('extras', '0106_bookmark_user_cascade_deletion'),
]
operations = [
migrations.RunPython(
code=convert_reportmodule_jobs,
reverse_code=migrations.RunPython.noop
),
migrations.DeleteModel(
name='Report',
),
migrations.DeleteModel(
name='ReportModule',
),
]

34
netbox/extras/migrations/0107_delete_report_delete_reportmodule.py
View File

@ -1,34 +0,0 @@
# Generated by Django 5.0.1 on 2024-01-26 22:22
from django.db import migrations
def migrate_report_jobs(apps, schema_editor):
ContentType = apps.get_model('contenttypes', 'ContentType')
Job = apps.get_model('core', 'Job')
# Delete the new ContentType effected by the introduction of core.ConfigRevision
if ContentType.objects.filter(app_label='extras', model='reportmodule'):
report_content_type = ContentType.objects.get(app_label='extras', model='reportmodule')
script_content_type = ContentType.objects.get(app_label='extras', model='scriptmodule')
jobs = Job.objects.filter(object_type_id=report_content_type.id).update(object_type_id=script_content_type.id)
class Migration(migrations.Migration):
dependencies = [
('extras', '0106_bookmark_user_cascade_deletion'),
]
operations = [
migrations.RunPython(
code=migrate_report_jobs,
reverse_code=migrations.RunPython.noop
),
migrations.DeleteModel(
name='Report',
),
migrations.DeleteModel(
name='ReportModule',
),
]