yapf and style

JoshuaGrams · Jan 11, 2020 · 81ec3b5 · 81ec3b5
1 parent 8f47e62
commit 81ec3b5
Showing 1 changed file with 10 additions and 11 deletions.
diff --git a/lib/python/qmk/cli/doctor.py b/lib/python/qmk/cli/doctor.py
@@ -10,14 +10,16 @@
 
 from milc import cli
 
-def _udev_rule(vid, pid = None):
+
+def _udev_rule(vid, pid=None):
     """ Helper function that return udev rules
     """
     if pid:
         return 'SUBSYSTEMS=="usb", ATTRS{idVendor}=="%s", ATTRS{idProduct}=="%s", MODE:="0666"' % (vid, pid)
     else:
         return 'SUBSYSTEMS=="usb", ATTRS{idVendor}=="%s", MODE:="0666"' % vid
 
+
 @cli.subcommand('Basic QMK environment checks')
 def doctor(cli):
     """Basic QMK environment checks.
@@ -57,17 +59,14 @@ def doctor(cli):
         # Checking for udev rules
         udev_dir = "/etc/udev/rules.d/"
         # These are the recommended udev rules
-        desired_rules = dict(dfu = {_udev_rule("03eb", "2ff4"),_udev_rule("03eb", "2ffb"), _udev_rule("03eb", "2ff0")},
-
-                             tmk = {_udev_rule("feed")},
-
-                             input_club = {_udev_rule("1c11")},
-
-                             stm32 = {_udev_rule("1eaf", "0003"),_udev_rule("0483", "df11")},
+        desired_rules = {
+            'dfu': {_udev_rule("03eb", "2ff4"), _udev_rule("03eb", "2ffb"), _udev_rule("03eb", "2ff0")},
+            'tmk': {_udev_rule("feed")},
+            'input_club': {_udev_rule("1c11")},
+            'stm32': {_udev_rule("1eaf", "0003"), _udev_rule("0483", "df11")},
+            'caterina': {'ATTRS{idVendor}=="2a03", ENV{ID_MM_DEVICE_IGNORE}="1"', 'ATTRS{idVendor}=="2341", ENV{ID_MM_DEVICE_IGNORE}="1"'},
+        }
 
-                             caterina = {'ATTRS{idVendor}=="2a03", ENV{ID_MM_DEVICE_IGNORE}="1"',
-                                         'ATTRS{idVendor}=="2341", ENV{ID_MM_DEVICE_IGNORE}="1"'}
-                             )
         if os.path.exists(udev_dir):
             udev_rules = [rule for rule in glob.iglob(os.path.join(udev_dir, "*.rules")) if os.path.isfile(rule)]
             # Collect all rules from the config files