|
1 | | -from django.apps import apps as global_apps |
2 | | -from django.conf import settings |
3 | | -from django.contrib.contenttypes.management import create_contenttypes |
4 | 1 | from django.db import migrations |
5 | 2 |
|
6 | 3 |
|
7 | | -try: |
8 | | - from djangocms_versioning.constants import DRAFT, PUBLISHED |
9 | | - |
10 | | - djangocms_versioning_installed = True |
11 | | -except ImportError: |
12 | | - djangocms_versioning_installed = False |
13 | | - |
14 | | -djangocms_versioning_config_enabled = getattr( |
15 | | - settings, 'DJANGOCMS_SNIPPET_VERSIONING_ENABLED', True |
16 | | -) |
17 | | - |
18 | | - |
19 | 4 | def cms4_grouper_version_migration(apps, schema_editor): |
20 | | - create_contenttypes(global_apps.get_app_config("djangocms_snippet")) |
21 | | - |
| 5 | + db_alias = schema_editor.connection.alias |
22 | 6 |
|
23 | | - |
24 | | - ContentType = apps.get_model('contenttypes', 'ContentType') |
25 | 7 | Snippet = apps.get_model('djangocms_snippet', 'Snippet') |
26 | 8 | SnippetGrouper = apps.get_model('djangocms_snippet', 'SnippetGrouper') |
27 | | - User = apps.get_model(*settings.AUTH_USER_MODEL.split('.')) |
28 | | - |
29 | | - snippet_contenttype = ContentType.objects.get(app_label='djangocms_snippet', model='snippet') |
30 | | - snippet_queryset = Snippet.objects.all() |
31 | 9 |
|
32 | | - # Get a migration user to create a version. |
33 | | - if djangocms_versioning_config_enabled and djangocms_versioning_installed and len(snippet_queryset): |
34 | | - Version = apps.get_model('djangocms_versioning', 'Version') |
35 | | - migration_user = User.objects.get(id=getattr(settings, "DJANGOCMS_SNIPPET_VERSIONING_MIGRATION_USER_ID", 1)) |
| 10 | + snippet_queryset = Snippet.objects.using(db_alias).all() |
36 | 11 |
|
37 | 12 | for snippet in snippet_queryset: |
38 | 13 | grouper = SnippetGrouper.objects.create() |
39 | 14 | snippet.snippet_grouper = grouper |
40 | 15 | snippet.save() |
41 | 16 |
|
42 | | - # Create initial Snippet Versions if versioning is enabled and installed. |
43 | | - # Publish the snippet because all snippets were assumed published before |
44 | | - if djangocms_versioning_config_enabled and djangocms_versioning_installed: |
45 | | - Version.objects.create( |
46 | | - created_by=migration_user, |
47 | | - state=PUBLISHED, |
48 | | - number=1, |
49 | | - object_id=snippet.pk, |
50 | | - content_type=snippet_contenttype, |
51 | | - ) |
52 | | - |
53 | 17 |
|
54 | 18 | class Migration(migrations.Migration): |
55 | 19 | dependencies = [ |
56 | | - # ('cms', '0034_remove_pagecontent_placeholders'), # Run after the CMS4 migrations |
57 | 20 | ('djangocms_snippet', '0009_auto_20210915_0445'), |
58 | 21 | ] |
59 | 22 |
|
60 | | - if djangocms_versioning_installed: |
61 | | - dependencies += [('djangocms_versioning', '0015_version_modified'), ] |
62 | | - |
63 | 23 | operations = [ |
64 | 24 | migrations.RunPython(cms4_grouper_version_migration) |
65 | 25 | ] |
0 commit comments