1 files changed, 12 insertions, 18 deletions

diff --git a/lisp/progmodes/python.el b/lisp/progmodes/python.el
index f73a0128b5c..f34fb28826f 100644
--- a/lisp/progmodes/python.el
+++ b/lisp/progmodes/python.el
@@ -7355,24 +7355,15 @@ implementations: `python-mode' and `python-ts-mode'."
                       #'python-eldoc-function))))
   (eldoc-add-command-completions "python-indent-dedent-line-backspace")
 
-  (add-to-list
-   'hs-modes-alist
-   `(python-mode
-     (start . ,python-nav-beginning-of-block-regexp)
-     ;; Use the empty string as end regexp so it doesn't default to
-     ;; "\\s)".  This way parens at end of defun are properly hidden.
-     (end . "")
-     (c-start . "#")
-     (forward-fn . python-hideshow-forward-sexp-function)
-     (find-beg-fn . python-nav-beginning-of-block)
-     (find-next-fn . python-hideshow-find-next-block)
-     (look-start-fn . python-info-looking-at-beginning-of-block)))
-
-  (add-to-list
-   'hs-modes-alist
-   '(python-ts-mode
-     (treesit-things . (or defun sexp))
-     (adjust-end-fn . python-ts-hs-adjust-block-end-fn)))
+  (setq-local hs-block-start-regexp python-nav-beginning-of-block-regexp)
+  ;; Use the empty string as end regexp so it doesn't default to
+  ;; "\\s)".  This way parens at end of defun are properly hidden.
+  (setq-local hs-block-end-regexp "")
+  (setq-local hs-c-start-regexp "#")
+  (setq-local hs-forward-sexp-func #'python-hideshow-forward-sexp-function)
+  (setq-local hs-find-block-beginning-func #'python-nav-beginning-of-block)
+  (setq-local hs-find-next-block-func #'python-hideshow-find-next-block)
+  (setq-local hs-looking-at-block-start-p-func #'python-info-looking-at-beginning-of-block)
 
   (setq-local outline-regexp (python-rx (* space) block-start))
   (setq-local outline-level
@@ -7446,6 +7437,9 @@ implementations: `python-mode' and `python-ts-mode'."
     (setq-local forward-sexp-function #'treesit-forward-sexp
                 treesit-sexp-thing 'sexp)
 
+    (setq-local hs-treesit-things '(or defun sexp))
+    (setq-local hs-adjust-block-end #'python-ts-hs-adjust-block-end-fn)
+
     (setq-local syntax-propertize-function #'python--treesit-syntax-propertize)
 
     (python-skeleton-add-menu-items)