Как можно изменить/сократить/украсить код?

Question

[helloworld21]

'''
1. Создавать воинов из json (также как оружие)
2. Добавить показатель брони воина (полученный урон = урон оружия - броня)
3. Добавить шанс попадания оружием
4. Для класса Weapon добавить метод, вычисляющий средний урон за удар
5. Добавить свои виды оружия, сбаллансировать их по урону (ср. урон 20 +- 10)
6. Возможность "битвы" 2х2, 4х4 ... NxN. Нужно продумать логику нанесения ударов.
7. На пов. оценку: визуализировать с помощью pygame либо другого игр. движка
'''
import random,os
from names import createWarriorTeam,createWeapons,Warrior,Weapons

warriors = createWarriorTeam(os.path.join(os.getcwd(),'WARRIORS.json'))
weapons = createWeapons(os.path.join(os.getcwd(),'WEAPONS.json'))


class Battlefield():
    def __init__(self,warriors,weapons):
        self.warriors = warriors
        self.weapons = weapons

    def _getweapons(self):
        self.weapons = [Weapons(x,self.weapons[x]) for x in self.weapons]

    def _getcomands(self):
        self.teams = {}
        self.teamscount = {}
        for warr in self.warriors:
            warr = warr.upper()
            self.team = [Warrior(x,warr,self.weapons) for x in self.warriors[warr]]
            self.teams[warr] = self.team
            self.teamscount[warr] = len(self.team)
        print('-' * 100)
        for tc in self.teamscount:
            print('{0}:{1}'.format(tc,self.teamscount[tc]),end=' ')
        print()
        print('-'*100)


    def _fight(self):
        while True:
            liveTeams = 0
            lteams = list(self.teams)
            random.shuffle(lteams)
            for lt in lteams:
                warrior = random.choice(self.teams[lt])
                otherTeams = [x for x in lteams if x!=lt]
                enemyTeam = random.choice(otherTeams)
                enemywarrior = random.choice(self.teams[enemyTeam])
                if enemywarrior.is_alive:
                    warrior.hit(enemywarrior)
            for ltlive in lteams:
                if (([False]*self.teamscount[ltlive])==[x.is_alive for x in self.teams[ltlive]]):
                    liveTeams+=1
            if liveTeams==len(self.teams)-1:
                break
        self._end()
        self._averageDamage()

    def _end(self):
        for x in self.teams:
            if  not all([not x.is_alive for x in self.teams[x]]):
                winner = x
                break
        print('Победила команда {0}!'.format(winner))
        print('-'*100)
        for t in self.teams:
            for w in self.teams[t]:
                print('Имя:{0} Команда:{1} Здоровье:{2} Броня:{3} Оружие:{4}({5})'.format(w.name, w.team, w.health,
                           w.armor, w.weapon.name,w.weapon.damage))

    def _averageDamage(self):
        print('-' * 100)
        warriors =[]
        for t in self.teams:
            warriors+=self.teams[t]
        for x in warriors:
            print('Оружие:{0} Средний урон:{1}({2})'.format(x.weapon.name,
                                    x.weapon.getAverageDamage(),x.weapon.damage))

    def start(self):
        self._getweapons()
        self._getcomands()
        self._fight()

a = Battlefield(warriors,weapons)
a.start()

Comments

[Денис Инешин]

Не очень понятно что именно вы хотите упростить, у игры есть некая сложная логика, как именно упростить?

В остальном выглядит как нормальный читабельный код.

[helloworld21]

Добрый день! Нужно изменить её как бы поменять код но при этом смысл оставить

[ScriptKiddo]

helloworld21, Здесь вам смогут помочь

[Ян]

Как я понял вы хотите то же самое, но не хотите встрять на авторские права?

[helloworld21]

Ян, Да именно так

[helloworld21]

Ян, Можете помочь пж ?

Answer 1

[Dimonchik]

нотацию строк замени

https://shultais.education/blog/python-f-strings

Comments

[helloworld21]

А как ? я просто недавно приступил к изучению и мне это очень сложно

[Dimonchik]

см

[helloworld21]

dimonchik2013, Посмотрел но не понял как тут это применить

[Владимир Куц]

helloworld21,

не понял как тут это применить

такая конструкция

'Оружие:{0} Средний урон:{1}({2})'.format(x.weapon.name,
                                    x.weapon.getAverageDamage(),x.weapon.damage)

преобразуется в такую:

f'Оружие:{x.weapon.name} Средний урон:{x.weapon.getAverageDamage()}({x.weapon.damage})'

Второй вариант выглядит более наглядно

[helloworld21]

Владимир Куц, Я вас очень прошу можете весь код так сделать

[Dimonchik]

helloworld21, я могу, сколько?

[helloworld21]

dimonchik2013, Что сколько

[Dimonchik]

кода переписать

[helloworld21]

dimonchik2013, Сколько возможно некоторую часть уже отправили то что осталось если бы сделали было бы просто супер

[helloworld21]

Заранее вам огромное спасибо!

Answer 2

[Владимир Куц]

Почитать про pep8:
к примеру:
otherTeams,  enemyTeam и т.д. - так переменные называть не рекомендуется

 За. f-strings уже сказали:

print('Имя:{0} Команда:{1} Здоровье:{2} Броня:{3} Оружие:{4}({5})'.format(w.name, w.team, w.health,
                           w.armor, w.weapon.name,w.weapon.damage))

меняется на более удобный вид:

print(
   f'Имя:{w.name} '
   f'Команда:{w.tem} '
   f'Здоровье:{w.health}'
   f'Броня:{w.armor}'
   f'Оружие:{w.weapon.name}({w.weapon.damage})'
)

Эту конструкцию не очень понял:

if ([False]*self.teamscount[ltlive])==[x.is_alive for x in self.teams[ltlive]])

Ее можно заменить на что-то вроде

if not any([x.is_alive for x in self.teams[ltlive]])

?
или у вас массивы слева и справа по размерам могут не совпадать?

В остальном - вроде норм

Comments

[helloworld21]

Пожалуйста если получается можете весь код так сделать.