def make_slot_msg(self, slot_ids=None): if not slot_ids: slot_ids = range(1, MAX_SLOT_AMOUNT + 1) sm = StaffManger(self.server_id, self.char_id) msgs = [] for _id in slot_ids: msg_slot = MsgFormationSlot() msg_slot.slot_id = int(_id) try: data = self.doc['slots'][str(_id)] except KeyError: msg_slot.status = FORMATION_SLOT_NOT_OPEN else: if not data['staff_id']: msg_slot.status = FORMATION_SLOT_EMPTY else: msg_slot.status = FORMATION_SLOT_USE msg_slot.staff_id = data['staff_id'] msg_slot.unit_id = data['unit_id'] if data['unit_id']: msg_slot.position = self.doc['position'].index(int(_id)) else: msg_slot.position = -1 msg_slot.staff_oid = sm.get_staff_object(data['staff_id']).oid msg_slot.policy = data.get('policy', 1) msgs.append(msg_slot) return msgs
def make_protobuf(self): msg = MsgPlunderFormation() msg.way = self.way_id msg.power = self.power sm = StaffManger(self.server_id, self.char_id) for k, v in self.doc['slots'].iteritems(): msg_slot = msg.formation.add() msg_slot.slot_id = int(k) if v['staff_id']: msg_slot.status = FORMATION_SLOT_USE msg_slot.staff_id = v['staff_id'] msg_slot.unit_id = v['unit_id'] if v['unit_id']: msg_slot.position = self.doc['position'].index(int(k)) else: msg_slot.position = -1 msg_slot.staff_oid = sm.get_staff_object(v['staff_id']).oid msg_slot.policy = v.get('policy', 1) else: msg_slot.status = FORMATION_SLOT_EMPTY skill_sequence = self.get_skill_sequence() for k, v in skill_sequence.iteritems(): notify_skill_sequence = msg.skill_sequence.add() notify_skill_sequence.id = k notify_skill_sequence.staff_id.extend(v) return msg
def _get_value_for_task_condition(self): # 给任务条件用的 # 要求上阵6人,每人都要有 键盘,鼠标,显示器装备 from core.bag import Bag staffs = self.in_formation_staffs().keys() if len(staffs) < 6: return 0, 0 qualities = [] levels = [] sm = StaffManger(self.server_id, self.char_id) bag = Bag(self.server_id, self.char_id) for sid in staffs: obj = sm.get_staff_object(sid) for bag_slot_id in [obj.equip_keyboard, obj.equip_mouse, obj.equip_monitor]: if not bag_slot_id: return 0, 0 item_data = bag.get_slot(bag_slot_id) qualities.append(ConfigItemNew.get(item_data['item_id']).quality) levels.append(item_data['level']) return min(qualities), min(levels)
def make_protobuf(self): msg = MsgPlunderFormation() msg.way = self.way_id msg.power = self.power sm = StaffManger(self.server_id, self.char_id) for k, v in self.doc['slots'].iteritems(): msg_slot = msg.formation.add() msg_slot.slot_id = int(k) if v['staff_id']: msg_slot.status = FORMATION_SLOT_USE msg_slot.staff_id = v['staff_id'] msg_slot.unit_id = v['unit_id'] if v['unit_id']: msg_slot.position = self.doc['position'].index(int(k)) else: msg_slot.position = -1 msg_slot.staff_oid = sm.get_staff_object(v['staff_id']).oid msg_slot.policy = v.get('policy', 1) else: msg_slot.status = FORMATION_SLOT_EMPTY skill_sequence = self.get_skill_sequence() for k, v in skill_sequence.iteritems(): notify_skill_sequence = msg.skill_sequence.add() notify_skill_sequence.id = k notify_skill_sequence.staff_id.extend(v) return msg
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 create(cls, server_id, char_id, club_name, club_flag): from core.staff import StaffManger from core.formation import Formation from core.mail import MailManager from apps.config.models import Mail as ModelMail doc = MongoCharacter.document() doc['_id'] = char_id doc['create_at'] = arrow.utcnow().timestamp doc['name'] = club_name doc['flag'] = club_flag doc['gold'] = CHAR_INIT_GOLD doc['diamond'] = CHAR_INIT_DIAMOND doc['crystal'] = CHAR_INIT_CRYSTAL doc['gas'] = CHAR_INIT_GAS sm = StaffManger(server_id, char_id) formation_init_data = [] for staff_id, unit_id in CHAR_INIT_STAFFS: uid = sm.add(staff_id, send_notify=False, trig_signal=False) formation_init_data.append((uid, unit_id)) fm = Formation(server_id, char_id) fm.initialize(formation_init_data) MongoCharacter.db(server_id).insert_one(doc) # add welfare mail start_time = get_start_time_of_today() condition = Q(send_at__gte=start_time.format("YYYY-MM-DD HH:mm:ssZ")) &\ Q(send_at__lte=arrow.utcnow().format("YYYY-MM-DD HH:mm:ssZ")) mails = ModelMail.objects.filter(condition) m = MailManager(server_id, char_id) for m_obj in mails: if not m_obj.welfare: continue ok = False if m_obj.condition_type == 1: ok = True elif m_obj.condition_type == 2 and server_id in m_obj.get_parsed_condition_value(): ok = True elif m_obj.condition_type == 3 and server_id not in m_obj.get_parsed_condition_value(): ok = True if not ok: continue if m_obj.items: rc = ResourceClassification.classify(m_obj.get_parsed_items()) attachment = rc.to_json() else: attachment = "" m.add(m_obj.title, m_obj.content, attachment=attachment, send_notify=False)
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 load_staffs(self): from core.staff import StaffManger from core.formation import Formation self.formation_staffs = [] sm = StaffManger(self.server_id, self.char_id) fm = Formation(self.server_id, self.char_id) for k in fm.in_formation_staffs(): self.formation_staffs.append(sm.get_staff_object(k))
def test_match(self): staff_ids = ConfigStaff.INSTANCES.keys() staff_ids = random.sample(staff_ids, 10) sm = StaffManger(1, 1) for i in staff_ids: sm.add(i) Club(1, 1).set_match_staffs(staff_ids) Challenge(1, 1).start()
def test_match(self): staff_ids = ConfigStaff.INSTANCES.keys() staff_ids = random.sample(staff_ids, 10) sm = StaffManger(1, 1) for i in staff_ids: sm.add(i) Club(1, 1).set_match_staffs(staff_ids) Challenge(1, 1).start()
def test_add(self): staff_ids = ConfigStaff.INSTANCES.keys() doc = MongoStaff.db(self.server_id).find_one({'_id': self.char_id}, {'staffs': 1}) add_id = 0 for staff_id in staff_ids: if str(staff_id) not in doc['staffs'].keys(): add_id = staff_id StaffManger(self.server_id, self.char_id).add(add_id) assert StaffManger(self.server_id, self.char_id).has_staff(add_id) is True
def test_add_duplicate(self): doc = MongoStaff.db(self.server_id).find_one({'_id': self.char_id}, {'staffs': 1}) staff_id = int(random.choice(doc['staffs'].keys())) assert StaffManger(self.server_id, self.char_id).has_staff(staff_id) is True try: StaffManger(self.server_id, self.char_id).add(staff_id) except GameException as e: assert e.error_id == ConfigErrorMessage.get_error_id("STAFF_ALREADY_HAVE") else: raise Exception("can not be here!")
def batch_destroy(request): server_id = request._game_session.server_id char_id = request._game_session.char_id staff_ids = request._proto.staff_ids rc = StaffManger(server_id, char_id).batch_destroy([i for i in staff_ids]) response = StaffBatchDestroyResponse() response.ret = 0 response.drop.MergeFrom(rc.make_protomsg()) return ProtobufResponse(response)
def test_match(self): self.set_friend_data() staff_ids = random.sample(ConfigStaff.INSTANCES.keys(), 10) for i in staff_ids: StaffManger(1, 1).add(i) StaffManger(1, 2).add(i) Club(1, 1).set_match_staffs(staff_ids) Club(1, 2).set_match_staffs(staff_ids) FriendManager(1, 1).match(2)
def batch_destroy(request): server_id = request._game_session.server_id char_id = request._game_session.char_id staff_ids = request._proto.staff_ids rc = StaffManger(server_id, char_id).batch_destroy([i for i in staff_ids]) response = StaffBatchDestroyResponse() response.ret = 0 response.drop.MergeFrom(rc.make_protomsg()) return ProtobufResponse(response)
def destroy(request): server_id = request._game_session.server_id char_id = request._game_session.char_id staff_id = request._proto.staff_id tp = request._proto.tp resource_classified = StaffManger(server_id, char_id).destroy(staff_id, tp) response = StaffDestroyResponse() response.ret = 0 response.drop.MergeFrom(resource_classified.make_protomsg()) return ProtobufResponse(response)
def destroy(request): server_id = request._game_session.server_id char_id = request._game_session.char_id staff_id = request._proto.staff_id tp = request._proto.tp resource_classified = StaffManger(server_id, char_id).destroy(staff_id, tp) response = StaffDestroyResponse() response.ret = 0 response.drop.MergeFrom(resource_classified.make_protomsg()) return ProtobufResponse(response)
def set_staff(self, slot_id, staff_id): from core.inspire import Inspire if str(slot_id) not in self.doc['slots']: raise GameException(ConfigErrorMessage.get_error_id("FORMATION_SLOT_NOT_OPEN")) sm = StaffManger(self.server_id, self.char_id) sm.check_staff(ids=[staff_id]) Inspire(self.server_id, self.char_id).check_staff_in(staff_id) this_staff_obj = sm.get_staff_object(staff_id) in_formation_staffs = self.in_formation_staffs() for k, v in in_formation_staffs.iteritems(): if k == staff_id: raise GameException(ConfigErrorMessage.get_error_id("FORMATION_STAFF_ALREADY_IN")) if v['slot_id'] != slot_id and sm.get_staff_object(k).oid == this_staff_obj.oid: # 其他位置 不能上 oid 一样的人 # 但是这个slot替换就可以 raise GameException(ConfigErrorMessage.get_error_id("FORMATION_STAFF_ALREADY_IN")) old_staff_id = self.doc['slots'][str(slot_id)]['staff_id'] self.doc['slots'][str(slot_id)]['staff_id'] = staff_id self.doc['slots'][str(slot_id)]['unit_id'] = 0 self.doc['slots'][str(slot_id)]['policy'] = 1 updater = { 'slots.{0}.staff_id'.format(slot_id): staff_id, 'slots.{0}.unit_id'.format(slot_id): 0, 'slots.{0}.policy'.format(slot_id): 1 } if slot_id in self.doc['position']: # 换完人,兵种清空了,也就从阵型位置上撤掉了 index = self.doc['position'].index(slot_id) self.doc['position'][index] = 0 updater['position.{0}'.format(index)] = 0 self.MONGO_COLLECTION.db(self.server_id).update_one( {'_id': self.char_id}, {'$set': updater} ) if old_staff_id: self.skill_sequence_try_unset_staff(old_staff_id) return old_staff_id
def test_remove(self): staff_ids = ConfigStaff.INSTANCES.keys() doc = MongoStaff.db(self.server_id).find_one({'_id': self.char_id}) staff_id = 0 for i in staff_ids: if str(i) not in doc['staffs'].keys(): staff_id = i break StaffManger(self.server_id, self.char_id).add(staff_id) assert StaffManger(self.server_id, self.char_id).has_staff(staff_id) is True StaffManger(self.server_id, self.char_id).remove(staff_id) assert StaffManger(self.server_id, self.char_id).has_staff(staff_id) is False
def test_set_staffs(self): staff_ids = random.sample(ConfigStaff.INSTANCES.keys(), 10) sm = StaffManger(1, 1) for sid in staff_ids: sm.add(sid) Club(1, 1).set_match_staffs(staff_ids) assert len(Club(1, 1).match_staffs) == 5 assert len(Club(1, 1).tibu_staffs) == 5 for c in Club(1, 1).match_staffs: assert c != 0 for c in Club(1, 1).tibu_staffs: assert c != 0
def test_set_staffs(self): staff_ids = random.sample(ConfigStaff.INSTANCES.keys(), 10) sm = StaffManger(1, 1) for sid in staff_ids: sm.add(sid) Club(1, 1).set_match_staffs(staff_ids) assert len(Club(1, 1).match_staffs) == 5 assert len(Club(1, 1).tibu_staffs) == 5 for c in Club(1, 1).match_staffs: assert c != 0 for c in Club(1, 1).tibu_staffs: assert c != 0
def check_exist(self, server_id, char_id): from core.club import Club from core.bag import Bag from core.staff import StaffManger, StaffRecruit from core.territory import Territory from core.arena import Arena from core.energy import Energy money_text = self.money_as_text_dict() if money_text: Club(server_id, char_id).check_money(**money_text) if self.bag: Bag(server_id, char_id).check_items(self.bag) if self.staff: StaffManger(server_id, char_id).check_original_staff_is_initial_state( self.staff) if self.territory_product: Territory(server_id, char_id).check_product(self.territory_product) if self.work_card: Territory(server_id, char_id).check_work_card(self.work_card) if self.arena_point: Arena(server_id, char_id).check_point(self.arena_point) if self.energy: Energy(server_id, char_id).check(self.energy) if self.staff_recruit_score: StaffRecruit(server_id, char_id).check_score(self.staff_recruit_score) if self.resource_data: _r = _Resource() _r.resource = dict(self.resource_data) _r.check_exists(server_id, char_id)
def get_addition_config(self): sm = StaffManger(self.server_id, self.char_id) # XXX 不能用 get_staff_object all_staffs = sm.get_staffs_data() levels = 0 steps = 0 for s in self.all_staffs(): data = all_staffs[s] levels += data['level'] steps += data['step'] config_level_addition = ConfigInspireAddition.get_by_level(levels) config_step_addition = ConfigInspireAddition.get_by_step(steps) return config_level_addition, config_step_addition
def get_addition_config(self): sm = StaffManger(self.server_id, self.char_id) # XXX 不能用 get_staff_object all_staffs = sm.get_staffs_data() levels = 0 steps = 0 for s in self.all_staffs(): data = all_staffs[s] levels += data['level'] steps += data['step'] config_level_addition = ConfigInspireAddition.get_by_level(levels) config_step_addition = ConfigInspireAddition.get_by_step(steps) return config_level_addition, config_step_addition
def test_level_up(self): doc = MongoStaff.db(self.server_id).find_one({'_id': self.char_id}, {'staffs': 1}) s = ConfigStaff.get(int(random.choice(doc['staffs'].keys()))) level = 5 exp = 0 for i in range(1, level): exp += ConfigStaffLevel.get(i).exp[s.quality] exp += 1 assert StaffManger(self.server_id, self.char_id).get_staff_object(s.id).level == 1 assert StaffManger(self.server_id, self.char_id).get_staff_object(s.id).exp == 0 StaffManger(self.server_id, self.char_id).update(s.id, exp=exp) assert StaffManger(self.server_id, self.char_id).get_staff_object(s.id).level == level assert StaffManger(self.server_id, self.char_id).get_staff_object(s.id).exp == 1
def is_staff_training_check_by_oid(self, oid): sm = StaffManger(self.server_id, self.char_id) for k, v in self.get_all_building_objects().iteritems(): if not v.open: continue for _, s in v.slots.iteritems(): if not s.open: continue if not s.staff_id: continue if sm.get_staff_object(s.staff_id).oid == oid: return True return False
def working_staff_oids(self): from core.inspire import Inspire oids = [] sm = StaffManger(self.server_id, self.char_id) # XXX: 这里不能用 get_staff_object # 因为get_staff_obj 可能用调用到 Club.force_load_staffs # 在 Club.force_load_staffs 中又会调用 这个方法 # 然后就死循环了 all_staffs = sm.get_staffs_data() working_staffs = self.in_formation_staffs().keys() working_staffs.extend(Inspire(self.server_id, self.char_id).all_staffs()) for s in working_staffs: oids.append(all_staffs[s]['oid']) return oids
def is_staff_training_check_by_oid(self, oid): sm = StaffManger(self.server_id, self.char_id) for k, v in self.get_all_building_objects().iteritems(): if not v.open: continue for _, s in v.slots.iteritems(): if not s.open: continue if not s.staff_id: continue if sm.get_staff_object(s.staff_id).oid == oid: return True return False
def harass(self, union_id, staff_id): doc = MongoUnion.db(self.server_id).find_one({'_id': union_id}, {'owner': 1}) if not doc: raise GameException( ConfigErrorMessage.get_error_id("UNION_NOT_EXIST")) staff = StaffManger(self.server_id, self.char_id).get_staff_object(staff_id) hi = HarassInfo(self.server_id, self.char_id) if not hi.remained_times: hi.buy_times() if staff.oid == self.member_doc['harass_staff']: param = 2 else: param = 1 hi.record() union_coin = int( (math.pow(staff.level, 0.2) + math.pow(staff.step + 1, 0.5)) * 3.5 * param) explore_point = int( (math.pow(staff.level, 0.2) + math.pow(staff.step + 1, 0.5)) * 7 * param) union_skill_point = int( (math.pow(staff.level, 0.2) + math.pow(staff.step + 1, 0.5)) * 5 * param) my_explore_point = int( (math.pow(staff.level, 0.2) + math.pow(staff.step + 1, 0.5)) * 5 * param) self.add_explore_point(my_explore_point) Union(self.server_id, doc['owner']).add_explore_point(-explore_point) reward = ConfigUnionExplore.get_harass_reward() reward = [(_id, _amount * param) for _id, _amount in reward] reward.append((UNION_COIN_ID, union_coin)) reward.append((UNION_SKILL_POINT_ID, union_skill_point)) rc = ResourceClassification.classify(reward) rc.add(self.server_id, self.char_id, message="UnionJoined.harass") self.member_doc['harass_staff'] = ConfigUnionExplore.get_staff_id() MongoUnionMember.db(self.server_id).update_one( {'_id': self.char_id}, {'$set': { 'harass_staff': self.member_doc['harass_staff'] }}) self.send_explore_notify() return explore_point, rc, my_explore_point
def step_up(request): server_id = request._game_session.server_id char_id = request._game_session.char_id staff_id = request._proto.staff_id StaffManger(server_id, char_id).step_up(staff_id) response = StaffStepUpResponse() response.ret = 0 return ProtobufResponse(response)
def test_recruit(self): config = ConfigStaffRecruit.get(RECRUIT_ENUM_TO_CONFIG_ID[RECRUIT_NORMAL]) cost_type = 'gold' if config.cost_type == 1 else 'diamond' self.update(**{cost_type: config.cost_value}) StaffRecruit(self.server_id, self.char_id).refresh(RECRUIT_NORMAL) staff_ids = StaffRecruit(self.server_id, self.char_id).get_self_refreshed_staffs() doc = MongoStaff.db(self.server_id).find_one({'_id': self.char_id}, {'staffs': 1}) test_id = 0 for staff_id in staff_ids: if str(staff_id) not in doc['staffs'].keys(): test_id = staff_id break assert StaffManger(self.server_id, self.char_id).has_staff(test_id) is False staff_cfg = ConfigStaff.get(test_id) tp = 'gold' if staff_cfg.buy_type == 1 else 'diamond' self.update(**{tp: staff_cfg.buy_cost}) StaffRecruit(self.server_id, self.char_id).recruit(test_id) assert StaffManger(self.server_id, self.char_id).has_staff(test_id) is True
def level_up(request): server_id = request._game_session.server_id char_id = request._game_session.char_id staff_id = request._proto.staff_id up_level = request._proto.up_level StaffManger(server_id, char_id).level_up(staff_id, up_level) response = StaffLevelUpResponse() response.ret = 0 return ProtobufResponse(response)
def equipment_change(request): server_id = request._game_session.server_id char_id = request._game_session.char_id staff_id = request._proto.staff_id slot_id = request._proto.slot_id tp = request._proto.tp StaffManger(server_id, char_id).equipment_change(staff_id, slot_id, tp) response = StaffEquipChangeResponse() response.ret = 0 return ProtobufResponse(response)
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 test_remove_not_exist(self): staff_ids = ConfigStaff.INSTANCES.keys() staff_id = 0 for i in range(1, 10000): if str(i) not in staff_ids: staff_id = i break try: StaffManger(self.server_id, self.char_id).remove(staff_id) except GameException as e: assert e.error_id == ConfigErrorMessage.get_error_id("STAFF_NOT_EXIST") else: raise Exception("can not be here!")
def test_add_not_exist(self): def get_id(): while True: staff_id = random.randint(1, 1000000) if staff_id not in ConfigStaff.INSTANCES.keys(): return staff_id staff_id = get_id() try: StaffManger(self.server_id, self.char_id).add(staff_id) except GameException as e: assert e.error_id == ConfigErrorMessage.get_error_id("STAFF_NOT_EXIST") else: raise Exception("can not be here!")
def star_up(request): server_id = request._game_session.server_id char_id = request._game_session.char_id staff_id = request._proto.staff_id single = request._proto.single crit, inc_exp, cost_item_id, cost_item_amount = StaffManger( server_id, char_id).star_up(staff_id, single) response = StaffStarUpResponse() response.ret = 0 response.crit = crit response.increase = inc_exp response.cost_item_id = cost_item_id response.cost_item_amount = cost_item_amount return ProtobufResponse(response)
def explore(self, staff_id): staff = StaffManger(self.server_id, self.char_id).get_staff_object(staff_id) ei = ExploreInfo(self.server_id, self.char_id) ei.check_cd() if not ei.remained_times: raise GameException( ConfigErrorMessage.get_error_id("UNION_EXPLORE_NO_TIMES")) if staff.oid == self.member_doc['explore_staff']: param = 2 else: param = 1 ei.record() union_coin = int( (math.pow(staff.level, 0.2) + math.pow(staff.step + 1, 0.5)) * 3 * param) explore_point = int( (math.pow(staff.level, 0.2) + math.pow(staff.step + 1, 0.5)) * 10 * param) union_skill_point = int( (math.pow(staff.level, 0.2) + math.pow(staff.step + 1, 0.5)) * 4 * param) self.add_explore_point(explore_point) reward = ConfigUnionExplore.get_explore_reward() reward = [(_id, _amount * param) for _id, _amount in reward] reward.append((UNION_COIN_ID, union_coin)) reward.append((UNION_SKILL_POINT_ID, union_skill_point)) rc = ResourceClassification.classify(reward) rc.add(self.server_id, self.char_id, message="UnionJoined.explore") self.member_doc['explore_staff'] = ConfigUnionExplore.get_staff_id() MongoUnionMember.db(self.server_id).update_one( {'_id': self.char_id}, {'$set': { 'explore_staff': self.member_doc['explore_staff'] }}) self.send_explore_notify() return explore_point, rc
def get_building_reward(self): if not self.open: return 0, 0 if not self.staff_id: return 0, 0 staff_obj = StaffManger(self.server_id, self.char_id).get_staff_object(self.staff_id) quality_modulus = STAFF_QUALITY_MODULUS[staff_obj.quality] slot_modulus = ConfigTerritoryBuilding.get( self.building_id).slots[self.id].exp_modulus exp = 10 * math.pow(self.hour, 0.8) * quality_modulus * slot_modulus product_amount = 100 * math.pow(self.hour, 0.8) * quality_modulus * slot_modulus return int(exp), int(product_amount)
def remove(self, server_id, char_id, message=""): from core.club import Club from core.bag import Bag from core.staff import StaffManger, StaffRecruit from core.territory import Territory from core.arena import Arena from core.energy import Energy money_text = self.money_as_text_dict() if money_text: money_text = {k: -v for k, v in money_text.iteritems()} money_text['message'] = message Club(server_id, char_id).update(**money_text) if self.bag: bag = Bag(server_id, char_id) for _id, _amount in self.bag: bag.remove_by_item_id(_id, _amount) if self.staff: StaffManger(server_id, char_id).internal_remove_by_oid(self.staff) if self.territory_product: Territory(server_id, char_id).remove_product(self.territory_product) if self.work_card: Territory(server_id, char_id).remove_work_card(self.work_card) if self.arena_point: Arena(server_id, char_id).remove_point(self.arena_point) if self.energy: Energy(server_id, char_id).remove(self.energy) if self.staff_recruit_score: StaffRecruit(server_id, char_id).remove_score(self.staff_recruit_score) if self.resource_data: _r = _Resource() _r.resource = dict(self.resource_data) _r.remove(server_id, char_id)
def test_recruit_already_have(self): config = ConfigStaffRecruit.get(RECRUIT_ENUM_TO_CONFIG_ID[RECRUIT_NORMAL]) cost_type = 'gold' if config.cost_type == 1 else 'diamond' self.update(**{cost_type: config.cost_value}) StaffRecruit(self.server_id, self.char_id).refresh(RECRUIT_NORMAL) staff_ids = StaffRecruit(self.server_id, self.char_id).get_self_refreshed_staffs() doc = MongoStaff.db(self.server_id).find_one({'_id': 1}, {'staffs': 1}) test_id = 0 for staff_id in staff_ids: if str(staff_id) not in doc['staffs'].keys(): StaffManger(self.server_id, self.char_id).add(staff_id) test_id = staff_id break try: StaffRecruit(self.server_id, self.char_id).recruit(test_id) except GameException as e: assert e.error_id == ConfigErrorMessage.get_error_id("STAFF_ALREADY_HAVE") else: raise Exception("can not be here!")
def setup_class(cls): MongoCharacter.db(1).update_one({'_id': 1}, {'$set': { 'club.level': 10 }}) doc = MongoStaff.db(1).find_one({'_id': 1}, {'staffs': 1}) staff_ids = random.sample(ConfigStaff.INSTANCES.keys(), 15) for i in staff_ids: if str(i) not in doc['staffs']: StaffManger(1, 1).add(i) match_staff_ids = [] for k in doc['staffs'].keys(): match_staff_ids.append(int(k)) for staff_id in staff_ids: if staff_id not in match_staff_ids: match_staff_ids.append(staff_id) Club(1, 1).set_match_staffs(random.sample(match_staff_ids, 10)) LeagueGame.new(1)
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)
def equipment_level_up(self, slot_id, times=1): from core.staff import StaffManger from core.formation import Formation from core.plunder import Plunder this_slot = self.doc['slots'][slot_id] item_id = this_slot['item_id'] if get_item_type(item_id) != TYPE_EQUIPMENT: raise GameException(ConfigErrorMessage.get_error_id("INVALID_OPERATE")) equip = Equipment.load_from_slot_data(this_slot) level = equip.level if equip.is_special: if level >= ConfigEquipmentSpecialLevel.MAX_LEVEL: raise GameException(ConfigErrorMessage.get_error_id("EQUIPMENT_SELF_REACH_MAX_LEVEL")) station_level_limit = Plunder(self.server_id, self.char_id).get_station_level() * 2 if level >= station_level_limit: raise GameException(ConfigErrorMessage.get_error_id("EQUIPMENT_SPECIAL_REACH_MAX_LEVEL")) max_level = min(ConfigEquipmentSpecialLevel.MAX_LEVEL, station_level_limit) else: config = ConfigEquipmentNew.get(item_id) if level >= config.max_level: raise GameException(ConfigErrorMessage.get_error_id("EQUIPMENT_SELF_REACH_MAX_LEVEL")) club_level_limit = get_club_property(self.server_id, self.char_id, 'level') * 2 if level >= club_level_limit: raise GameException(ConfigErrorMessage.get_error_id("EQUIPMENT_REACH_MAX_LEVEL")) max_level = min(config.max_level, club_level_limit) can_add_level = max_level - level if times > can_add_level: times = can_add_level def do_level_up(_add): _item_needs = equip.get_level_up_needs_items(_add) resource_classified = ResourceClassification.classify(_item_needs) resource_classified.check_exist(self.server_id, self.char_id) resource_classified.remove(self.server_id, self.char_id, message="Bag.equipment_level_up:{0}".format(item_id)) old_level = level error_code = 0 for i in range(times, 0, -1): try: do_level_up(i) except GameException as e: error_code = e.error_id else: level += i break if level > old_level: self.doc['slots'][slot_id]['level'] = level MongoBag.db(self.server_id).update_one( {'_id': self.char_id}, {'$set': { 'slots.{0}.level'.format(slot_id): level }} ) ValueLogEquipmentLevelUpTimes(self.server_id, self.char_id).record(value=level - old_level) self.send_notify(slot_ids=[slot_id]) sm = StaffManger(self.server_id, self.char_id) staff_id = sm.find_staff_id_with_equip(slot_id) if staff_id: s_obj = sm.get_staff_object(staff_id) s_obj.calculate() s_obj.make_cache() sm.send_notify(ids=[staff_id]) fm = Formation(self.server_id, self.char_id) if staff_id in fm.in_formation_staffs(): Club(self.server_id, self.char_id).send_notify() task_condition_trig_signal.send( sender=None, server_id=self.server_id, char_id=self.char_id, condition_name='core.formation.Formation' ) return error_code, level != old_level, Equipment.load_from_slot_data(self.doc['slots'][slot_id])
def training_star(self, building_id, slot_id, staff_id, hour): if hour not in TRAINING_HOURS: raise GameException(ConfigErrorMessage.get_error_id('BAD_MESSAGE')) sm = StaffManger(self.server_id, self.char_id) sm.check_staff([staff_id]) building = self.get_building_object(building_id, slots_ids=[slot_id]) try: slot = building.slots[slot_id] except KeyError: raise GameException(ConfigErrorMessage.get_error_id("INVALID_OPERATE")) if not slot.open: raise GameException(ConfigErrorMessage.get_error_id("TERRITORY_SLOT_LOCK")) if slot.staff_id: raise GameException(ConfigErrorMessage.get_error_id("TERRITORY_SLOT_HAS_STAFF")) if self.is_staff_training_check_by_oid(sm.get_staff_object(staff_id).oid): raise GameException(ConfigErrorMessage.get_error_id("TERRITORY_STAFF_IS_TRAINING")) config_slot = ConfigTerritoryBuilding.get(building_id).slots[slot_id] cost_amount = config_slot.get_cost_amount(building.level, TRAINING_HOURS.index(hour)) new_amount = self.check_work_card(cost_amount) self.doc['work_card'] = new_amount slot.staff_id = staff_id slot.hour = hour start_at = arrow.utcnow().timestamp slot_doc = MongoTerritory.document_slot() slot_doc['staff_id'] = staff_id slot_doc['start_at'] = start_at slot_doc['hour'] = hour building_exp, product_amount = slot.get_building_reward() reward = { 'building_exp': building_exp, 'product_amount': product_amount, 'items': [] } ri = ResultItems() # 固定奖励 report = [ ( 1, [str(building_exp), ConfigItemNew.get(slot.product_id).name, str(product_amount)], 0, ), ] # 选手特产 config_staff_product = ConfigTerritoryStaffProduct.get(sm.get_staff_object(staff_id).oid) if config_staff_product: _id, _amount = config_staff_product.get_product(TRAINING_HOURS.index(hour)) if _amount: report.append(( 2, [ConfigItemNew.get(_id).name, str(_amount)], 0, )) ri.add(_id, _amount) # 概率奖励 end_at = start_at + hour * 3600 extra_at = start_at + 3600 while extra_at < end_at: _id, _amount = ConfigTerritoryBuilding.get(building_id).slots[slot_id].get_extra_product(1) report.append(( 3, [ConfigItemNew.get(_id).name, str(_amount)], extra_at )) ri.add(_id, _amount) extra_at += 3600 reward['items'] = ri.items.items() slot_doc['report'] = report slot_doc['reward'] = reward self.doc['buildings'][str(building_id)]['slots'][str(slot_id)] = slot_doc MongoTerritory.db(self.server_id).update_one( {'_id': self.char_id}, {'$set': { 'work_card': self.doc['work_card'], 'buildings.{0}.slots.{1}'.format(building_id, slot_id): slot_doc }} ) ValueLogTerritoryTrainingTimes(self.server_id, self.char_id).record() self.send_notify(building_id=building_id, slot_id=slot_id)
def _single_response(_char): context['show'] = True context['char_id'] = _char.id context['char_name'] = _char.name context['server_id'] = _char.server_id _club = Club(_char.server_id, _char.id) _bag = Bag(_char.server_id, _char.id) _vip = VIP(_char.server_id, _char.id) _sm = StaffManger(_char.server_id, _char.id) _fm = Formation(_char.server_id, _char.id) _te = Territory(_char.server_id, _char.id) _union = Union(_char.server_id, _char.id) context['last_login'] = arrow.get(_club.last_login).to(settings.TIME_ZONE).format("YYYY-MM-DD HH:mm:ss") context['club_level'] = _club.level context['vip_level'] = _vip.doc['vip'] context['vip_exp'] = _vip.doc['exp'] context['power'] = _club.power context['diamond'] = _club.diamond context['gold'] = _club.gold context['crystal'] = _club.crystal context['gas'] = _club.gas context['item_30007'] = _bag.get_amount_by_item_id(30007) context['item_30008'] = _bag.get_amount_by_item_id(30008) context['item_30009'] = _bag.get_amount_by_item_id(30009) context['item_30015'] = _te.get_work_card_amount() _res = MongoResource.db(_char.server_id).find_one({'_id': _char.id}) if not _res: _res = {'resource': {}} context['item_30022'] = _res['resource'].get('30022', 0) context['item_30023'] = _res['resource'].get('30023', 0) context['item_30019'] = _res['resource'].get('30019', 0) _union_id = _union.get_joined_union_id() if not _union_id: context['union'] = {} else: context['union'] = { 'id': _union_id, 'name': _union.union_doc['name'], 'joined_at': arrow.get(_union.member_doc['joined_at']).to(settings.TIME_ZONE).format( "YYYY-MM-DD HH:mm:ss"), 'contribution': _union.member_doc['contribution'] } in_formation_staffs = _fm.in_formation_staffs() staffs_data = [] equip_data = [] for k, v in in_formation_staffs.iteritems(): staff_obj = _sm.get_staff_object(k) staff_name = ConfigItemNew.get(staff_obj.oid).name staffs_data.append({ 'id': staff_obj.id, 'oid': staff_obj.oid, 'name': staff_name, 'level': staff_obj.level, 'step': staff_obj.step, 'star': staff_obj.star, 'power': staff_obj.power, 'unit_id': v['unit_id'], 'position': v['position'], }) for bag_slot_id in [staff_obj.equip_special, staff_obj.equip_decoration, staff_obj.equip_keyboard, staff_obj.equip_monitor, staff_obj.equip_mouse]: if not bag_slot_id: continue equip = _bag.get_slot(bag_slot_id) equip_data.append({ 'oid': equip['item_id'], 'level': equip['level'], 'name': ConfigItemNew.get(equip['item_id']).name, 'staff_name': staff_name }) context['staffs'] = staffs_data context['equips'] = equip_data return render_to_response( 'dianjing_statistics_char.html', context=context )