-
Notifications
You must be signed in to change notification settings - Fork 7
/
logs_pet_bullshit.py
208 lines (177 loc) · 6.63 KB
/
logs_pet_bullshit.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
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
from collections import defaultdict
from c_bosses import convert_to_fight_name
SPEC_DATA = {
"Unholy": {
"spell_id": "50526",
"pet_name": "Ghoul",
"pet_spells": {
"47468": "Claw",
"47481": "Gnaw",
"47482": "Leap",
},
},
"Affliction": {
"spell_id": "59164",
"pet_name": "Felhunter",
"pet_spells": {
"19647": "Spell Lock",
"19658": "Devour Magic Effect",
"48011": "Devour Magic",
"54053": "Shadow Bite",
"54425": "Improved Felhunter",
"57567": "Fel Intelligence",
},
},
"Frost Mage": {
"spell_id": "44572",
"pet_name": "Water Elemental",
"pet_spells": {
"72898": "Waterbolt",
},
},
}
def sdjfsiodjfsdiojfio(owners1, owners2, pets1, pets2):
owners_diff = owners1 - owners2
print()
print(owners_diff)
if len(owners_diff) == 1:
pets_diff = pets1 - pets2
print(pets_diff)
if len(pets_diff) == 1:
return owners_diff.pop(), pets_diff.pop()
def check_negative(data):
for i, (owners1, pets1) in enumerate(data, 1):
for owners2, pets2 in data[i:]:
if owners1 == owners2 or pets1 == pets2:
continue
q = sdjfsiodjfsdiojfio(owners1, owners2, pets1, pets2)
if q:
yield q
q = sdjfsiodjfsdiojfio(owners2, owners1, pets2, pets1)
if q:
yield q
class PET_BULLSHIT:
def __init__(self, logs: list[str], enc_data: dict[str, list], data: dict, player_spec: str) -> None:
if player_spec not in SPEC_DATA:
return
pet_data = SPEC_DATA[player_spec]
self.spec_spell_id = pet_data["spell_id"]
self.pet_name = pet_data["pet_name"]
self.pet_spell_ids = pet_data["pet_spells"]
self.logs = logs
self.enc_data = enc_data
self.everything: dict[str, dict[str, str]] = data["everything"]
self.missing_owner: list[str] = data["missing_owner"]
self.spec_pets: defaultdict[str, set[str]] = data["other_perma_pets"][self.pet_name]
def combine_pet_by_target(self):
q: defaultdict[str, set[str]] = defaultdict(set)
for pet_guid, targets in self.spec_pets.items():
pet_id = pet_guid[6:-6]
for target_guid in targets:
enc_name = convert_to_fight_name(target_guid)
if enc_name:
q[enc_name].add(pet_id)
return q
@property
def pet_data(self):
try:
return self._pet_data
except AttributeError:
pass
q: dict[str, dict[str, str]] = {}
for pet_guid in self.spec_pets:
pet_id = pet_guid[6:-6]
if pet_id in q:
continue
q[pet_id] = self.everything[pet_guid].get('master_guid')
self._pet_data = q
return self._pet_data
def get_all_classes_in_segment(self, s, f):
spell_id = self.spec_spell_id
spell_id_filter = f",{spell_id},"
players = set()
for line in self.logs[s:f]:
if spell_id_filter not in line:
continue
try:
_, _, sGUID, _, _, _, _spell_id, _ = line.split(',', 7)
if _spell_id == spell_id and sGUID.startswith("0x0"):
players.add(sGUID)
except:
pass
return players
def get_all_pets_in_segment(self, s, f):
pets = set()
for line in self.logs[s:f]:
if "0xF14" not in line:
continue
try:
_, _, sGUID, _, _, _, spell_id, _ = line.split(',', 7)
if self.pet_spell_ids[spell_id] and sGUID.startswith("0xF14"):
pets.add(sGUID)
except:
pass
print("get_all_pets_in_segment", pets)
return pets
def update_pet_owner(self, owner_guid, pet_id):
if self.pet_data.get(pet_id):
return
self.pet_data[pet_id] = owner_guid
PET_DATA = {
'name': self.pet_name,
'master_name': self.everything[owner_guid]['name'],
'master_guid': owner_guid
}
for guid in self.spec_pets:
if guid[6:-6] == pet_id:
self.everything[guid] = PET_DATA
print(f"set pet owner for {guid} {PET_DATA}")
def filter_pets_by_combat(self):
iosadjfiosajfuisdoj = []
for enc_name, pet_ids in self.combine_pet_by_target().items():
if not pet_ids:
continue
boss_segments = self.enc_data.get(enc_name)
if not boss_segments:
continue
for s, f in boss_segments:
# for i, (s, f) in enumerate(boss_segments):
_owners = self.get_all_classes_in_segment(s, f)
_pets = self.get_all_pets_in_segment(s, f)
if self.filter_out_pets(_pets, _owners):
self.update_pet_owner(_owners.pop(), _pets.pop()[6:-6])
if len(_owners) != len(_pets):
continue
_pets = {k[6:-6] for k in _pets}
iosadjfiosajfuisdoj.append((_owners, _pets))
# print(self.spec_pets)
for _o, _p in check_negative(iosadjfiosajfuisdoj):
self.update_pet_owner(_o, _p)
def filter_out_pets(self, pets: set[str], owners: set[str]):
for pet_guid in set(pets):
if pet_guid not in self.everything:
continue
master_guid = self.everything[pet_guid].get('master_guid')
if master_guid:
if pet_guid in pets:
pets.remove(pet_guid)
if master_guid in owners:
owners.remove(master_guid)
return len(pets) == 1 and len(owners) == 1
def filter_loop(self):
for n in range(len(self.pet_data)):
if None not in self.pet_data.values():
break
print('filter_pets_by_combat Loop:', n+1)
print(self.pet_data)
pet_data_cached = dict(self.pet_data)
self.filter_pets_by_combat()
if self.pet_data == pet_data_cached:
break
for pet_id in list(self.missing_owner):
if self.pet_data.get(pet_id):
self.missing_owner.remove(pet_id)
def find_pet_owners(logs: list[str], enc_data: dict[str, list], data: dict):
for spec_name in SPEC_DATA:
z = PET_BULLSHIT(logs, enc_data, data, spec_name)
z.filter_loop()