Merge pull request cstrap#17 from macintacos/fix-class-args

cstrap · web-flow · commit 63e81105a8c0 · 2021-10-13T14:15:04.000+02:00
Fix `class` snippet so it updates `arg` parameter inside `__init__` function on-type
diff --git a/snippets/base.json b/snippets/base.json
@@ -47,7 +47,14 @@
     },
     "New class": {
         "prefix": "class",
-        "body": "class ${1:ClassName}(${2:object}):\n\t\"\"\"${3:docstring for $1.}\"\"\"\n\tdef __init__(self, ${4:arg}):\n\t\t${5:super($1, self).__init__()}\n\t\tself.arg = arg\n\t\t$0",
+        "body": [
+            "class ${1:ClassName}(${2:object}):",
+            "\t\"\"\"${3:docstring for $1.}\"\"\"",
+            "\tdef __init__(self, ${4:arg}):",
+            "\t\t${5:super($1, self).__init__()}",
+            "\t\tself.$4 = $4",
+            "\t\t$0"
+        ],
         "description" : "Code snippet for a class definition."
     },
     "New method": {
