python class method that receives an object of its class

Question

I want to create a Square class that has method to calculate the distance from another Square. Here is how I have defined it:

class Square:
    def __init__(self, _x: int, _y: int):
        self.x = _x
        self.y = _y

    def distance(self, _other_square: Square) -> int:
        pass

The _other_square is an object of type Square. This gives me an Unresolved reference 'Square' error.

Is there a way to get around it?

@TomKarzes This seems to solve the issue: from __future__ import annotations Thanks — thanos.a, Jun 23 '19 at 10:06

score 2 · Accepted Answer · answered Jun 23 '19 at 10:04

2

Change the function definition to this:

def distance(self, _other_square: 'Square') -> int:
        pass

The type hint is now a str instance which will be resolved after the module has been loaded, and therefore the Square type is defined. See here for all details.

answered Jun 23 '19 at 10:04

handras

1,548
14
28

This clearly solves my issue. Thanks – thanos.a Jun 23 '19 at 10:09

python class method that receives an object of its class

1 Answers1