def get_entity_two(self): from discograph.library.sqlite.SqliteArtist import SqliteArtist from discograph.library.sqlite.SqliteLabel import SqliteLabel entity_id = self.entity_two_id entity_type = self.entity_two_type if entity_type == 1: return SqliteArtist.from_id(entity_id) return SqliteLabel.from_id(entity_id)