Use black as formatter for python files

startup_scripts/startup_script_utils/__init__.py

@@ -1,3 +1,3 @@
+from .custom_fields import pop_custom_fields, set_custom_fields_values
 from .load_yaml import load_yaml
 from .permissions import set_permissions
-from .custom_fields import set_custom_fields_values, pop_custom_fields

startup_scripts/startup_script_utils/custom_fields.py

@@ -1,15 +1,16 @@
 def set_custom_fields_values(entity, custom_field_data):
-  if not custom_field_data:
-    return
+    if not custom_field_data:
+        return
+
+    entity.custom_field_data = custom_field_data
+    return entity.save()
 
-  entity.custom_field_data = custom_field_data
-  return entity.save()
 
 def pop_custom_fields(params):
-  if 'custom_field_data' in params:
-    return params.pop('custom_field_data')
-  elif 'custom_fields' in params:
-    print("⚠️ Please rename 'custom_fields' to 'custom_field_data'!")
-    return params.pop('custom_fields')
+    if "custom_field_data" in params:
+        return params.pop("custom_field_data")
+    elif "custom_fields" in params:
+        print("⚠️ Please rename 'custom_fields' to 'custom_field_data'!")
+        return params.pop("custom_fields")
 
-  return None
+    return None

startup_scripts/startup_script_utils/load_yaml.py

@@ -1,11 +1,12 @@
 from pathlib import Path
+
 from ruamel.yaml import YAML
 
 
 def load_yaml(yaml_file: str):
-  yf = Path(yaml_file)
-  if not yf.is_file():
-    return None
-  with yf.open("r") as stream:
-    yaml = YAML(typ="safe")
-    return yaml.load(stream)
+    yf = Path(yaml_file)
+    if not yf.is_file():
+        return None
+    with yf.open("r") as stream:
+        yaml = YAML(typ="safe")
+        return yaml.load(stream)

startup_scripts/startup_script_utils/permissions.py

@@ -2,17 +2,21 @@ from django.contrib.auth.models import Permission
 
 
 def set_permissions(subject, permission_filters):
-  if subject is None or permission_filters is None:
-    return
-  subject.clear()
-  for permission_filter in permission_filters:
-    if "*" in permission_filter:
-      permission_filter_regex = "^" + permission_filter.replace("*", ".*") + "$"
-      permissions = Permission.objects.filter(codename__iregex=permission_filter_regex)
-      print("  ⚿ Granting", permissions.count(), "permissions matching '" + permission_filter + "'")
-    else:
-      permissions = Permission.objects.filter(codename=permission_filter)
-      print("  ⚿ Granting permission", permission_filter)
+    if subject is None or permission_filters is None:
+        return
+    subject.clear()
+    for permission_filter in permission_filters:
+        if "*" in permission_filter:
+            permission_filter_regex = "^" + permission_filter.replace("*", ".*") + "$"
+            permissions = Permission.objects.filter(codename__iregex=permission_filter_regex)
+            print(
+                "  ⚿ Granting",
+                permissions.count(),
+                "permissions matching '" + permission_filter + "'",
+            )
+        else:
+            permissions = Permission.objects.filter(codename=permission_filter)
+            print("  ⚿ Granting permission", permission_filter)
 
-    for permission in permissions:
-      subject.add(permission)
+        for permission in permissions:
+            subject.add(permission)