Эх сурвалжийг харах

HUE-8758 [model] Adding Document2 connector FK migration org off

Catching-up on one missing beewax migration
Romain 5 жил өмнө
parent
commit
2c5947c7de

+ 25 - 0
apps/beeswax/src/beeswax/migrations/0002_auto_20200320_0746.py

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.28 on 2020-03-20 14:46
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('beeswax', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='session',
+            name='guid',
+            field=models.TextField(max_length='100'),
+        ),
+        migrations.AlterField(
+            model_name='session',
+            name='secret',
+            field=models.TextField(max_length='100'),
+        ),
+    ]

+ 21 - 0
desktop/core/src/desktop/migrations/0011_document2_connector.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.28 on 2020-03-20 14:46
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('desktop', '0010_auto_20200115_0908'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='document2',
+            name='connector',
+            field=models.ForeignKey(blank=True, help_text='Connector.', null=True, on_delete=django.db.models.deletion.CASCADE, to='desktop.Connector', verbose_name='Connector'),
+        ),
+    ]

+ 9 - 0
desktop/core/src/desktop/models.py

@@ -52,6 +52,7 @@ from desktop.conf import get_clusters, CLUSTER_ID, IS_MULTICLUSTER_ONLY, IS_K8S_
     has_connectors, TASK_SERVER, ENABLE_GIST, APP_BLACKLIST
 from desktop.lib import fsmanager
 from desktop.lib.connectors.api import _get_installed_connectors
+from desktop.lib.connectors.models import Connector
 from desktop.lib.i18n import force_unicode
 from desktop.lib.exceptions_renderable import PopupException
 from desktop.lib.paths import get_run_root, SAFE_CHARACTERS_URI_COMPONENTS
@@ -1132,6 +1133,14 @@ class Document2(models.Model):
       db_index=True,
       help_text=_t('Type of document, e.g. Hive query, Oozie workflow, Search Dashboard...')
   )
+  connector = models.ForeignKey(
+      Connector,
+      verbose_name=_t('Connector'),
+      help_text=_t('Connector.'),
+      blank=True,
+      null=True,
+      db_index=True
+  )
 
   data = models.TextField(default='{}')
   extra = models.TextField(default='')