def community_init(self, group): infected_id = random.sample( range(self.size), int(round(self.size * setting.INITIAL_CHANCE, 0))) self.get_empty_big_pos() # infected_id = random.sample(range(self.size), 1) for num in range(self.size): ent = Entities.Entity() _id = num + 1 ent.entity_init(setting.RESOLUTION, _id) # ent.entity_init(world_size=self.world_size, id=_id) ent.is_dist_kept = self.keep_dist_generator() self.entity_pool[f'{ent.id}'] = ent if _id in infected_id: ent.status.is_infected = 1 self.infected_id_pool.append((f'{_id}')) self.big_pos_getter(ent) group.add(ent)