PyCharm template for python class __init__ functio

2019-02-10 16:44发布

问题:

I have a python class with several init variables:

class Foo(object):
    def __init__(self, d1, d2):
        self.d1 = d1
        self.d2 = d2

Is there a way to create this code automatically in PyCharm, so I don't have to type explicitly:

self.dn = dn

This pattern happens very often in my code. Is there a better (Pythonic) way to initialize classes?

I have already seen this post ( What is the best way to do automatic attribute assignment in Python, and is it a good idea?), but I don't want to use a decorator as it makes the code less readable.

回答1:

You can start by making a custom Live Template. I'm not sure about whether you can auto-generate variable number of arguments this way, but, for two constructor arguments the live template may look like:

class $class_name$:
    def __init__(self, $arg1$, $arg2$):
        self.$arg1$ = $arg1$
        self.$arg2$ = $arg2$
$END$

This is how I've set it up in PyCharm:


Or, you can change the way you set the instance attributes and generalize it for N arguments, see:

  • How can you set class attributes from variable arguments (kwargs) in python


回答2:

I learned of another way to solve this.

  • Move cursor to argument
  • Press alt + enter
  • Select Add field '[arg]' to class for single arg -or-
  • Press right arrow key to open sub menu
  • Select Fix all 'unused local' problems to add all args