Typing fix: More direct access to the Family enum

hypothesis · Mar 14, 2024 · 48937ae · 48937ae
1 parent ae053bc
commit 48937ae
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/lms/product/blackboard/product.py b/lms/product/blackboard/product.py
@@ -3,14 +3,14 @@
 from lms.product.blackboard._plugin.course_copy import BlackboardCourseCopyPlugin
 from lms.product.blackboard._plugin.grouping import BlackboardGroupingPlugin
 from lms.product.blackboard._plugin.misc import BlackboardMiscPlugin
-from lms.product.product import PluginConfig, Product, Routes
+from lms.product.product import Family, PluginConfig, Product, Routes
 
 
 @dataclass
 class Blackboard(Product):
     """A product for Blackboard specific settings and tweaks."""
 
-    family: Product.Family = Product.Family.BLACKBOARD
+    family: Family = Family.BLACKBOARD
 
     route: Routes = Routes(
         oauth2_authorize="blackboard_api.oauth.authorize",

diff --git a/lms/product/canvas/product.py b/lms/product/canvas/product.py
@@ -3,14 +3,14 @@
 from lms.product.canvas._plugin.course_copy import CanvasCourseCopyPlugin
 from lms.product.canvas._plugin.grouping import CanvasGroupingPlugin
 from lms.product.canvas._plugin.misc import CanvasMiscPlugin
-from lms.product.product import PluginConfig, Product, Routes
+from lms.product.product import Family, PluginConfig, Product, Routes
 
 
 @dataclass
 class Canvas(Product):
     """A product for Canvas specific settings and tweaks."""
 
-    family: Product.Family = Product.Family.CANVAS
+    family: Family = Family.CANVAS
 
     route: Routes = Routes(
         oauth2_authorize="canvas_api.oauth.authorize",

diff --git a/lms/product/d2l/product.py b/lms/product/d2l/product.py
@@ -3,14 +3,14 @@
 from lms.product.d2l._plugin.course_copy import D2LCourseCopyPlugin
 from lms.product.d2l._plugin.grouping import D2LGroupingPlugin
 from lms.product.d2l._plugin.misc import D2LMiscPlugin
-from lms.product.product import PluginConfig, Product, Routes
+from lms.product.product import Family, PluginConfig, Product, Routes
 
 
 @dataclass
 class D2L(Product):
     """A product for D2L specific settings and tweaks."""
 
-    family: Product.Family = Product.Family.D2L
+    family: Family = Family.D2L
 
     plugin_config: PluginConfig = PluginConfig(
         grouping=D2LGroupingPlugin, misc=D2LMiscPlugin, course_copy=D2LCourseCopyPlugin

diff --git a/lms/product/moodle/product.py b/lms/product/moodle/product.py
@@ -3,12 +3,12 @@
 from lms.product.moodle._plugin.course_copy import MoodleCourseCopyPlugin
 from lms.product.moodle._plugin.grouping import MoodleGroupingPlugin
 from lms.product.moodle._plugin.misc import MoodleMiscPlugin
-from lms.product.product import PluginConfig, Product, Routes
+from lms.product.product import Family, PluginConfig, Product, Routes
 
 
 @dataclass
 class Moodle(Product):
-    family: Product.Family = Product.Family.MOODLE
+    family: Family = Family.MOODLE
 
     plugin_config: PluginConfig = PluginConfig(
         grouping=MoodleGroupingPlugin,