Python dictionaries turned into dict(...)

Author: AndrewRadev

doc/switch.txt

@@ -800,6 +800,11 @@ String style (g:switch_builtins.python_string_style)
     "Ham"
     f"Ham"
 <
+Dictionary style (g:switch_builtins.python_dict_style)
+>
+    {'foo': 'bar', 'bar': 'baz'}
+    dict(foo='bar', bar='baz')
+<
 Jinja ~
 
 Tag type (g:switch_builtins.jinja_tag_type)

ftplugin/python/switch.vim

@@ -6,4 +6,5 @@ let b:switch_definitions =
       \ [
       \   g:switch_builtins.python_dict_get,
       \   g:switch_builtins.python_string_style,
+      \   g:switch_builtins.python_dict_style,
       \ ]

plugin/switch.vim

@@ -236,6 +236,18 @@ let g:switch_builtins =
       \     '\v(%(\%@<!\k)+)\[(.{-})\]': '\1.get(\2)',
       \     '\v(\k+)\.get\((.{-})\)':    '\1[\2]',
       \   },
+      \   'python_dict_style': {
+      \     '{\(\s*[''"]\k\+[''"]:\s*.\+,\?\)\+}': {
+      \       '{':                      'dict(',
+      \       '[''"]\(\k\+\)[''"]:\s*': '\1=',
+      \       '}':                      ')',
+      \     },
+      \     'dict(\(\s*\k\+=\s*.\+,\?\)\+)': {
+      \       'dict(':     '{',
+      \       '\(\k\+\)=': '''\1'': ',
+      \       ')':         '}',
+      \     },
+      \   },
       \   'python_string_style': {
       \     'f"\(\%([^"\\]\|\\.\)*\)"':    '''\1''',
       \     '''\(\%([^''\\]\|\\.\)*\)''':  '"\1"',

spec/plugin/python_spec.rb

@@ -25,4 +25,14 @@
     vim.switch
     assert_file_contents "foo['one']"
   end
+
+  specify "dict style" do
+    set_file_contents "data = {'foo': 'bar', 'bar': 'baz'}"
+
+    vim.search('{').switch
+    assert_file_contents "data = dict(foo='bar', bar='baz')"
+
+    vim.switch
+    assert_file_contents "data = {'foo': 'bar', 'bar': 'baz'}"
+  end
 end