Some jinja and python support

Author: AndrewRadev

ftplugin/htmldjango/switch.vim

@@ -0,0 +1,8 @@
+if !exists("g:loaded_switch")
+  finish
+endif
+
+let b:switch_definitions =
+      \ [
+      \   g:switch_builtins.jinja_tag_type,
+      \ ]

ftplugin/python/switch.vim

@@ -0,0 +1,9 @@
+if !exists("g:loaded_switch")
+  finish
+endif
+
+let b:switch_definitions =
+      \ [
+      \   g:switch_builtins.python_dict_get,
+      \   g:switch_builtins.python_string_style,
+      \ ]

plugin/switch.vim

@@ -226,7 +226,21 @@ let g:switch_builtins =
       \   'markdown_task_item': {
       \     '^\(\s*\)- \[ \] \(.*\)':  '\1- [x] \2',
       \     '^\(\s*\)- \[x\] \(.*\)':  '\1- [ ] \2',
-      \   }
+      \   },
+      \   'jinja_tag_type': {
+      \     '{{ \(.*\) }}': '{% \1 %}',
+      \     '{% \(.*\) %}': '{# \1 #}',
+      \     '{# \(.*\) #}': '{{ \1 }}',
+      \   },
+      \   'python_dict_get': {
+      \     '\v(%(\%@<!\k)+)\[(.{-})\]': '\1.get(\2)',
+      \     '\v(\k+)\.get\((.{-})\)':    '\1[\2]',
+      \   },
+      \   'python_string_style': {
+      \     'f"\(\%([^"\\]\|\\.\)*\)"':    '''\1''',
+      \     '''\(\%([^''\\]\|\\.\)*\)''':  '"\1"',
+      \     'f\@!"\(\%([^"\\]\|\\.\)*\)"': 'f"\1"',
+      \   },
       \ }
 
 let g:switch_definitions =

spec/plugin/python_spec.rb

@@ -0,0 +1,28 @@
+require 'spec_helper'
+
+describe "python definitions" do
+  let(:filename) { 'test.py' }
+
+  specify "string type" do
+    set_file_contents 'foo = "bar?"'
+
+    vim.search('bar').switch
+    assert_file_contents 'foo = f"bar?"'
+
+    vim.switch
+    assert_file_contents "foo = 'bar?'"
+
+    vim.switch
+    assert_file_contents 'foo = "bar?"'
+  end
+
+  specify "fetch array access" do
+    set_file_contents "foo['one']"
+
+    vim.search('foo').switch
+    assert_file_contents "foo.get('one')"
+
+    vim.switch
+    assert_file_contents "foo['one']"
+  end
+end