Raise proper PostgresPartitioningError when PSQLEXTRA_PARTITIONING_MANAGER is not set

Photonios · Photonios · commit 69470a094933 · 2022-02-07T21:38:05.000+02:00
diff --git a/psqlextra/management/commands/pgpartition.py b/psqlextra/management/commands/pgpartition.py
@@ -107,7 +107,7 @@ def _ask_for_confirmation() -> bool:
     @staticmethod
     def _partitioning_manager():
         partitioning_manager = getattr(
-            settings, "PSQLEXTRA_PARTITIONING_MANAGER"
+            settings, "PSQLEXTRA_PARTITIONING_MANAGER", None
         )
         if not partitioning_manager:
             raise PostgresPartitioningError(

Original file line number	Diff line number	Diff line change
`@@ -107,7 +107,7 @@ def _ask_for_confirmation() -> bool:`
`107`	`107`	`@staticmethod`
`108`	`108`	`def _partitioning_manager():`
`109`	`109`	`partitioning_manager = getattr(`
`110`		`- settings, "PSQLEXTRA_PARTITIONING_MANAGER"`
	`110`	`+ settings, "PSQLEXTRA_PARTITIONING_MANAGER", None`
`111`	`111`	`)`
`112`	`112`	`if not partitioning_manager:`
`113`	`113`	`raise PostgresPartitioningError(`