feat(nvim): add python snippets

modules/programs/nvim/plugins/lua/nvim-cmp.lua

@@ -10,6 +10,9 @@ cmp.setup({
       ellipsis_char = "...", -- when popup menu exceed maxwidth, the truncated part would show ellipsis_char instead
     }),
   },
+  enabled = function()
+    return not luasnip.jumpable(1)
+  end,
   snippet = {
     -- REQUIRED - you must specify a snippet engine
     expand = function(args)
@@ -23,19 +26,19 @@ cmp.setup({
     ["<S-CR>"] = cmp.mapping.abort(),
     ["<CR>"] = cmp.mapping.confirm({ select = true }),
     ["<Tab>"] = cmp.mapping(function(fallback)
-      if luasnip.jumpable(1) then
+      if cmp.visible() then
-        luasnip.jump(1)
-      elseif cmp.visible() then
         cmp.select_next_item({ behavior = cmp.SelectBehavior.Select })
+      elseif luasnip.jumpable(1) then
+        luasnip.jump(1)
       else
         fallback()
       end
     end, { "i", "s" }),
     ["<S-Tab>"] = cmp.mapping(function(fallback)
-      if luasnip.jumpable(-1) then
+      if cmp.visible() then
-        luasnip.jump(-1)
-      elseif cmp.visible() then
         cmp.select_prev_item({ behavior = cmp.SelectBehavior.Select })
+      elseif luasnip.jumpable(-1) then
+        luasnip.jump(-1)
       else
         fallback()
       end
@@ -43,8 +46,6 @@ cmp.setup({
     ["<C-n>"] = cmp.mapping(function(fallback)
       if luasnip.choice_active() then
         luasnip.change_choice(1)
-      elseif cmp.visible() then
-        cmp.select_next_item({ behavior = cmp.SelectBehavior.Select })
       else
         fallback()
       end
@@ -52,8 +53,6 @@ cmp.setup({
     ["<C-p>"] = cmp.mapping(function(fallback)
       if luasnip.choice_active() then
         luasnip.change_choice(-1)
-      elseif cmp.visible() then
-        cmp.select_prev_item({ behavior = cmp.SelectBehavior.Select })
       else
         fallback()
       end

modules/programs/nvim/plugins/snippets/python.lua

@@ -183,6 +183,51 @@ local dot_try = postfix(".try", {
   end),
 })
 
+local parr = s(
+  "parr",
+  fmt(
+    [[
+    :param {name}: {description}
+    :type {name}: {type}
+    ]],
+    {
+      name = i(1, "name"),
+      description = i(2, "description"),
+      type = i(3, "type"),
+    },
+    {
+      repeat_duplicates = true,
+    }
+  )
+)
+
+local retr = s(
+  "retr",
+  fmt(
+    [[
+    :return: {description}
+    :rtype: {rtype}
+    ]],
+    {
+      description = i(1, "description"),
+      rtype = i(2, "rtype"),
+    }
+  )
+)
+
+local raisr = s(
+  "raisr",
+  fmt(
+    [[
+    :raises {exception}: {description}
+    ]],
+    {
+      exception = i(1, "Exception"),
+      description = i(2, "description"),
+    }
+  )
+)
+
 return {
   def,
   defs,
@@ -191,4 +236,7 @@ return {
   dot_items,
   dot_try,
   enum,
+  parr,
+  retr,
+  raisr,
 }