Run black

Author: AlanCoding

ansible_base/rbac/management/create_types.py

@@ -10,10 +10,7 @@ def get_local_DAB_contenttypes(using: str, ct_model: Type[models.Model], service
     # This should work in migration scenarios, but other code checks for existence of it on manager
     ct_model.objects.clear_cache()
 
-    return {
-        (ct.service, ct.model): ct
-        for ct in ct_model.objects.using(using).filter(service=service)
-    }
+    return {(ct.service, ct.model): ct for ct in ct_model.objects.using(using).filter(service=service)}
 
 
 def create_DAB_contenttypes(
@@ -31,9 +28,7 @@ def create_DAB_contenttypes(
     """
     DABContentType = apps.get_model("dab_rbac", "DABContentType")
 
-    content_types = get_local_DAB_contenttypes(
-        using, DABContentType, service
-    )
+    content_types = get_local_DAB_contenttypes(using, DABContentType, service)
 
     cts = [
         DABContentType(
@@ -49,7 +44,4 @@ def create_DAB_contenttypes(
     DABContentType.objects.using(using).bulk_create(cts)
     if verbosity >= 2:
         for ct in cts:
-            print(
-                "Adding DAB content type "
-                f"'{ct.service}:{ct.app_label} | {ct.model}'"
-            )
+            print("Adding DAB content type " f"'{ct.service}:{ct.app_label} | {ct.model}'")

ansible_base/rbac/models/permission.py

@@ -9,7 +9,14 @@ class DABPermission(models.Model):
 
     name = models.CharField("name", max_length=255, help_text=_("The name of this permission."))
     # TODO: make non-null, make db_constraint
-    content_type = models.ForeignKey(DABContentType, models.CASCADE, null=True, verbose_name="content type", help_text=_("The content type this permission will apply to."), db_constraint=False)
+    content_type = models.ForeignKey(
+        DABContentType,
+        models.CASCADE,
+        null=True,
+        verbose_name="content type",
+        help_text=_("The content type this permission will apply to."),
+        db_constraint=False,
+    )
     # tmp_content_type = models.ForeignKey(ContentType, models.CASCADE, verbose_name="content type", help_text=_("The content type this permission will apply to."))
     codename = models.CharField(
         "codename",

ansible_base/rbac/models/role.py

@@ -317,7 +317,9 @@ class Meta:
 
     # role_definition set on child models to set appropriate help_text and related_name
     content_type = models.ForeignKey(
-        DABContentType, on_delete=models.CASCADE, help_text=_("The content type of the subject of permission assignments. Duplicated from related RoleDefinition.")
+        DABContentType,
+        on_delete=models.CASCADE,
+        help_text=_("The content type of the subject of permission assignments. Duplicated from related RoleDefinition."),
     )
     object_id = models.TextField(null=False, help_text=_("The database primary key of the subject of permission assignments."))
     content_object = FederatedForeignKey('content_type', 'object_id')

test_app/tests/rbac/models/test_content_type.py

@@ -17,6 +17,7 @@ def test_post_migrate_creates_contenttype():
 @pytest.mark.django_db
 class DABContentTypeTests(TestCase):
     """These tests originally came from Django contenttypes"""
+
     def setUp(self):
         DABContentType.objects.clear_cache()
         self.addCleanup(DABContentType.objects.clear_cache)