temp

pulp_2to3_migration/app/migration.py

@@ -58,7 +58,7 @@ async def migrate_repositories(plan):
             pulp2repos_qs = Pulp2Repository.objects.filter(
                 pulp3_repository_version=None,
                 not_in_plan=False,
-                type=plugin.type,
+                pulp2_repo_type=plugin.type,
             )
             repos_to_create = plugin.get_repo_creation_setup()
 
@@ -180,11 +180,12 @@ async def migrate_repo_distributor(pb, dist_migrator, pulp2dist, repo_version=No
     )
     with ProgressReport(**progress_data) as pb:
         for plugin in plan.get_plugin_plans():
+            distributor_types = list(plugin.migrator.distributor_migrators.keys())
             pulp2distributors_qs = Pulp2Distributor.objects.filter(
                 pulp3_distribution=None,
                 pulp3_publication=None,
                 not_in_plan=False,
-                pulp2_type_id=plugin.type
+                pulp2_type_id__in=distributor_types
             )
             pb.total = pulp2distributors_qs.count()
             pb.save()
@@ -213,11 +214,9 @@ async def migrate_repo_distributor(pb, dist_migrator, pulp2dist, repo_version=No
                             # not in Pulp 2 anymore
                             continue
                         else:
-                            distributor_types = plugin.migrator.distributor_migrators.keys()
-
                             pulp2dist = Pulp2Distributor.objects.filter(
                                 pulp2_repo_id__in=dist_repositories,
-                                pulp2_type_id__in=list(distributor_types)
+                                pulp2_type_id__in=distributor_types
                             )
                             for dist in pulp2dist:
                                 dist_migrator = distributor_migrators.get(dist.pulp2_type_id)
@@ -284,7 +283,7 @@ def create_repo_version(migrator, pulp3_repo_name, pulp2_repo, pulp3_remote=None
     for plugin in plan.get_plugin_plans():
         pulp3_repo_setup = plugin.get_repo_creation_setup()
         if not pulp3_repo_setup:
-            repos_to_migrate = Pulp2Repository.objects.filter(type=plugin.type,
+            repos_to_migrate = Pulp2Repository.objects.filter(pulp2_repo_type=plugin.type,
                                                               not_in_plan=False)
             for pulp2_repo in repos_to_migrate:
                 # Create one repo version for each pulp 2 repo if needed.

pulp_2to3_migration/app/migrations/0001_initial.py

@@ -1,4 +1,4 @@
-# Generated by Django 2.2.9 on 2020-01-22 15:53
+# Generated by Django 2.2.10 on 2020-02-07 16:17
 
 import django.contrib.postgres.fields
 import django.contrib.postgres.fields.jsonb
@@ -58,9 +58,9 @@ class Migration(migrations.Migration):
                 ('pulp2_last_unit_removed', models.DateTimeField(null=True)),
                 ('is_migrated', models.BooleanField(default=False)),
                 ('not_in_plan', models.BooleanField(default=False)),
-                ('type', models.CharField(max_length=25)),
+                ('pulp2_repo_type', models.CharField(max_length=25)),
                 ('pulp3_repository_remote', models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, to='core.Remote')),
-                ('pulp3_repository_version', models.OneToOneField(null=True, on_delete=django.db.models.deletion.SET_NULL, to='core.RepositoryVersion')),
+                ('pulp3_repository_version', models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, to='core.RepositoryVersion')),
             ],
             options={
                 'abstract': False,

pulp_2to3_migration/app/models/repository.py

@@ -41,7 +41,7 @@ class Pulp2Repository(BaseModel):
     pulp2_last_unit_removed = models.DateTimeField(null=True)
     is_migrated = models.BooleanField(default=False)
     not_in_plan = models.BooleanField(default=False)
-    type = models.CharField(max_length=25)
+    pulp2_repo_type = models.CharField(max_length=25)
 
     pulp3_repository_version = models.ForeignKey(RepositoryVersion,
                                                  on_delete=models.SET_NULL,

pulp_2to3_migration/app/pre_migration.py

@@ -269,7 +269,7 @@ async def pre_migrate_repo(record, repo_id_to_type):
                   'pulp2_last_unit_added': last_unit_added,
                   'pulp2_last_unit_removed': last_unit_removed,
                   'pulp2_description': record.description,
-                  'type': repo_id_to_type[record.repo_id],
+                  'pulp2_repo_type': repo_id_to_type[record.repo_id],
                   'is_migrated': False})
 
     if not created:
@@ -438,7 +438,7 @@ async def mark_removed_resources(plan, type_to_repo_ids):
         mongo_repo_object_ids = set(
             str(i.id) for i in Repository.objects(mongo_repo_q).only('id'))
 
-        premigrated_repos = Pulp2Repository.objects.filter(type=plugin_plan.type)
+        premigrated_repos = Pulp2Repository.objects.filter(pulp2_repo_type=plugin_plan.type)
         premigrated_repo_object_ids = set(premigrated_repos.values_list('pulp2_object_id',
                                                                         flat=True))
         removed_repo_object_ids = premigrated_repo_object_ids - mongo_repo_object_ids

pulp_2to3_migration/app/serializers.py

@@ -156,7 +156,7 @@ class Pulp2RepositoriesSerializer(ModelSerializer):
     )
     pulp2_object_id = serializers.CharField(max_length=255)
     pulp2_repo_id = serializers.CharField()
-    type = serializers.CharField()
+    pulp2_repo_type = serializers.CharField()
     is_migrated = serializers.BooleanField(default=False)
     not_in_plan = serializers.BooleanField(default=False)
 
@@ -233,7 +233,7 @@ class Meta:
         fields = ModelSerializer.Meta.fields + (
             "pulp2_object_id",
             "pulp2_repo_id",
-            "type",
+            "pulp2_repo_type",
             "is_migrated",
             "not_in_plan",
             "pulp3_repository_version",