mirror of
https://github.com/netbox-community/netbox.git
synced 2025-07-13 16:47:34 -06:00
a9af541e81
* 19529 fix custom script path * 19529 fix custom script path * 19529 fix custom script path * 19529 fix custom script path * 19680 add object_change migrator * 19680 optimize migration * 19680 optimize migration
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
from django.conf import settings
|
|
from django.core.files.storage import storages
|
|
from django.db import migrations
|
|
from urllib.parse import urlparse
|
|
|
|
from extras.storage import ScriptFileSystemStorage
|
|
|
|
|
|
def normalize(url):
|
|
parsed_url = urlparse(url)
|
|
if not parsed_url.path.endswith('/'):
|
|
return url + '/'
|
|
return url
|
|
|
|
|
|
def fix_script_paths(apps, schema_editor):
|
|
"""
|
|
Fix script paths for scripts that had incorrect path from NB 4.3.
|
|
"""
|
|
storage = storages.create_storage(storages.backends["scripts"])
|
|
if not isinstance(storage, ScriptFileSystemStorage):
|
|
return
|
|
|
|
ScriptModule = apps.get_model('extras', 'ScriptModule')
|
|
script_root_path = normalize(settings.SCRIPTS_ROOT)
|
|
for script in ScriptModule.objects.filter(file_path__startswith=script_root_path):
|
|
script.file_path = script.file_path[len(script_root_path):]
|
|
script.save()
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('extras', '0128_tableconfig'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(code=fix_script_paths, reverse_code=migrations.RunPython.noop),
|
|
]
|
|
|
|
|
|
def oc_fix_script_paths(objectchange, reverting):
|
|
script_root_path = normalize(settings.SCRIPTS_ROOT)
|
|
|
|
for data in (objectchange.prechange_data, objectchange.postchange_data):
|
|
if data is None:
|
|
continue
|
|
|
|
if file_path := data.get('file_path'):
|
|
if file_path.startswith(script_root_path):
|
|
data['file_path'] = file_path[len(script_root_path):]
|
|
|
|
|
|
objectchange_migrators = {
|
|
'extras.scriptmodule': oc_fix_script_paths,
|
|
}
|