Support electric-layout-mode in python-base-mode

* lisp/progmodes/python.el (python-base-mode): Add support for
'electric-layout-mode'.

1 files changed, 5 insertions, 0 deletions

diff --git a/lisp/progmodes/python.el b/lisp/progmodes/python.el
index b350b5dadf9..5f85fecf016 100644
--- a/lisp/progmodes/python.el
+++ b/lisp/progmodes/python.el
@@ -7071,6 +7071,11 @@ implementations: `python-mode' and `python-ts-mode'."
   (setq-local electric-indent-inhibit t)
   (setq-local electric-indent-chars
               (cons ?: electric-indent-chars))
+  (setq-local electric-layout-rules
+              `((?: . ,(lambda ()
+                         (and (zerop (car (syntax-ppss)))
+                              (python-info-statement-starts-block-p)
+                              'after)))))
 
   ;; Add """ ... """ pairing to electric-pair-mode.
   (add-hook 'post-self-insert-hook