monster.py

'''
Core class for the combat system, used to instantiate a Monster object
with ability to attack other Monster objects, contains lots of crappy code
to show the monsters
'''

class Monster:
    # Action is printed
    def __init__(self, name, hp, dmg, action='hit', ):
        self.name = name
        self.max_hp = hp
        self.hp = hp
        self.dmg = dmg
        self.action = action
        self.alive = True
        self.length = 0

    def line(self, shape):
        if shape == 'top':
            return '┌' + '-' * self.length + '┐'
        elif shape == 'mid':
            return '---' * self.length
        elif shape == 'bot':
            return '└' + '-' * self.length + '┘'
        elif shape == 'clear':
            return '\n' * 120

    def attack(self, enemy):
        enemy.hp -= self.dmg
        enemy.update_data()

    # Returns a multi-line string of the stats
    def show(self, size='min'):
        total = ''
        self.update_data()
        if size == 'max':
            if self.alive:
                n = '|' + self.name
                total += self.line('top') + '\n'
                total += n + ((self.length - len(n)) * ' ') + ' |\n'
                n = '|'+'HP:'+ str(1 * self.hp)
                total += n + ((self.length - len(n)) * ' ') + ' |\n'
                n = '|'+'DMG:'+ str(self.dmg)
                total += n + ((self.length - len(n)) * ' ') + ' |'
                total += '\n' + self.line('bot')
                return total
            else:
                n = '|' + self.name
                total += self.line('top') + '\n'
                total += n + ((self.length - len(n)) * ' ') + ' |'
                total += '\n' + self.line('bot')
                return total
        elif size == 'min':
            if self.alive:
                n = '|' + self.name + ' |HP: ' + '█' * self.hp + (self.max_hp - self.hp) * '░' + ' ' + str(self.hp) + ' |DMG: ' + str(self.dmg)
                self.length = len(n)
                total += self.line('top') + '\n'
                total += n +((self.length - len(n)) * ' ') + ' |'
                total += '\n' + self.line('bot')
                return total
            else:
                n = '|' + self.name
                self.length = len(n)
                total += self.line('top') + '\n'
                total += n + ((self.length - len(n)) * ' ') + ' |'
                total += '\n' + self.line('bot')
                return total

    # Sets life state and changes name accordingly
    def update_data(self):
        if self.hp <= 0:
            self.alive = False

        if not self.alive:
            if not self.name.startswith('Dead'):
                # Checks if enemy got variated name and replaces it for 'Dead'
                if len(self.name.split(' ')) < 2:
                    self.name = 'Dead %s' % self.name
                else:
                    no_variation = self.name.split(' ', 1)
                    self.name = no_variation[1]
                    self.name = '{} {}'.format('Dead', self.name)              
        if self.max_hp + 4 > len(self.name) + 1:
            self.length = self.max_hp + 4
        else:
            self.length = len(self.name) + 1