/
Attack.py
169 lines (136 loc) · 6.71 KB
/
Attack.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
from Move import Move
from Pokemon import Pokemon
import random
# Stat modification function; will be called inside the attack function if the move alters the defending Pokemon's stats
# Takes the current statStage as input and returns a multiplier that will be used to calculate the new statStage
def statMod(statStage):
if statStage == 1:
multiplier = 1.5
elif statStage == -1:
multiplier = 2/3
elif statStage == 2:
multiplier = 2
elif statStage == -2:
multiplier = 1/2
elif statStage == 3:
multiplier = 2.5
elif statStage == -3:
multiplier = 0.4
elif statStage == 4:
multiplier = 3
elif statStage == -4:
multiplier = 1/3
elif statStage == 5:
multiplier = 3.5
elif statStage == -5:
multiplier = 2/7
elif statStage == 6:
multiplier = 4
elif statStage == -6:
multiplier = 1/4
return multiplier # This multiplier affects the value of the in-battle stat
# Will take a move, the attacking Pokemon object, and the defending Pokemon object as input
# Will return a string that contains the amount of damage done and the effectiveness of the move
def attack(move, pokemon1, pokemon2):
# Creating an empty string to store the results of the attack function
tempMsg= ""
# Reading "Type Advantages.csv" file to determine type advantages and the damage modifier
# Stores the line number in the csv as the key and a list giving information about type advantage for the value
fin = open("Type Advantages.csv", 'r')
typeDic = {}
for line in fin:
line = line.strip()
typeList = line.split(",")
typeDic[typeList[0]] = typeList
# This list contains a number in the first position, the attack type in the second, the defending type in the third,
# and the appropriate damage multiplier in the fourth
fin.close()
# Making the input string into an actual move object
move = Move(move)
# This modifier is used in damage calculations; it takes into account type advantage and STAB bonus
modifier = 1
# Calculating Type advantages using "Type Advantages.csv" file
for key in typeDic:
# If the attacking and defending types match up, multiply the modifier by the damage multiplier from the list
if typeDic[key][1] == move.type and typeDic[key][2] == pokemon2.type1:
modifier *= float(typeDic[key][3])
# Didn't use elif; Just in case you get a 4x or 0.25x modifier based on double type
if typeDic[key][1] == move.type and typeDic[key][2] == pokemon2.type2:
modifier *= float(typeDic[key][3])
# Calculating STAB (Same-type Attack Bonus)
if move.type == pokemon1.type1:
modifier *= Pokemon.STAB
elif move.type == pokemon1.type2:
modifier *= Pokemon.STAB
# Damage formula also has a random element
modifier *= random.uniform(0.85, 1.0)
print()
# Appending the useMove function to the output
tempMsg += pokemon1.useMove(move)
# ATK/DEF or SpATK/SpDEF or Status? Using the Pokemon damage formula
# If the move is "Physical", the damage formula will take into account attack and defense
if move.kind == "Physical":
damage = int((((2*pokemon1.getLevel()) + 10)/250 * (pokemon1.battleATK/pokemon2.battleDEF) * move.getPower() + 2) * modifier)
tempMsg += "\n" + pokemon2.loseHP(damage)
# If the move is "Special", the damage formula will take into account special attack and special defense
elif move.kind == "Special":
damage = int((((2*pokemon1.getLevel()) + 10)/250 * (pokemon1.battleSpATK/pokemon2.battleSpDEF) * move.getPower() + 2) * modifier)
tempMsg += "\n" + pokemon2.loseHP(damage)
# Stat Changing moves
else:
# If the move is stat-changing, it does 0 damage and the modifier is set to 1 (so it doesn't return super effective or not very effective)
damage = 0
modifier = 1
# Going through each kind of different stat change based on the move type
if move.kind == "a-":
pokemon2.atkStage -= 1
pokemon2.battleATK = pokemon2.originalATK * statMod(pokemon2.atkStage)
tempMsg += "\n" + pokemon2.name + "'s attack fell! "
elif move.kind == "a+":
pokemon1.atkStage +=1
pokemon1.battleATK = pokemon1.originalATK * statMod(pokemon1.atkStage)
tempMsg += "\n" + pokemon1.name + "'s attack rose! "
elif move.kind == "d+":
pokemon1.defStage +=1
pokemon1.battleDEF = pokemon1.originalDEF * statMod(pokemon1.defStage)
print(pokemon1.name + "'s defense rose! ")
tempMsg += "\n" + pokemon1.name + "'s defense rose! "
elif move.kind == "sa+":
pokemon1.spAtkStage +=1
pokemon1.battleSpATK = pokemon1.originalSpATK * statMod(pokemon1.spAtkStage)
print(pokemon1.name + "'s special attack rose! ")
tempMsg += "\n" + pokemon1.name + "'s special attack rose! "
elif move.kind == "sd+":
pokemon1.spDefStage +=1
pokemon1.battleSpDef = pokemon1.originalSpDEF * statMod(pokemon1.spDefStage)
tempMsg += "\n" + pokemon1.name + "'s special defense rose! "
elif move.kind == "s+":
pokemon1.speedStage +=1
pokemon1.battleSpeed = pokemon1.originalSpeed * statMod(pokemon1.speedStage)
tempMsg += "\n" + pokemon1.name + "'s speed fell! "
elif move.kind == "d-":
pokemon2.defStage -=1
pokemon2.battleDEF = pokemon2.originalDEF * statMod(pokemon2.defStage)
tempMsg += "\n" + pokemon2.name + "'s defense fell! "
elif move.kind == "sa-":
pokemon2.spAtkStage -=1
pokemon2.battleSpATK = pokemon2.originalSpATK * statMod(pokemon2.spAtkStage)
tempMsg += "\n" + pokemon2.name + "'s special attack fell! "
elif move.kind == "sd-":
pokemon2.spDefStage -=1
pokemon2.battleSpDEF = pokemon2.originalSpDEF * statMod(pokemon2.spDefStage)
tempMsg += "\n" + pokemon2.name + "'s special defense fell! "
elif move.kind == "s-":
pokemon2.speedStage -=1
pokemon2.battleSpeed = pokemon2.originalSpeed * statMod(pokemon2.speedStage)
tempMsg += "\n" + pokemon2.name + "'s speed fell! "
# Super effective, not very effective, or no effect?
# Appending the result to tempMsg
if modifier < 0.85 and modifier > 0:
tempMsg += "\nIt's not very effective..."
elif modifier > 1.5:
tempMsg += "\nIt's super effective!"
elif modifier == 0.0:
tempMsg += "\nIt doesn't affect " + pokemon2.name + "..."
# String containing useMove(), damage, and type effectiveness
return tempMsg