Add Domain support.

closes #3008

(based on PR #2920)
Co-authored by: Grant Gainey <ggainey@redhat.com>

CHANGES/3008.feature

@@ -0,0 +1 @@
+Added support for Domains.

pulp_rpm/app/__init__.py

@@ -10,3 +10,4 @@ class PulpRpmPluginAppConfig(PulpPluginAppConfig):
     label = "rpm"
     version = "3.21.0.dev"
     python_package_name = "pulp-rpm"
+    domain_compatible = True

pulp_rpm/app/migrations/0051_domains.py

@@ -0,0 +1,150 @@
+# Generated by Django 3.2.18 on 2023-04-19 00:43
+
+from django.db import migrations, models
+import django.db.models.deletion
+import pulpcore.app.util
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('core', '0102_add_domain_relations'),
+        ('rpm', '0050_alter_addon_pulp_id_alter_checksum_pulp_id_and_more'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='distributiontree',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_distributiontree', to='core.domain'),
+        ),
+        migrations.AddField(
+            model_name='modulemd',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_modulemd', to='core.domain'),
+        ),
+        migrations.AddField(
+            model_name='modulemddefaults',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_modulemddefaults', to='core.domain'),
+        ),
+        migrations.AddField(
+            model_name='modulemdobsolete',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_modulemdobsolete', to='core.domain'),
+        ),
+        migrations.AddField(
+            model_name='package',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_package', to='core.domain'),
+        ),
+        migrations.AddField(
+            model_name='packagecategory',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_packagecategory', to='core.domain'),
+        ),
+        migrations.AddField(
+            model_name='packageenvironment',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_packageenvironment', to='core.domain'),
+        ),
+        migrations.AddField(
+            model_name='packagegroup',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_packagegroup', to='core.domain'),
+        ),
+        migrations.AddField(
+            model_name='packagelangpacks',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_packagelangpacks', to='core.domain'),
+        ),
+        migrations.AddField(
+            model_name='repometadatafile',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_repometadatafile', to='core.domain'),
+        ),
+        migrations.AddField(
+            model_name='updaterecord',
+            name='_pulp_domain',
+            field=models.ForeignKey(default=pulpcore.app.util.get_domain_pk, on_delete=django.db.models.deletion.PROTECT, related_name='rpm_updaterecord', to='core.domain'),
+        ),
+        migrations.AlterField(
+            model_name='modulemddefaults',
+            name='digest',
+            field=models.TextField(db_index=True),
+        ),
+        migrations.AlterField(
+            model_name='package',
+            name='pkgId',
+            field=models.TextField(db_index=True),
+        ),
+        migrations.AlterField(
+            model_name='packagecategory',
+            name='digest',
+            field=models.TextField(db_index=True),
+        ),
+        migrations.AlterField(
+            model_name='packageenvironment',
+            name='digest',
+            field=models.TextField(db_index=True),
+        ),
+        migrations.AlterField(
+            model_name='packagegroup',
+            name='digest',
+            field=models.TextField(db_index=True),
+        ),
+        migrations.AlterField(
+            model_name='packagelangpacks',
+            name='digest',
+            field=models.TextField(db_index=True),
+        ),
+        migrations.AlterField(
+            model_name='updaterecord',
+            name='digest',
+            field=models.TextField(db_index=True),
+        ),
+        migrations.AlterUniqueTogether(
+            name='distributiontree',
+            unique_together={('_pulp_domain', 'digest')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='modulemd',
+            unique_together={('_pulp_domain', 'name', 'stream', 'version', 'context', 'arch')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='modulemddefaults',
+            unique_together={('_pulp_domain', 'digest')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='modulemdobsolete',
+            unique_together={('_pulp_domain', 'modified', 'module_name', 'module_stream')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='package',
+            unique_together={('_pulp_domain', 'name', 'epoch', 'version', 'release', 'arch', 'checksum_type', 'pkgId')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='packagecategory',
+            unique_together={('_pulp_domain', 'digest')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='packageenvironment',
+            unique_together={('_pulp_domain', 'digest')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='packagegroup',
+            unique_together={('_pulp_domain', 'digest')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='packagelangpacks',
+            unique_together={('_pulp_domain', 'digest')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='repometadatafile',
+            unique_together={('_pulp_domain', 'data_type', 'checksum', 'relative_path')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='updaterecord',
+            unique_together={('_pulp_domain', 'digest')},
+        ),
+    ]

pulp_rpm/app/models/advisory.py

@@ -8,6 +8,7 @@
     BaseModel,
     Content,
 )
+from pulpcore.plugin.util import get_domain_pk
 
 from pulp_rpm.app.constants import (
     CR_UPDATE_COLLECTION_ATTRS,
@@ -100,14 +101,9 @@ class UpdateRecord(Content):
     # A field that represents the hash digest of the update record. Used to track differences
     # between two UpdateRecord objects without having to examine the associations like
     # UpdateCollection or UpdateCollectionPackage.
-    digest = models.TextField(unique=True)
+    digest = models.TextField(db_index=True)
 
-    @classmethod
-    def natural_key_fields(cls):
-        """
-        Digest is used as a natural key for UpdateRecords.
-        """
-        return ("digest",)
+    _pulp_domain = models.ForeignKey("core.Domain", default=get_domain_pk, on_delete=models.PROTECT)
 
     @classmethod
     def createrepo_to_dict(cls, update):
@@ -228,6 +224,7 @@ def get_module_list(self):
 
     class Meta:
         default_related_name = "%(app_label)s_%(model_name)s"
+        unique_together = ("_pulp_domain", "digest")
 
 
 class UpdateCollection(BaseModel):

pulp_rpm/app/models/comps.py

@@ -5,6 +5,7 @@
 from django.db import models
 
 from pulpcore.plugin.models import Content
+from pulpcore.plugin.util import get_domain_pk
 
 from pulp_rpm.app.constants import (
     LIBCOMPS_CATEGORY_ATTRS,
@@ -73,19 +74,15 @@ class PackageGroup(Content):
     desc_by_lang = models.JSONField(default=dict)
     name_by_lang = models.JSONField(default=dict)
 
-    digest = models.TextField(unique=True)
+    digest = models.TextField(db_index=True)
 
     repo_key_fields = ("id",)
 
+    _pulp_domain = models.ForeignKey("core.Domain", default=get_domain_pk, on_delete=models.PROTECT)
+
     class Meta:
         default_related_name = "%(app_label)s_%(model_name)s"
-
-    @classmethod
-    def natural_key_fields(cls):
-        """
-        Digest is used as a natural key for PackageGroups.
-        """
-        return ("digest",)
+        unique_together = ("_pulp_domain", "digest")
 
     @classmethod
     def pkglist_to_list(cls, value):
@@ -230,19 +227,15 @@ class PackageCategory(Content):
     desc_by_lang = models.JSONField(default=dict)
     name_by_lang = models.JSONField(default=dict)
 
-    digest = models.TextField(unique=True)
+    digest = models.TextField(db_index=True)
 
     repo_key_fields = ("id",)
 
+    _pulp_domain = models.ForeignKey("core.Domain", default=get_domain_pk, on_delete=models.PROTECT)
+
     class Meta:
         default_related_name = "%(app_label)s_%(model_name)s"
-
-    @classmethod
-    def natural_key_fields(cls):
-        """
-        Digest is used as a natural key for PackageCategory.
-        """
-        return ("digest",)
+        unique_together = ("_pulp_domain", "digest")
 
     @classmethod
     def grplist_to_lst(cls, value):
@@ -357,19 +350,15 @@ class PackageEnvironment(Content):
     desc_by_lang = models.JSONField(default=dict)
     name_by_lang = models.JSONField(default=dict)
 
-    digest = models.TextField(unique=True)
+    digest = models.TextField(db_index=True)
 
     repo_key_fields = ("id",)
 
+    _pulp_domain = models.ForeignKey("core.Domain", default=get_domain_pk, on_delete=models.PROTECT)
+
     class Meta:
         default_related_name = "%(app_label)s_%(model_name)s"
-
-    @classmethod
-    def natural_key_fields(cls):
-        """
-        Digest is used as a natural key for PackageEnvironment.
-        """
-        return ("digest",)
+        unique_together = ("_pulp_domain", "digest")
 
     @classmethod
     def grplist_to_lst(cls, value):
@@ -463,17 +452,13 @@ class PackageLangpacks(Content):
 
     matches = models.JSONField(default=dict)
 
-    digest = models.TextField(unique=True)
+    digest = models.TextField(db_index=True)
+
+    _pulp_domain = models.ForeignKey("core.Domain", default=get_domain_pk, on_delete=models.PROTECT)
 
     class Meta:
         default_related_name = "%(app_label)s_%(model_name)s"
-
-    @classmethod
-    def natural_key_fields(cls):
-        """
-        Digest is used as a natural key for PackageLangpacks.
-        """
-        return ("digest",)
+        unique_together = ("_pulp_domain", "digest")
 
     @classmethod
     def libcomps_to_dict(cls, langpacks):

pulp_rpm/app/models/custom_metadata.py

@@ -3,6 +3,7 @@
 from django.db import models
 
 from pulpcore.plugin.models import Content
+from pulpcore.plugin.util import get_domain_pk
 from pulp_rpm.app.constants import CHECKSUM_CHOICES
 
 log = getLogger(__name__)
@@ -32,9 +33,11 @@ class RepoMetadataFile(Content):
 
     repo_key_fields = ("data_type",)
 
+    _pulp_domain = models.ForeignKey("core.Domain", default=get_domain_pk, on_delete=models.PROTECT)
+
     class Meta:
         default_related_name = "%(app_label)s_%(model_name)s"
-        unique_together = ("data_type", "checksum", "relative_path")
+        unique_together = ("_pulp_domain", "data_type", "checksum", "relative_path")
 
     @property
     def unsupported_metadata_type(self):

pulp_rpm/app/models/distribution.py

@@ -12,6 +12,7 @@
     ContentArtifact,
     Repository,
 )
+from pulpcore.plugin.util import get_domain_pk
 
 log = getLogger(__name__)
 
@@ -94,6 +95,8 @@ class DistributionTree(Content):
 
     digest = models.TextField(null=False)
 
+    _pulp_domain = models.ForeignKey("core.Domain", default=get_domain_pk, on_delete=models.PROTECT)
+
     def repositories(self):
         """
         Return the subrepos in this DistributionTree.
@@ -134,7 +137,7 @@ def artifacts(self):
 
     class Meta:
         default_related_name = "%(app_label)s_%(model_name)s"
-        unique_together = ("digest",)
+        unique_together = ("_pulp_domain", "digest")
 
 
 class Checksum(BaseModel):

pulp_rpm/app/models/modulemd.py

@@ -3,6 +3,7 @@
 from django.db import models
 
 from pulpcore.plugin.models import Content
+from pulpcore.plugin.util import get_domain_pk
 
 from pulp_rpm.app.models.package import Package
 
@@ -65,9 +66,11 @@ class Modulemd(Content):
 
     snippet = models.TextField()
 
+    _pulp_domain = models.ForeignKey("core.Domain", default=get_domain_pk, on_delete=models.PROTECT)
+
     class Meta:
         default_related_name = "%(app_label)s_%(model_name)s"
-        unique_together = ("name", "stream", "version", "context", "arch")
+        unique_together = ("_pulp_domain", "name", "stream", "version", "context", "arch")
 
 
 class ModulemdDefaults(Content):
@@ -92,21 +95,17 @@ class ModulemdDefaults(Content):
     module = models.TextField()
     stream = models.TextField()
     profiles = models.JSONField(default=list)
-    digest = models.TextField(unique=True)
+    digest = models.TextField(db_index=True)
 
     snippet = models.TextField()
 
     repo_key_fields = ("module",)
 
-    @classmethod
-    def natural_key_fields(cls):
-        """
-        Digest is used as a natural key for ModulemdDefaults.
-        """
-        return ("digest",)
+    _pulp_domain = models.ForeignKey("core.Domain", default=get_domain_pk, on_delete=models.PROTECT)
 
     class Meta:
         default_related_name = "%(app_label)s_%(model_name)s"
+        unique_together = ("_pulp_domain", "digest")
 
 
 class ModulemdObsolete(Content):
@@ -151,6 +150,8 @@ class ModulemdObsolete(Content):
 
     snippet = models.TextField()
 
+    _pulp_domain = models.ForeignKey("core.Domain", default=get_domain_pk, on_delete=models.PROTECT)
+
     class Meta:
         default_related_name = "%(app_label)s_%(model_name)s"
-        unique_together = ("modified", "module_name", "module_stream")
+        unique_together = ("_pulp_domain", "modified", "module_name", "module_stream")