PyCharm template for python class init function

Question

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.

Community · Accepted Answer · 2017-05-23 12:16:35Z

6

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

edited May 23, 2017 at 12:16

CommunityBot

11 silver badge

answered Oct 18, 2016 at 17:20

alecxe

476k127 gold badges1.1k silver badges1.2k bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Marcel Wilson · Accepted Answer · 2018-12-06 19:00:54Z

5

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

answered Dec 6, 2018 at 19:00

Marcel Wilson

4,6284 gold badges35 silver badges67 bronze badges

2 Comments

Ramon Royo Over a year ago

Very handy thank you. In version 2020.2 the "Fix all" option is gone. Anyone knows how to add it back?

Marcel Wilson Over a year ago

It looks like they may have removed it. Not sure why. I hope this isn't intentional.

Daraan · Accepted Answer · 2023-12-10 03:53:17Z

1

A bit later but i think i find a solution, not sure if there is a short way but my solution is to put the args in the init like $args$ or something like that, and below the init definition where the instance variables must be put something like $argsSplit$ and in the expression of that variable put the following code:

groovyScript("def result = ''; _1.split(', ').each{ item -> result = result + 'self.' + item + ' = ' + item + System.lineSeparator() + '        '}; return result;", args)

screenshot

edited Dec 10, 2023 at 3:53

Daraan

5,1667 gold badges25 silver badges51 bronze badges

answered Dec 7, 2023 at 5:33

Camilo Sánchez Jiménez

114 bronze badges

Comments

Ramon Royo · Accepted Answer · 2020-08-19 06:30:37Z

Partially answering my own question:

Very handy thank you. In version 2020.2 the "Fix all" option is gone. Anyone knows how to add it back?

I haven't found a way to add the missing option back, but a workaround that will accomplish the same, adding all missing fields to the class.

The answer comes from this post: https://stackoverflow.com/a/62060345/1067293

Place the mouse cursor over an unused argument and hit Alt+Enter.
Then hit the right arrow and select "Run inspection on..."

On the new floating window select the file scope.

The inspection panel will appear on the bottom.
On the left select the file.
Then on the right, click the "Add field to class" button.

It's a little bit too elaborate, but nonetheless comes in handy when you've more than a few args.

Collectives™ on Stack Overflow

PyCharm template for python class init function

4 Answers 4

Comments

2 Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

2 Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related