def test_add_talent_points(self):
amount = 10000
TalentManager(self.server_id, self.char_id).add_talent_points(amount)
doc = MongoTalent.db(self.server_id).find_one({'_id': self.char_id},
{'total': 1})
assert doc['total'] == amount
def add(self, server_id, char_id, message=""):
from core.club import Club
from core.bag import Bag
from core.staff import StaffManger, StaffRecruit
from core.talent import TalentManager
from core.territory import Territory
from core.vip import VIP
from core.arena import Arena
from core.energy import Energy
from core.plunder import Plunder
club_property = self.money_as_text_dict()
if self.club_exp:
club_property['exp'] = self.club_exp
if club_property:
club_property['message'] = message
Club(server_id, char_id).update(**club_property)
if self.vip_exp:
VIP(server_id, char_id).add_exp(self.vip_exp)
if self.bag:
bag = Bag(server_id, char_id)
bag.batch_add(self.bag)
sm = StaffManger(server_id, char_id)
if self.staff:
sm.batch_add(self.staff)
if self.staff_exp_pool:
sm.add_exp_pool(self.staff_exp_pool)
if self.talent_point:
TalentManager(server_id,
char_id).add_talent_points(self.talent_point)
if self.arena_point:
Arena(server_id, char_id).add_point(self.arena_point)
if self.territory_product:
Territory(server_id, char_id).add_product(self.territory_product)
if self.work_card:
Territory(server_id, char_id).add_work_card(self.work_card)
if self.energy:
Energy(server_id, char_id).add(self.energy)
if self.staff_recruit_score:
StaffRecruit(server_id,
char_id).add_score(self.staff_recruit_score)
if self.station_exp:
Plunder(server_id, char_id).add_station_exp(self.station_exp)
if self.resource_data:
_r = _Resource()
_r.resource = dict(self.resource_data)
_r.add(server_id, char_id)
def load_formation_staffs(self):
# NOTE: 这段代码其实是从 Club.force_load_staffs 抄来的
from core.formation import Formation
from core.unit import UnitManager
from core.talent import TalentManager
from core.collection import Collection
from core.party import Party
from core.inspire import Inspire
self.formation_staffs = []
""":type: list[core.staff.Staff]"""
sm = StaffManger(self.server_id, self.char_id)
fm = Formation(self.server_id, self.char_id)
um = UnitManager(self.server_id, self.char_id)
ins = Inspire(self.server_id, self.char_id)
staffs = sm.get_staffs_data()
in_formation_staffs = self.in_formation_staffs()
for k, v in in_formation_staffs.iteritems():
obj = Staff(self.server_id, self.char_id, k, staffs[k])
self.formation_staffs.append(obj)
obj.policy = in_formation_staffs[k]['policy']
obj.formation_position = in_formation_staffs[k]['position']
unit_id = in_formation_staffs[k]['unit_id']
if unit_id:
obj.set_unit(um.get_unit_object(unit_id))
club = Club(self.server_id, self.char_id, load_staffs=False)
club.formation_staffs = self.formation_staffs
working_staff_oids = self.working_staff_oids()
for obj in self.formation_staffs:
obj.check_qianban(working_staff_oids)
obj.add_self_talent_effect(club)
talent_effects_1 = TalentManager(self.server_id,
self.char_id).get_talent_effects()
talent_effects_2 = Collection(self.server_id,
self.char_id).get_talent_effects()
talent_effects_3 = fm.get_talent_effects()
talent_effects_4 = Party(self.server_id,
self.char_id).get_talent_effects()
club.add_talent_effects(talent_effects_1)
club.add_talent_effects(talent_effects_2)
club.add_talent_effects(talent_effects_3)
club.add_talent_effects(talent_effects_4)
config_inspire_level_addition, config_inspire_step_addition = ins.get_addition_config(
)
for obj in self.formation_staffs:
obj.config_inspire_level_addition = config_inspire_level_addition
obj.config_inspire_step_addition = config_inspire_step_addition
obj.calculate()
def test_check_talent_points_not_enough(self):
amount = 10000
try:
TalentManager(self.server_id,
self.char_id).check_talent_points(amount)
except GameException as e:
assert e.error_id == ConfigErrorMessage.get_error_id("BAD_MESSAGE")
else:
Exception('Error')
def reset(request):
server_id = request._game_session.server_id
char_id = request._game_session.char_id
TalentManager(server_id, char_id).reset()
response = TalentResetTalentResponse()
response.ret = 0
return ProtobufResponse(response)
def level_up(request):
server_id = request._game_session.server_id
char_id = request._game_session.char_id
talent_id = request._proto.talent_id
TalentManager(server_id, char_id).level_up(talent_id)
response = TalentLevelUpResponse()
response.ret = 0
return ProtobufResponse(response)
def test_level_up_lock(self):
doc = MongoTalent.db(self.server_id).find_one({'_id': self.char_id},
{'talent': 1})
_id = 0
for k in ConfigTalent.INSTANCES.keys():
if k not in doc['talent']:
_id = k
break
try:
TalentManager(self.server_id, self.char_id).level_up(_id)
except GameException as e:
assert e.error_id == ConfigErrorMessage.get_error_id(
"TALENT_LOCKED")
else:
Exception('error')
def test_unlock(self):
doc = MongoTalent.db(self.server_id).find_one({'_id': self.char_id},
{'talent': 1})
_id = 0
keys = ConfigTalent.INSTANCES.keys()
for k in keys:
if k not in doc['talent']:
_id = k
break
TalentManager(self.server_id, self.char_id).unlock(_id)
doc_af = MongoTalent.db(self.server_id).find_one({'_id': self.char_id},
{'talent': 1})
assert _id in doc_af['talent']
def test_level_up(self):
self.test_add_talent_points()
doc = MongoTalent.db(self.server_id).find_one({'_id': self.char_id},
{'talent': 1})
_id = random.choice(doc['talent'])
conf = ConfigTalent.get(_id)
TalentManager(self.server_id, self.char_id).level_up(_id)
doc_af = MongoTalent.db(self.server_id).find_one({'_id': self.char_id},
{
'talent': 1,
'cost': 1
})
assert _id not in doc_af['talent']
assert conf.next_id in doc_af['talent']
assert doc_af['cost'] == conf.up_need
def test_level_up_level_max(self):
self.test_add_talent_points()
doc = MongoTalent.db(self.server_id).find_one({'_id': self.char_id},
{'talent': 1})
_id = random.choice(doc['talent'])
try:
for i in range(0, 10):
TalentManager(self.server_id, self.char_id).level_up(_id)
conf = ConfigTalent.get(_id)
if conf.next_id:
_id = conf.next_id
else:
break
except GameException as e:
assert e.error_id == ConfigErrorMessage.get_error_id(
"TALENT_LEVEL_REACH_MAX")
else:
Exception("error")
def test_deduct_talent_points(self):
self.test_add_talent_points()
TalentManager(self.server_id, self.char_id).deduct_talent_points(1000)
def test_get_talent_tree(self):
talent_ids = TalentManager(self.server_id,
self.char_id).get_talent_tree()
init_ids = get_init_talent_doc()
for k in init_ids:
assert k in talent_ids
def test_send_notify(self):
TalentManager(self.server_id, self.char_id).send_notify()
def test_get_talent_effect(self):
assert TalentManager(self.server_id, self.char_id).get_talent_effects()
def game_start_handler(server_id, char_id, **kwargs):
MessagePipe(char_id).clean()
msg = UTCNotify()
msg.timestamp = arrow.utcnow().timestamp
MessagePipe(char_id).put(msg=msg)
msg = SocketServerNotify()
ss = random.choice(settings.SOCKET_SERVERS)
msg.ip = ss['host']
msg.port = ss['tcp']
MessagePipe(char_id).put(msg=msg)
_Resource.send_notify(server_id, char_id)
UnitManager(server_id, char_id).send_notify()
Bag(server_id, char_id).send_notify()
StaffManger(server_id, char_id).send_notify()
StaffRecruit(server_id, char_id).send_notify()
f = Formation(server_id, char_id)
f.send_formation_notify()
f.send_slot_notify()
club = Club(server_id, char_id)
club.set_login()
club.send_notify()
msg = CreateDaysNotify()
msg.days = days_passed(club.create_at)
msg.create_at = club.create_at
MessagePipe(char_id).put(msg=msg)
chall = Challenge(server_id, char_id)
chall.send_chapter_notify()
chall.send_challenge_notify()
FriendManager(server_id, char_id).send_notify()
MailManager(server_id, char_id).send_notify()
TaskMain(server_id, char_id).send_notify()
TaskDaily(server_id, char_id).send_notify()
Chat(server_id, char_id).send_notify()
Notification(server_id, char_id).send_notify()
FinanceStatistics(server_id, char_id).send_notify()
TalentManager(server_id, char_id).send_notify()
Dungeon(server_id, char_id).send_notify()
a = Arena(server_id, char_id)
a.send_notify()
a.send_honor_notify()
t = Tower(server_id, char_id)
t.send_notify()
t.send_goods_notify()
Territory(server_id, char_id).send_notify()
TerritoryStore(server_id, char_id).send_notify()
TerritoryFriend(server_id, char_id).send_remained_times_notify()
Store(server_id, char_id).send_notify()
VIP(server_id, char_id).send_notify()
Collection(server_id, char_id).send_notify()
Energy(server_id, char_id).send_notify()
w = Welfare(server_id, char_id)
w.send_signin_notify()
w.send_new_player_notify()
w.send_level_reward_notify()
w.send_energy_reward_notify()
Union(server_id, char_id).send_all_notify()
Purchase(server_id, char_id).send_notify()
ac = ActivityNewPlayer(server_id, char_id)
ac.send_notify()
ac.send_daily_buy_notify()
ActivityOnlineTime(server_id, char_id).send_notify()
ActivityChallenge(server_id, char_id).send_notify()
ActivityPurchaseDaily(server_id, char_id).send_notify()
ActivityPurchaseContinues(server_id, char_id).send_notify()
ActivityLevelGrowing(server_id, char_id).send_notify()
p = Plunder(server_id, char_id)
p.send_search_notify()
p.send_result_notify()
p.send_revenge_notify()
p.send_station_notify()
p.send_formation_notify()
p.send_plunder_times_notify()
p.send_plunder_daily_reward_notify()
SpecialEquipmentGenerator(server_id, char_id).send_notify()
Party(server_id, char_id).send_notify()
ins = Inspire(server_id, char_id)
ins.try_open_slots(send_notify=False)
ins.send_notify()
cs = Championship(server_id, char_id)
cs.try_initialize(send_notify=False)
cs.send_notify()
WinningPlunder(server_id, char_id).send_notify()
WinningArena(server_id, char_id).send_notify()
WinningChampionship(server_id, char_id).send_notify()
Worship(server_id, char_id).send_notify()
CommonArenaWinningChat(server_id, char_id).send_notify()
CommonPlunderWinningChat(server_id, char_id).send_notify()
CommonChampionshipChat(server_id, char_id).send_notify()
send_system_notify(server_id, char_id)
BroadCast(server_id, char_id).try_cast_login_notify()
def setup(self):
TalentManager(self.server_id, self.char_id)
def force_load_staffs(self, send_notify=False):
from core.staff import StaffManger, Staff
from core.formation import Formation
from core.unit import UnitManager
from core.talent import TalentManager
from core.collection import Collection
from core.party import Party
from core.inspire import Inspire
from core.union import Union
from core.bag import Bag
self.formation_staffs = []
sm = StaffManger(self.server_id, self.char_id)
fm = Formation(self.server_id, self.char_id)
um = UnitManager(self.server_id, self.char_id)
ins = Inspire(self.server_id, self.char_id)
staffs = sm.get_staffs_data()
in_formation_staffs = fm.in_formation_staffs()
staff_objs = {}
""":type: dict[str, core.staff.Staff]"""
for k, v in staffs.items():
staff_objs[k] = Staff(self.server_id, self.char_id, k, v)
for k, v in staff_objs.iteritems():
if k in in_formation_staffs:
self.formation_staffs.append(v)
v.policy = in_formation_staffs[k]['policy']
v.formation_position = in_formation_staffs[k]['position']
unit_id = in_formation_staffs[k]['unit_id']
if unit_id:
v.set_unit(um.get_unit_object(unit_id))
working_staff_oids = fm.working_staff_oids()
for k in in_formation_staffs:
staff_objs[k].check_qianban(working_staff_oids)
staff_objs[k].add_self_talent_effect(self)
talent_effects_1 = TalentManager(self.server_id, self.char_id).get_talent_effects()
talent_effects_2 = Collection(self.server_id, self.char_id).get_talent_effects()
talent_effects_3 = fm.get_talent_effects()
talent_effects_4 = Party(self.server_id, self.char_id).get_talent_effects()
talent_effects_5 = Union(self.server_id, self.char_id).get_union_skill_talent_effects()
self.add_talent_effects(talent_effects_1)
self.add_talent_effects(talent_effects_2)
self.add_talent_effects(talent_effects_3)
self.add_talent_effects(talent_effects_4)
self.add_talent_effects(talent_effects_5)
config_inspire_level_addition, config_inspire_step_addition = ins.get_addition_config()
bag = Bag(self.server_id, self.char_id)
for _, v in staff_objs.iteritems():
v.config_inspire_level_addition = config_inspire_level_addition
v.config_inspire_step_addition = config_inspire_step_addition
v.calculate(bag=bag, um=um)
v.make_cache()
if send_notify:
self.send_notify()
in_formation_staff_ids = fm.in_formation_staffs().keys()
sm.send_notify(ids=in_formation_staff_ids)