|
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