def assign_generators(generators, context, warnings):
for generator in generators:
assign(generator.target,
generator.iter,
context,
warnings,
generator=True)
def new_ext(self):
self.filecheck()
total = self.usercount()
result = pd.DataFrame(columns=["Name", "Direct Number", "Extension"])
with open(self.filepath) as userlist:
reader = csv.DictReader(userlist)
for line_num, row in enumerate(reader):
# row keys: ['givenName'], ['surname'], ['name'], ['emailAddress'], ['Title'], ['Department']
name, ext = assign(row, total, line_num)
if name:
num = set_forward(name, ext)
result = result.append({"Name": name, "Direct Number": num, "Extension": ext}, ignore_index=True)
self.pnl('Extension assignment and configuration for {0} successful.'.format(name))
else:
result = result.append({"Name": name, "Direct Number": "Assignment Failed",
"Extension": "Assignment failed"}, ignore_index=True)
self.pnl('RingCentral accounts created successfully.')
self.pnl(result)
outfile = export_csv(self.filepath, result)
if outfile != IOError:
self.pnl('Results written to {0}'.format(outfile))
kill_browser()
def manga_refresh(manga):
query = session.query(Manga).filter(Manga.name==manga.name).first()
if(query):
assign(manga.name, query.name)
assign(manga.added_at, query.added_at)
assign(manga.update_at, query.update_at)
assign(manga.author, query.author)
assign(manga.source, query.source)
assign(manga.poster, query.poster)
assign(manga.introduction, query.introduction)
# 更新剧情信息
for item in manga.plot:
query_plot = queryPlotWithIndex(item.index, query.plot)
if (query_plot):
assign(item.title, query_plot.title)
assign(item.link, query_plot.link)
else:
plot_new = Plot(manga_id=query.id, link=item.link, title=item.title, index=item.index)
query.plot.append(plot_new)
else:
session.add(manga)
session.flush()
pass
import time
from assign import write_csv, assign
# for 14 tasks
start = time.time()
ppl = 'data/in/people.csv'
skills = 'data/in/skillMatrix.csv'
tasks = 'data/in/tasks-small.csv'
filename = 'data/out/assignments-small.csv'
write_csv(assign(ppl, skills, tasks), filename)
end = time.time()
print('Time taken for 14 tasks: ', end - start)
# for 300 tasks
start = time.time()
ppl = 'data/in/people.csv'
skills = 'data/in/skillMatrix.csv'
tasks = 'data/in/tasks-large.csv'
filename = 'data/out/assignments-large.csv'
write_csv(assign(ppl, skills, tasks), filename)
end = time.time()
print('Time taken for 300 tasks: ', end - start)
def test_cns_fmt_dict(self):
for cns, qfeat, sfeat in assign(self.cns_dict, self.qbed, self.sbed,
self.qpair_map, self.spair_map):
d = cns_fmt_dict(cns, qfeat, sfeat, self.evalue_dict)
print "dddddddd", d
def test_assign(self):
assign(self.cns_dict, self.qbed, self.sbed, self.qpair_map,
self.spair_map)
def get_action(self, referee, obstacle_avoidance, ball_info, robot_info=None, defense_num=0):
self._control_target.control_enable = True
remake_path = False # 経路再生成のフラグ TODO:remake_pathを活用する
avoid_obstacle = True # 障害物回避の経路追加フラグ
avoid_ball = False # ボール回避の経路追加フラグ
zone_enable = False
# パラメータ初期化
self._control_target.dribble_power = 0.0
self._control_target.kick_power = 0.0
if referee.can_move_robot is False or ball_info.disappeared:
# 移動禁止 or ボールの消失で制御を停止する
rospy.logdebug("HALT")
self._control_target, remake_path= normal.stop(self._control_target)
avoid_obstacle = False # 障害物回避しない
elif referee.is_inplay:
rospy.logdebug("IN-PLAY")
zone_enable = True
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
if tool.is_in_defense_area(ball_info.pose, 'our'):
self._control_target = offense.outside_shoot(
self._my_pose, ball_info, self._control_target)
else:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
if tool.is_in_defense_area(ball_info.pose, 'our'):
# ボールが自チームのディフェンスエリアにある場合は行動を変える
self._control_target = normal.move_to(
self._control_target, Pose2D(0,0,0), ball_info, look_ball=True)
elif tool.is_in_defense_area(ball_info.pose, 'their'):
# ボールが相手チームのディフェンスエリアにある場合は行動を変える
self._control_target = normal.keep_x(
self._control_target,
Field.penalty_pose('their', 'upper_front').x - 1.0,
ball_info)
else:
self._control_target = offense.inplay_shoot(
self._my_pose, ball_info, self._control_target)
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info, zone_enable)
else:
if referee.referee_id == ref.REFEREE_ID["STOP"]:
rospy.logdebug("STOP")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target = offense.interpose(ball_info,
self._control_target, dist_from_target=0.7)
avoid_ball = True
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info)
elif referee.referee_id == ref.REFEREE_ID["OUR_KICKOFF_PREPARATION"]:
rospy.logdebug("OUR_KICKOFF_PREPARATION")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target, avoid_ball = offense.setplay_shoot(
self._my_pose, ball_info, self._control_target,
kick_enable=False)
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info)
elif referee.referee_id == ref.REFEREE_ID["OUR_KICKOFF_START"]:
rospy.logdebug("OUR_KICKOFF_START")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target, avoid_ball = offense.setplay_shoot(
self._my_pose, ball_info, self._control_target,
kick_enable=True)
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info)
elif referee.referee_id == ref.REFEREE_ID["OUR_PENALTY_PREPARATION"]:
rospy.logdebug("OUR_PENALTY_PREPARATION")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target, avoid_ball = offense.setplay_shoot(
self._my_pose, ball_info, self._control_target,
kick_enable=False)
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info)
elif referee.referee_id == ref.REFEREE_ID["OUR_PENALTY_START"]:
rospy.logdebug("OUR_PENALTY_START")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target, avoid_ball = offense.setplay_shoot(
self._my_pose, ball_info, self._control_target,
kick_enable=True, penalty=True)
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info)
elif referee.referee_id == ref.REFEREE_ID["OUR_DIRECT_FREE"]:
rospy.logdebug("OUR_DIRECT_FREE")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target, avoid_ball = offense.setplay_pass(
self._my_pose, ball_info, self._control_target,
Pose2D(Field.field('length')*0.25, 0, 0),
receive_enable=True, receiver_role_exist=Observer.role_is_exist(role.ROLE_ID["ROLE_ZONE_1"]),
robot_info=robot_info, direct=True)
elif self._my_role == role.ROLE_ID["ROLE_ZONE_1"]:
self._control_target = normal.move_to(
self._control_target, Pose2D(Field.field('length')*0.25,0,0), ball_info, look_ball=True)
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info)
elif referee.referee_id == ref.REFEREE_ID["OUR_INDIRECT_FREE"]:
rospy.logdebug("OUR_INDIRECT_FREE")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target, avoid_ball = offense.setplay_pass(
self._my_pose, ball_info, self._control_target,
Pose2D(Field.field('length')*0.25, 0, 0),
receive_enable=True, receiver_role_exist=Observer.role_is_exist(role.ROLE_ID["ROLE_ZONE_1"]),
robot_info=robot_info)
elif self._my_role == role.ROLE_ID["ROLE_ZONE_1"]:
self._control_target = normal.move_to(
self._control_target, Pose2D(Field.field('length')*0.25,0,0), ball_info, look_ball=True)
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info)
elif referee.referee_id == ref.REFEREE_ID["OUR_TIMEOUT"]:
rospy.logdebug("OUR_TIMEOUT")
# 自チームのタイムアウトではロボットを停止させる
self._control_target, remake_path= normal.stop(self._control_target)
avoid_obstacle = False # 障害物回避しない
elif referee.referee_id == ref.REFEREE_ID["OUR_BALL_PLACEMENT"]:
rospy.logdebug("OUR_BALL_PLACEMENT")
replace_pose = referee.placement_position
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target, avoid_ball = ball_placement.basic_ball_placement(self._control_target, replace_pose, ball_info, robot_info, self._MY_ID, 'atk', [Field.field('length'), Field.field('width')])
elif self._my_role == role.ROLE_ID["ROLE_CENTER_BACK_1"]:
self._control_target, avoid_ball = ball_placement.basic_ball_placement(self._control_target, replace_pose, ball_info, robot_info, self._MY_ID, 'recv', [Field.field('length'), Field.field('width')])
else:
self._control_target, avoid_ball = ball_placement.avoid_ball_place_line(
self._my_pose, ball_info, replace_pose, self._control_target)
elif referee.referee_id == ref.REFEREE_ID["THEIR_KICKOFF_PREPARATION"] \
or referee.referee_id == ref.REFEREE_ID["THEIR_KICKOFF_START"]:
rospy.logdebug("THEIR_KICKOFF")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target = offense.interpose(ball_info,
self._control_target, dist_from_target=0.6)
avoid_ball = True
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info)
elif referee.referee_id == ref.REFEREE_ID["THEIR_PENALTY_PREPARATION"] \
or referee.referee_id == ref.REFEREE_ID["THEIR_PENALTY_START"]:
rospy.logdebug("THEIR_PENALTY")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
else:
self._control_target, remake_path = normal.make_line(
self._my_role, ball_info, self._control_target,
start_x=-Field.field("length")*0.5,
start_y=Field.field("width")*0.4,
add_x=0.4, add_y=0)
elif referee.referee_id == ref.REFEREE_ID["THEIR_DIRECT_FREE"]:
rospy.logdebug("THEIR_DIRECT")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target = offense.interpose(ball_info,
self._control_target, dist_from_target=0.6)
avoid_ball = True
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info,zone_enable=True)
elif referee.referee_id == ref.REFEREE_ID["THEIR_INDIRECT_FREE"]:
rospy.logdebug("THEIR_INDIRECT")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target = offense.interpose(ball_info,
self._control_target, dist_from_target=0.6)
avoid_ball = True
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info, zone_enable=True)
elif referee.referee_id == ref.REFEREE_ID["THEIR_TIMEOUT"]:
rospy.logdebug("THEIR_TIMEOUT")
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
self._control_target = offense.interpose(ball_info,
self._control_target, dist_from_target=0.6)
avoid_ball = True
else:
self._control_target = assign.assign(
self._my_role, ball_info, self._control_target,
self._my_pose, defense_num, robot_info)
elif referee.referee_id == ref.REFEREE_ID["THEIR_BALL_PLACEMENT"]:
rospy.logdebug("THEIR_BALL_PLACEMENT")
replace_pose = referee.placement_position
if self._my_role == role.ROLE_ID["ROLE_GOALIE"]:
self._control_target = goalie.interpose(
ball_info, robot_info, self._control_target)
avoid_obstacle = False # 障害物回避しない
# elif self._my_role == role.ROLE_ID["ROLE_ATTACKER"]:
# self._control_target = offense.interpose(ball_info,
# self._control_target, dist_from_target=0.6)
# avoid_ball = True
else:
self._control_target, avoid_ball = ball_placement.avoid_ball_place_line(
self._my_pose, ball_info, replace_pose, self._control_target,
force_avoid=True)
# self._control_target = assign.assign(
# self._my_role, ball_info, self._control_target,
# self._my_pose, defense_num, robot_info)
# 障害物回避の経路作成
if avoid_obstacle:
self._control_target.path = obstacle_avoidance.add_path(
self._control_target.path, self._my_pose, avoid_ball)
return self._control_target
def test_cns_fmt_dict(self):
for cns, qfeat, sfeat in assign(self.cns_dict, self.qbed, self.sbed, self.qpair_map, self.spair_map):
d = cns_fmt_dict(cns, qfeat, sfeat, self.evalue_dict)
print "dddddddd", d
def test_assign(self): assign(self.cns_dict, self.qbed, self.sbed, self.qpair_map, self.spair_map)
def assign_generators(generators, context, warnings):
for generator in generators:
assign(generator.target, generator.iter, context,
warnings, generator=True)
