Как переписать класс?

Question

[Владислав]

class Vehicle(API):
	def vehicle_stats(self, lang='ru', fields='', account_id=''):
		endpoint = '/tanks/stats/'
		if type(fields) is list:
			fields = self._format_fields(fields)	
		return self._api_call(endpoint=endpoint, fields=fields, language=lang, account_id=account_id)
 
	def vehicle_achievements(self, lang='ru', fields='', account_id=''):
		endpoint = '/tanks/achievements/'
		if type(fields) is list:
			fields = self._format_fields(fields)	
		return self._api_call(endpoint=endpoint, fields=fields, language=lang, account_id=account_id)

попался такой код в одном из проектов на github.
Это собственно весь модуль. Данный класс дает доступ к статистике танка игрока в wot.
Очевидно же, что код избыточен и можно его уменьшить.
Как переписать класс, чтобы один и тот же метод вызвать с разным параметром endpoint?

Answer 1

[Tark]

Если нужно именно убрать избыточность и при условии, что код будет вызываться только вашим приложением, то можно его изменить таким образом.

class Vehicle(API):
    def get(self, endpoint='/tanks/stats/', lang='ru', fields='', account_id=''):
        if type(fields) is list:
            fields = self._format_fields(fields)	
        return self._api_call(endpoint=endpoint, 
                              fields=fields, 
                              language=lang, 
                              account_id=account_id)

vehicle = Vehicle(...)

stats = vehicle.get()
achievements = vehicle.get('/tanks/achievements/')

Если же этот код вызывается кем-то ещё, то сами понимаете...

Comments

[Владислав]

я уже сел сам переписывать) прям мои мысли изложили.
Но чувствую, что можно вообще без класса обойтись, раз там всего лишь одна функция

[Tark]

Владислав, скорее всего нет: там ведь ещё рассматривается тип поля fields и проворачивается операция, специфичная для класса API. При очень большом желании можно переписать и правда использовать только функцию, да, но экономии ни места, ни времени не будет.

def get_endpoint(endpoint='/tanks/stats/', lang='ru', fields='', account_id=''):
    api = API()
    if type(fields) is list:
        fields = api._format_fields(fields)
    return api._api_call(endpoint=endpoint, 
                                  fields=fields, 
                                  language=lang, 
                                  account_id=account_id)

Если методы _format_fields и _api_call обёрнуты декоратором @classmethod, можно не создавать экземпляр класса API, а прямо так и вызывать: API._format_fields ... Но это уже довольно странное решение) Всё-таки библиотеки и модули предназначены как раз для того, чтобы вам было нужно меньше писать, а тут вы отказываетесь от этого и пишете всё сами. Хотя оригинальный класс Vehicle странноват, да, принцип DRY напрочь не соблюдён.

[Владислав]

Tark: могу дать ссыль на проект, чтоб вам стало понятнее че за API и какое там дерево проекта. Там везде индусский код слошь и рядом.

Answer 2

[SlivTime]

Я бы сделал так. Переменную endpoint, конечно, можно не инициализировать, а сразу передавать строку в вызов _call_with_endpoint. Но так нагляднее.

class Vehicle(API):
    def _call_with_endpoint(self, lang, fields, account_id, endpoint):
        if type(fields) is list:
          fields = self._format_fields(fields)  
        return self._api_call(endpoint=endpoint,
                              fields=fields,
                              language=lang,
                              account_id=account_id)

    def vehicle_stats(self, lang='ru', fields='', account_id=''):
        endpoint = '/tanks/stats/'
        return self._call_with_endpoint(lang=lang,
                                        field=fields,
                                        account_id=account_id,
                                        endpoint=endpoint)
        

    def vehicle_achievements(self, lang='ru', fields='', account_id=''):
        endpoint = '/tanks/achievements/'
        return self._call_with_endpoint(lang=lang,
                                        field=fields,
                                        account_id=account_id,
                                        endpoint=endpoint)

Answer 3

[Mintormo]

class Vehicle(API):
    def vehicle_stats(self, lang='ru', fields='', account_id='', endpoint = '/tanks/stats/'):  
        if type(fields) is list:
            fields = self._format_fields(fields)	
        return self._api_call(endpoint=endpoint, fields=fields, language=lang, account_id=account_id)

    def vehicle_achievements(self, lang='ru', fields='', account_id='', endpoint = '/tanks/achievements/'):    
        if type(fields) is list:
            fields = self._format_fields(fields)	
        return self._api_call(endpoint=endpoint, fields=fields, language=lang, account_id=account_id

Как-то так. Если надо оставить все по-умолчанию, то просто не пишите последний параметр. Если надо изменить endpoint, то пишите в конце endpoint = "то_что_надо".