def test_adj_default():
board_adj_default_cross = Board(10, 10)
p1 = Pos(5, 5)
p2 = Pos(6, 5)
p3 = Pos(4, 4)
# cross:
assert board_adj_default_cross.is_adjacent(p1, p2) == True
# diag :
assert board_adj_default_cross.is_adjacent(p1, p3) == False
set_default_adjacency(AdjacencyEvaluatorCross)
board_adj_default_cross_2 = Board(10, 10)
# cross:
assert board_adj_default_cross_2.is_adjacent(p1, p2) == True
# diag :
assert board_adj_default_cross_2.is_adjacent(p1, p3) == False
set_default_adjacency(AdjacencyEvaluatorCrossDiag)
board_adj_default_cross_diag = Board(10, 10)
# cross:
assert board_adj_default_cross_diag.is_adjacent(p1, p2) == True
# diag :
assert board_adj_default_cross_diag.is_adjacent(p1, p3) == True
def test_getitem_fail():
board = Board(5, 14)
failed_at_failing = False
try:
a = board[0, 14]
failed_at_failing = True
except BoardIndexError as e:
print(e)
try:
p = Pos(5, 0)
a = board[p]
failed_at_failing = True
except BoardIndexError as e:
print(e)
try:
a = board[0, -15]
failed_at_failing = True
except BoardIndexError as e:
print(e)
try:
p = Pos(-6, 0)
a = board[p]
failed_at_failing = True
except BoardIndexError as e:
print(e)
assert failed_at_failing == False
def test_permute_simple():
board = Board(5, 3)
setting_data = ("ABCDE", "FGHIJ", "KLMNO")
board.set_data_from_string(setting_data)
tile_with_c = board[2, 0]
tile_with_n = board[3, 2]
board.circular_permute_tiles([Pos(2, 0), Pos(3, 2)])
print(board.render())
assert tile_with_c.x == 3
assert tile_with_c.y == 2
assert tile_with_c.data == "C"
assert board[3, 2].data == "C"
assert tile_with_n.x == 2
assert tile_with_n.y == 0
assert tile_with_n.data == "N"
assert board[2, 0].data == "N"
render_result = """
ABNDE
FGHIJ
KLMCO
"""
assert strip_multiline(board.render()) == strip_multiline(render_result)
def circular_permute_tiles(self, positions):
"""
positions est un itérable.
"""
made_first_iteration = False
for pos in positions:
if made_first_iteration:
cur_pos = pos
cur_tile = self._tiles[cur_pos.y][cur_pos.x]
cur_tile.pos = Pos(prev_pos)
cur_tile.x = prev_pos.x
cur_tile.y = prev_pos.y
self._tiles[prev_pos.y][prev_pos.x] = cur_tile
prev_pos = cur_pos
else:
first_pos = pos
first_tile = self._tiles[first_pos.y][first_pos.x]
prev_pos = first_pos
made_first_iteration = True
first_tile.pos = Pos(pos)
first_tile.x = pos.x
first_tile.y = pos.y
self._tiles[pos.y][pos.x] = first_tile
def generate_training_stats(years_in_majors, position):
exp_term = 0
if years_in_majors == 0: pass
else: exp_term = 16.0 / (1 + 60.0 / (years_in_majors * years_in_majors))
params1 = parameters()
params2 = parameters()
stat1 = redistribute(int(rand_skill_seed(params1[0], params1[1])))
stat2 = redistribute(int(rand_skill_seed(params2[0], params2[1])))
total = stat1 + stat2
hitting, fielding, pitching = 0, 0, 0
if Pos.is_infield(position):
hitting = int(round(total * 1.0 / 2.0))
fielding = int(round(total * 1.0 / 2.0))
pitching = MIN_VALUE
elif Pos.is_outfield(position):
hitting = int(round(total * 3.0 / 5.0))
fielding = int(round(total * 2.0 / 5.0))
pitching = MIN_VALUE
elif Pos.is_pitcher(position):
index = int(total * 0.02999)
params = distro_set[index]
pitching = redistribute(int(rand_skill_seed(params[0], params[1])))
fielding = redistribute(
int(rand_skill_seed(distro_set[3][0], distro_set[3][1])))
hitting = redistribute(
int(rand_skill_seed(distro_set[1][0], distro_set[1][1])))
else:
hitting = int(round(total * 4.0 / 5.0))
fielding = int(round(total * 1.0 / 5.0))
pitching = MIN_VALUE
return redistribute(hitting), redistribute(fielding), redistribute(
pitching)
def generate_player(): r = random(); name = '' if r < 0.5: name = rand_male_name() + ' ' + rand_surname() else: name = rand_female_name() + ' ' + rand_surname() pos = Pos.rand_position() weight, height = 0, 0 if pos == Pos.CATCHER: weight, height = catcher_height_weight() elif Pos.is_infield(pos): weight, height = infield_height_weight() elif Pos.is_outfield(pos): weight, height = outfield_height_weight() else: weight, height = pitcher_height_weight() birthdate = generate_birthdate(start_date.year) bats, throws = generate_bats_throws() years_in_majors = int( gauss(years_old(birthdate, start_date) - 23, 2) ) if years_in_majors < 0: years_in_majors = 0 caps = Capabilities.generate_all_caps(height, weight, birthdate, start_date, years_in_majors, pos) pos_caps = PositionCapabilities.generate_pos_caps(pos) contract = Contract(get_FA(), 0, 0) personality = Personality.rand_personality() return Player(name, pos, height, weight, birthdate, bats, throws, caps, pos_caps, contract, personality)
def test_eq():
p_1 = Pos(34, 78)
p_2 = Pos(35, 78)
p_3 = Pos(34, 77)
p_4 = Pos(34, 78)
p_5 = Pos(0, 0)
assert p_1 == p_1
assert p_1 != p_2
assert p_1 != p_3
assert p_1 == p_4
assert p_1 != p_5
def __init__(self,
board,
pos_start,
pos_end,
pass_through_condition=propag_cond_default):
# FUTURE : pathfinding avec tous les shortest paths possibles.
# pathfinding avec tous les paths possibles
super().__init__(board)
self.pass_through_condition = pass_through_condition
pos_start = Pos(pos_start)
pos_end = Pos(pos_end)
self.pos_start = pos_start
self.pos_end = pos_end
iter_propag = BoardIteratorPropagation(self.board, self.pos_start,
pass_through_condition)
try:
while pos_end not in iter_propag.propagated_poss:
next(iter_propag)
except StopIteration:
self.path = None
return
propagated_poss = iter_propag.propagated_poss
# Et maintenant, on parcourt la propagation à l'envers,
# pour retrouver le chemin.
pos_cur = pos_end
dist_cur = propagated_poss[pos_cur]
self.path = [pos_cur]
while pos_cur != pos_start:
advanced = False
for adj_pos in self.board.adjacency.adjacent_positions(pos_cur):
if (propagated_poss.get(adj_pos, -2)
== dist_cur - 1) and pass_through_condition(
self.board[adj_pos], self.board[pos_cur]):
pos_cur = adj_pos
dist_cur -= 1
self.path.append(pos_cur)
advanced = True
break
if not advanced:
raise Exception(
"No adj pos with dist-1. Not supposed to happen")
def __next__(self):
while self.nb_sub_coord_to_skip:
self._apply_skip_sub_coord()
self.nb_sub_coord_to_skip -= 1
self._update_col_line_modification(True)
try:
val_coord_main = next(self.iter_main)
must_change_sub = False
except StopIteration:
# Faut repartir à la "ligne" suivante.
must_change_sub = True
self._update_col_line_modification(must_change_sub)
if must_change_sub:
self._apply_skip_sub_coord()
val_coord_main = next(self.iter_main)
self._update_col_line_modification(True)
if self.id_coord_main == Coord.X:
x = val_coord_main
y = self.val_coord_sub
else:
x = self.val_coord_sub
y = val_coord_main
new_pos = Pos(x, y)
self._update_indicators(new_pos)
return self.board.get_tile(self.current_pos)
def test_adj_diag():
simple_board = Board(10, 10, class_adjacency=AdjacencyEvaluatorCrossDiag)
p1 = Pos(5, 5)
p2 = Pos(6, 5)
p3 = Pos(4, 4)
# same :
assert simple_board.is_adjacent(p1, p1) == False
# cross :
assert simple_board.is_adjacent(p1, p2) == True
# diag :
assert simple_board.is_adjacent(p1, p3) == True
# none :
assert simple_board.is_adjacent(p2, p3) == False
def test_permute_column():
board = Board(5, 7)
setting_data = ("ABCDE", "FGHIJ", "KLMNO", "PQRST", "UVWXY", "01234",
"56789")
board.set_data_from_string(setting_data)
pos_to_permute = [Pos(tile.x, tile.y) for tile in board[2, :]]
assert len(pos_to_permute) == 7
board.circular_permute_tiles(pos_to_permute)
print(board.render())
render_result = """
ABHDE
FGMIJ
KLRNO
PQWST
UV2XY
01734
56C89
"""
assert strip_multiline(board.render()) == strip_multiline(render_result)
# Pour vérifier que la fonction de permutation ne vide pas la liste.
# Ça le faisait avant, et c'était mal.
assert len(pos_to_permute) == 7
def test_replace_simple():
board = Board(5, 3)
setting_data = ("ABCDE", "FGHIJ", "KLMNO")
board.set_data_from_string(setting_data)
new_tile = Tile()
new_tile.data = "Z"
board.replace_tile(new_tile, Pos(3, 1))
print(board.render())
assert new_tile.x == 3
assert new_tile.y == 1
assert board[3, 1].data == "Z"
render_result = """
ABCDE
FGHZJ
KLMNO
"""
assert strip_multiline(board.render()) == strip_multiline(render_result)
print(board.render())
def __init__(self):
board = []
for x in range(0, 8):
row = [Pos(x, y) for y in range(0, 8)]
board.append(tuple(row))
self.board = tuple(board)
def test_from_obj():
class Whatever:
def __init__(self):
self.x = 23.9
self.y = 45.8
p = Pos(Whatever())
assert p.x == 23 and p.y == 45
def getPlayerPositions(self):
poss = []
for r in range(1, 6):
for c in range(1, 6):
pos = Pos(r, c)
if self.board.hasPiece(pos, self.playerPiece):
poss.append(pos)
return poss
def __init__(self, board, pos_start, propag_condition=propag_cond_default):
# TODO : avec plusieurs pos_start.
super().__init__(board)
self.propag_condition = propag_condition
# Dict
# - clé : la pos propagée.
# - valeur : la distance depuis la pos de départ jusqu'à la pos propagée.
self.propagated_poss = {}
# liste de tuple de 2 éléments : la distance et la pos propagée.
self.to_propagate_poss = [(0, Pos(pos_start))]
def test_getitem_pos():
board = Board(3, 3)
p = Pos(1, 0)
board[p].data = "|"
board[Pos(0, -2)].data = "-"
board[Pos(-2, 1)].data = "*"
board[Pos(-1, -2)].data = "~"
board[{"x": 1, "y": 2}].data = "I"
render_result = """
.|.
-*~
.I.
"""
print(board.render())
assert strip_multiline(board.render()) == strip_multiline(render_result)
def __next__(self):
self.current_posis_index += 1
if self.current_posis_index >= len(self.posis):
raise StopIteration
new_pos = Pos(self.posis[self.current_posis_index])
self._update_indicators(new_pos)
return self.board.get_tile(self.current_pos)
def __getitem__(self, args):
# FUTURE : on a le droit de faire du *args, **kwargs avec getitem ?
# Et ça donne quoi si on le fait ? À tester.
if not args:
return BoardIteratorRect(self)
try:
pos = Pos(args)
except ValueError:
pos = None
if pos is not None:
# Mode un seul élément
return self._get_tile(pos.x, pos.y)
slice_x = None
slice_y = None
id_coord_main = Coord.X
try:
iter_on_args = iter(args)
slice_x = next(iter_on_args)
slice_y = next(iter_on_args)
id_coord_main = next(iter_on_args)
except TypeError:
slice_x = args
except StopIteration:
pass
if (slice_x is None or slice_y is None or isinstance(slice_x, slice)
or isinstance(slice_y, slice)):
# Mode itération
if slice_x is None:
slice_x = slice(None, None, None)
if isinstance(slice_x, int):
slice_x = slice(slice_x, slice_x + 1, None)
if slice_y is None:
slice_y = slice(None, None, None)
if isinstance(slice_y, int):
slice_y = slice(slice_y, slice_y + 1, None)
dict_coord_from_str = {"X": Coord.X, "Y": Coord.Y}
if isinstance(id_coord_main, str):
id_coord_main = id_coord_main.upper()
if id_coord_main in dict_coord_from_str:
id_coord_main = dict_coord_from_str[id_coord_main]
return BoardIteratorRect(self, slice_x, slice_y, id_coord_main)
# Mode fail
raise Exception("TODO fail get item" + "".join(args))
def __init__(self, x=None, y=None, board_owner=None):
# TODO : il faut accepter le même bazar de param que pour l'objet Pos. Ou pas.
self.x = x
self.y = y
# TODO : est-ce qu'on autorise des tiles sans coord, qui "flotte un peu dans les airs", ou pas ?
try:
self.pos = Pos(x, y)
except:
self.pos = None
self.board_owner = board_owner
self.data = "."
self.mobile_items = []
def generate_pos_caps(cls, player_pos):
skills = {}
skills[Pos.CATCHER] = cls.redistribute(
int(
gauss(
cls.skill_levels[Pos.compare_pos(player_pos, Pos.CATCHER)],
cls.sigma)))
skills[Pos.FIRST_BASE] = cls.redistribute(
int(
gauss(
cls.skill_levels[Pos.compare_pos(player_pos,
Pos.FIRST_BASE)],
cls.sigma)))
skills[Pos.SECOND_BASE] = cls.redistribute(
int(
gauss(
cls.skill_levels[Pos.compare_pos(player_pos,
Pos.SECOND_BASE)],
cls.sigma)))
skills[Pos.THIRD_BASE] = cls.redistribute(
int(
gauss(
cls.skill_levels[Pos.compare_pos(player_pos,
Pos.THIRD_BASE)],
cls.sigma)))
skills[Pos.SHORTSTOP] = cls.redistribute(
int(
gauss(
cls.skill_levels[Pos.compare_pos(player_pos,
Pos.SHORTSTOP)],
cls.sigma)))
skills[Pos.LEFT_FIELD] = cls.redistribute(
int(
gauss(
cls.skill_levels[Pos.compare_pos(player_pos,
Pos.LEFT_FIELD)],
cls.sigma)))
skills[Pos.CENTER_FIELD] = cls.redistribute(
int(
gauss(
cls.skill_levels[Pos.compare_pos(player_pos,
Pos.CENTER_FIELD)],
cls.sigma)))
skills[Pos.RIGHT_FIELD] = cls.redistribute(
int(
gauss(
cls.skill_levels[Pos.compare_pos(player_pos,
Pos.RIGHT_FIELD)],
cls.sigma)))
return PositionCapabilities(skills)
def adjacent_positions(self, pos):
# TODO : faudrait renvoyer des tiles ou des positions ?
# Il est conseillé de mettre dans le même ordre que l'ordre des Direction.
# C'est à dire dans le sens des aiguilles d'une montre.
# (Mais ce n'est pas tout le temps possible avec des fonctions d'adjacences tordues)
offsets = [(0, -1), (+1, 0), (0, +1), (-1, 0)]
for offset in offsets:
x = pos.x + offset[0]
y = pos.y + offset[1]
# TODO : le check de inbounds devrait être dans la classe board, tellement c'est un truc basique.
if (0 <= x < self.board.w) and (0 <= y < self.board.h):
yield Pos(x, y)
def movePlayerPiece(self):
inp = input("Player " + self.playerPiece +
" move piece (position + direction, e.g. 'A1 NE'): ")
try:
pos = Pos.parse(inp[0:2])
dir = inp[3:].rstrip()
self.board.move(pos, self.playerPiece, dir)
except TypeError as e1:
print(str(e1))
self.movePlayerPiece()
except MoveError as e2:
print(str(e2))
self.movePlayerPiece()
def move(self, board_owner=None, tile_owner=None, z_index=None, *args, **kwargs):
"""
Param prioritaire : tile_owner.
Sinon : les autres params.
"""
# FUTURE : j'ai plein de fonctions qui crée une pos à partir de args et kwargs.
# Y'aurait peut-être moyen de le factoriser avec un décorateur.
if self.tile_owner is not None:
# --- suppression du mobitem de la tile où il était avant ---
# Si self n'est pas mobile_items, ça va raiser une exception.
# C'est ce qu'on veut, parce que not supposed to happen.
index_myself = self.tile_owner.mobile_items.index(self)
del self.tile_owner.mobile_items[index_myself]
# --- définition éventuelle de board_owner, à partir de l'actuel board_owner ---
if board_owner is None:
board_owner = self.tile_owner.board_owner
# --- définition éventuelle de board_owner, à partir du nouveau tile_owner ---
if tile_owner is not None:
board_owner = tile_owner.board_owner
# --- définition éventuelle de tile_owner, à partir de board_owner et des param de pos ---
if tile_owner is None and board_owner is not None:
try:
pos = Pos(*args, **kwargs)
tile_owner = board_owner[pos]
except:
tile_owner = None
# --- Enregistrement dans le nouveau tile_owner, si défini ---
if tile_owner is not None:
self.tile_owner = tile_owner
if z_index is None:
tile_owner.mobile_items.append(self)
else:
tile_owner.mobile_items.insert(z_index, self)
def test_directions_computing():
center = Pos(4, 7)
assert compute_direction(center, Pos(4, 5)) == Dir.UP
assert compute_direction(center, Pos(5, 5)) == Dir.UP_RIGHT
assert compute_direction(center, Pos(6, 5)) == Dir.UP_RIGHT
assert compute_direction(center, Pos(6, 6)) == Dir.UP_RIGHT
assert compute_direction(center, Pos(6, 7)) == Dir.RIGHT
assert compute_direction(center, Pos(6, 8)) == Dir.DOWN_RIGHT
assert compute_direction(center, Pos(6, 9)) == Dir.DOWN_RIGHT
assert compute_direction(center, Pos(5, 9)) == Dir.DOWN_RIGHT
assert compute_direction(center, Pos(4, 9)) == Dir.DOWN
assert compute_direction(center, Pos(3, 9)) == Dir.DOWN_LEFT
assert compute_direction(center, Pos(2, 9)) == Dir.DOWN_LEFT
assert compute_direction(center, Pos(2, 8)) == Dir.DOWN_LEFT
assert compute_direction(center, Pos(2, 7)) == Dir.LEFT
assert compute_direction(center, Pos(2, 6)) == Dir.UP_LEFT
assert compute_direction(center, Pos(2, 5)) == Dir.UP_LEFT
assert compute_direction(center, Pos(3, 5)) == Dir.UP_LEFT
def test_coord_outing():
p = Pos(23.9, 45.8)
t = p.as_tuple()
assert t[0] == 23 and t[1] == 45
d = p.as_dict()
assert d["x"] == 23 and d["y"] == 45
def find_drugs(code, batchno, num, index):
print(code, num, index)
global ishave_num
# 添加明细
win32api.SetCursorPos([Pos.TJMX_X(), Pos.TJMX_Y()]) # 设置鼠标位置
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0) # 左键点击
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 0, 0, 0, 0) # 左键点击
time.sleep(1)
pass
# 药品编码搜索框
win32api.SetCursorPos([Pos.YPBMSS_X(), Pos.YPBMSS_Y()]) # 设置鼠标位置
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0) # 左键点击
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 0, 0, 0, 0) # 左键点击
time.sleep(0.1)
pass
# 清空输入框原有参数
win32api.keybd_event(win32con.VK_LCONTROL, 0, 0, 0)
win32api.keybd_event(KEYJSON['A'], 0, 0, 0)
win32api.keybd_event(KEYJSON['A'], 0, win32con.KEYEVENTF_KEYUP, 0)
win32api.keybd_event(win32con.VK_LCONTROL, 0, win32con.KEYEVENTF_KEYUP,
0)
pass
# 把code转list，一个一个 输入进去
code_list = list(code)
for value in code_list:
if value in 'ABCDEFGHIJKLMNOPQRSTUVWXYZ':
win32api.keybd_event(win32con.VK_SHIFT, 0, 0, 0)
pass
win32api.keybd_event(KEYJSON[value], 0, 0, 0)
win32api.keybd_event(KEYJSON[value], 0, win32con.KEYEVENTF_KEYUP,
0)
if value in 'ABCDEFGHIJKLMNOPQRSTUVWXYZ':
win32api.keybd_event(win32con.VK_SHIFT, 0,
win32con.KEYEVENTF_KEYUP, 0)
pass
pass
# 点击搜索按钮
win32api.SetCursorPos([Pos.YPBMSSBTN_X(), Pos.YPBMSSBTN_Y()]) # 设置鼠标位置
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0) # 左键点击
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 0, 0, 0, 0) # 左键点击
time.sleep(1)
pass
# 全选订单明细
win32api.SetCursorPos([Pos.YPBMSSCHECK_X(),
Pos.YPBMSSCHECK_Y()]) # 设置鼠标位置
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0) # 左键点击
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 0, 0, 0, 0) # 左键点击
time.sleep(1)
pass
img = ImageGrab.grab(Pos.SCREEN_IMAGE_POSITION())
px = img.load()
for x in range(img.width):
for y in range(img.height):
if (str(px[x, y]) != str((255, 255, 255))):
ishave_num = True
break
pass
pass
# 确定选中
win32api.SetCursorPos([Pos.SUREBTN_X(), Pos.SUREBTN_Y()]) # 设置鼠标位置
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0) # 左键点击
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 0, 0, 0, 0) # 左键点击
time.sleep(1)
pass
if ishave_num:
# 初始化tab焦点
win32api.SetCursorPos([Pos.FIRSTDATA_X(),
Pos.FIRSTDATA_Y()]) # 设置鼠标位置
DOUBLE_CLICK()
time.sleep(0.5)
count = 0
# while count < 5 * index + 1:
while count < 5 * index + 2: # 2.0
TAB_KEYUP()
count += 1
pass
time.sleep(0.5)
# 把code转list，一个一个 输入进去
batch_list = list(batchno)
print(batch_list)
for value in batch_list:
if value in 'ABCDEFGHIJKLMNOPQRSTUVWXYZ':
win32api.keybd_event(win32con.VK_SHIFT, 0, 0, 0)
pass
if value == '-':
print(KEYJSON[value])
win32api.keybd_event(KEYJSON[value], 0, 0, 0)
win32api.keybd_event(KEYJSON[value], 0,
win32con.KEYEVENTF_KEYUP, 0)
if value in 'ABCDEFGHIJKLMNOPQRSTUVWXYZ':
win32api.keybd_event(win32con.VK_SHIFT, 0,
win32con.KEYEVENTF_KEYUP, 0)
pass
pass
TAB_KEYUP()
num_list = list(str(int(num)))
for value in num_list:
win32api.keybd_event(KEYJSON[value], 0, 0, 0)
win32api.keybd_event(KEYJSON[value], 0,
win32con.KEYEVENTF_KEYUP, 0)
pass
time.sleep(0.5)
TAB_KEYUP()
print(code, num)
pass
def test_push_cols_lines():
"""
Test de déplacement de toutes les tiles d'une ligne ou d'une colonne,
en ajoutant une nouvelle tile qui va pousser les autres.
Comme dans le jeu de plateau 'Labyrinthe', et dans le challenge CodinGame 'Xmas Rush'
"""
# PUSH 3 RIGHT
board = Board(5, 7)
setting_data = ("ABCDE", "FGHIJ", "KLMNO", "PQRST", "UVWXY", "01234",
"56789")
board.set_data_from_string(setting_data)
added_tile = Tile()
added_tile.data = "#"
pos_to_permute = [Pos(tile.x, tile.y) for tile in board[::-1, 3]]
board.circular_permute_tiles(pos_to_permute)
removed_tile = board[0, 3]
board.replace_tile(added_tile, Pos(0, 3))
print(board.render())
assert removed_tile.data == "T"
render_result = """
ABCDE
FGHIJ
KLMNO
#PQRS
UVWXY
01234
56789
"""
assert strip_multiline(board.render()) == strip_multiline(render_result)
print("")
# PUSH 0 LEFT
board = Board(5, 7)
setting_data = ("ABCDE", "FGHIJ", "KLMNO", "PQRST", "UVWXY", "01234",
"56789")
board.set_data_from_string(setting_data)
added_tile = Tile()
added_tile.data = "#"
pos_to_permute = [Pos(tile.x, tile.y) for tile in board[:, 0]]
board.circular_permute_tiles(pos_to_permute)
removed_tile = board[4, 0]
board.replace_tile(added_tile, Pos(4, 0))
print(board.render())
assert removed_tile.data == "A"
render_result = """
BCDE#
FGHIJ
KLMNO
PQRST
UVWXY
01234
56789
"""
assert strip_multiline(board.render()) == strip_multiline(render_result)
print("")
# PUSH 4 DOWN
board = Board(5, 7)
setting_data = ("ABCDE", "FGHIJ", "KLMNO", "PQRST", "UVWXY", "01234",
"56789")
board.set_data_from_string(setting_data)
added_tile = Tile()
added_tile.data = "#"
pos_to_permute = [Pos(tile.x, tile.y) for tile in board[4, ::-1]]
board.circular_permute_tiles(pos_to_permute)
removed_tile = board[4, 0]
board.replace_tile(added_tile, Pos(4, 0))
print(board.render())
assert removed_tile.data == "9"
render_result = """
ABCD#
FGHIE
KLMNJ
PQRSO
UVWXT
0123Y
56784
"""
assert strip_multiline(board.render()) == strip_multiline(render_result)
print("")
# PUSH 1 UP
board = Board(5, 7)
setting_data = ("ABCDE", "FGHIJ", "KLMNO", "PQRST", "UVWXY", "01234",
"56789")
board.set_data_from_string(setting_data)
added_tile = Tile()
added_tile.data = "#"
pos_to_permute = [Pos(tile.x, tile.y) for tile in board[1, :]]
board.circular_permute_tiles(pos_to_permute)
removed_tile = board[1, board.h - 1]
board.replace_tile(added_tile, Pos(1, board.h - 1))
print(board.render())
assert removed_tile.data == "B"
render_result = """
AGCDE
FLHIJ
KQMNO
PVRST
U1WXY
06234
5#789
"""
assert strip_multiline(board.render()) == strip_multiline(render_result)
print("")
def do_shopcart(txt=''):
global ishave_num
file_arr = txt.split(' || ')
txt_arr = []
for item in file_arr:
txt_arr.extend(item.split('\n'))
pass
txt_data = []
print('do_shopcart')
try:
# txt = open(file_path)
for item in txt_arr:
item_text = item.split('&')
if not (len(item_text) == 1 and len(item_text[0]) == 0):
txt_data.append(item_text)
pass
nrows = len(txt_data)
pass
except Exception:
return Exception
print('txt_data', txt_data)
i = 0
drug_arr = []
batch_arr = []
num_arr = []
drug_code_index = 0 # 医院药品编码的位置索引
batch_no_index = 1 # 批号
num_index = 2 # 药品数量的位置索引
while i < nrows:
row = txt_data[i]
print('row', row)
if len(row) >= drug_code_index and len(
row) >= batch_no_index and len(row) >= num_index and len(
str(row[drug_code_index])) > 0 and len(
str(row[drug_code_index])
) > 1 and len(str(row[batch_no_index])) > 0 and len(
str(row[num_index])) > 0 and row[num_index].isdigit():
drug_arr.append(str(row[drug_code_index]))
batch_arr.append(str(row[batch_no_index]))
num_arr.append(str(row[num_index]))
pass
i += 1
pass
print('drug_arr', drug_arr)
print('batch_arr', batch_arr)
print('num_arr', num_arr)
windows = {} # 所有的窗口
win_id = 0 # 准备要操作的窗口id
def find_window_callback(id, key):
# 获取窗口时候的回调
if win32gui.GetWindowText(id).strip():
_item = []
_item.append(hex(id))
_item.append(win32gui.GetClassName(id))
_item.append(win32gui.GetWindowText(id))
if key == 'windows':
windows[id] = _item
pass
pass
# 获取所有打开的窗口
win32gui.EnumWindows(find_window_callback, 'windows')
shell.SendKeys('%')
ishave_trade = False
# 遍历一下，取到当前的
for item in windows:
if '东莞交易' in windows[item][2]:
ishave_trade = True
win_id = item
# 最大化
win32api.PostMessage(win_id, win32con.WM_SYSCOMMAND,
win32con.SC_MAXIMIZE, 0)
win32gui.SetForegroundWindow(win_id)
time.sleep(1)
Pos.now_hWnd = win32gui.WindowFromPoint(
(int(Pos.SCREEN_X / 2), int(Pos.SCREEN_Y / 2)))
(Pos.now_hWnd_X1, Pos.now_hWnd_Y1, Pos.now_hWnd_X2,
Pos.now_hWnd_Y2) = win32gui.GetWindowRect(Pos.now_hWnd)
break
pass
if ishave_trade == False:
win32api.MessageBox(0, '没有找到交易平台', '错误', win32con.MB_SYSTEMMODAL)
# os._exit(0)
# sys.exit()
return False
print('-------------------------------------------------')
print('-------------------------------------------------')
# 判断当前是否在自动入库页面
isIn = True
img = ImageGrab.grab(Pos.TJMX_SCREEN_POS())
px = img.load()
for x in range(img.width):
for y in range(img.height):
print(str(px[x, y]))
if (str(px[x, y]) != str((8, 104, 207))):
isIn = False
break
pass
pass
if isIn == False:
win32api.MessageBox(0, '当前不在自动入库模块', '错误', win32con.MB_SYSTEMMODAL)
return False
def find_drugs(code, batchno, num, index):
print(code, num, index)
global ishave_num
# 添加明细
win32api.SetCursorPos([Pos.TJMX_X(), Pos.TJMX_Y()]) # 设置鼠标位置
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0) # 左键点击
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 0, 0, 0, 0) # 左键点击
time.sleep(1)
pass
# 药品编码搜索框
win32api.SetCursorPos([Pos.YPBMSS_X(), Pos.YPBMSS_Y()]) # 设置鼠标位置
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0) # 左键点击
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 0, 0, 0, 0) # 左键点击
time.sleep(0.1)
pass
# 清空输入框原有参数
win32api.keybd_event(win32con.VK_LCONTROL, 0, 0, 0)
win32api.keybd_event(KEYJSON['A'], 0, 0, 0)
win32api.keybd_event(KEYJSON['A'], 0, win32con.KEYEVENTF_KEYUP, 0)
win32api.keybd_event(win32con.VK_LCONTROL, 0, win32con.KEYEVENTF_KEYUP,
0)
pass
# 把code转list，一个一个 输入进去
code_list = list(code)
for value in code_list:
if value in 'ABCDEFGHIJKLMNOPQRSTUVWXYZ':
win32api.keybd_event(win32con.VK_SHIFT, 0, 0, 0)
pass
win32api.keybd_event(KEYJSON[value], 0, 0, 0)
win32api.keybd_event(KEYJSON[value], 0, win32con.KEYEVENTF_KEYUP,
0)
if value in 'ABCDEFGHIJKLMNOPQRSTUVWXYZ':
win32api.keybd_event(win32con.VK_SHIFT, 0,
win32con.KEYEVENTF_KEYUP, 0)
pass
pass
# 点击搜索按钮
win32api.SetCursorPos([Pos.YPBMSSBTN_X(), Pos.YPBMSSBTN_Y()]) # 设置鼠标位置
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0) # 左键点击
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 0, 0, 0, 0) # 左键点击
time.sleep(1)
pass
# 全选订单明细
win32api.SetCursorPos([Pos.YPBMSSCHECK_X(),
Pos.YPBMSSCHECK_Y()]) # 设置鼠标位置
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0) # 左键点击
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 0, 0, 0, 0) # 左键点击
time.sleep(1)
pass
img = ImageGrab.grab(Pos.SCREEN_IMAGE_POSITION())
px = img.load()
for x in range(img.width):
for y in range(img.height):
if (str(px[x, y]) != str((255, 255, 255))):
ishave_num = True
break
pass
pass
# 确定选中
win32api.SetCursorPos([Pos.SUREBTN_X(), Pos.SUREBTN_Y()]) # 设置鼠标位置
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0) # 左键点击
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 0, 0, 0, 0) # 左键点击
time.sleep(1)
pass
if ishave_num:
# 初始化tab焦点
win32api.SetCursorPos([Pos.FIRSTDATA_X(),
Pos.FIRSTDATA_Y()]) # 设置鼠标位置
DOUBLE_CLICK()
time.sleep(0.5)
count = 0
# while count < 5 * index + 1:
while count < 5 * index + 2: # 2.0
TAB_KEYUP()
count += 1
pass
time.sleep(0.5)
# 把code转list，一个一个 输入进去
batch_list = list(batchno)
print(batch_list)
for value in batch_list:
if value in 'ABCDEFGHIJKLMNOPQRSTUVWXYZ':
win32api.keybd_event(win32con.VK_SHIFT, 0, 0, 0)
pass
if value == '-':
print(KEYJSON[value])
win32api.keybd_event(KEYJSON[value], 0, 0, 0)
win32api.keybd_event(KEYJSON[value], 0,
win32con.KEYEVENTF_KEYUP, 0)
if value in 'ABCDEFGHIJKLMNOPQRSTUVWXYZ':
win32api.keybd_event(win32con.VK_SHIFT, 0,
win32con.KEYEVENTF_KEYUP, 0)
pass
pass
TAB_KEYUP()
num_list = list(str(int(num)))
for value in num_list:
win32api.keybd_event(KEYJSON[value], 0, 0, 0)
win32api.keybd_event(KEYJSON[value], 0,
win32con.KEYEVENTF_KEYUP, 0)
pass
time.sleep(0.5)
TAB_KEYUP()
print(code, num)
pass
i = 0
table_index = 0
while i < len(drug_arr):
ishave_num = False
print('num_arr[i]', num_arr[i])
find_drugs(drug_arr[i], batch_arr[i], float(num_arr[i]), table_index)
if ishave_num:
table_index += 1
i += 1
pass
win32api.MessageBox(0, '录入完成', '成功', win32con.MB_SYSTEMMODAL)
# txt = open(file_path, 'w')
# txt.write('')
# txt.close()
return True
def test_str():
p = Pos(23.9, 45.8)
assert str(p) == "<Pos 23, 45 >"
