def clean_inventory(self, to_tab, force_sync=False, expected_items=None):
'''
Transfer back all items from inventory to stash.
Move any trash item to recycle bin.
'''
# Check we are in foreground, set otherwise
self.is_foreground(set_foreground=True)
# Check if stash is open, open it otherwise
self.check_stash_open(open_otherwise=True)
# Go to the currency stash tab
self.go_to_tab(to_tab, check_stash_open=False)
# Check if the inventory is synchronized
if not inventory.synchronzed or force_sync:
self.sync_inventory()
# Clear clipboard text
self.clear_clipboard_text()
autogui.keyDown('ctrl')
for slot in inventory.iter():
# Get item stack
currency_stack = copy.deepcopy(inventory.get(*slot))
# Request bank diposit
if not bank.deposit(currency_stack):
# bank is full or something went wrong
raise CouldNotClearInventoryException
else:
# Transfer
(point_x, point_y) = inventory.to_screen_position(*slot)
logger.debug(
"Stashing item at ({},{})".format(*slot))
autogui.moveTo(point_x, point_y)
autogui.click()
# Attempt to get item text at current position
autogui.press('c')
# If clipboard is not empty, the item didn't get transfered
if self.get_clipboard_text() != "":
logger.error(
'Could not remove item from inventory at position [{},{}]'.format(*slot))
raise Exception
# Clear item from inventory
inventory.clear(*slot)
autogui.keyUp('ctrl')
# Reset mouse to avoid screen polution
autogui.moveTo(1, 1)
return True
def cancel_trade(self, ensure=False):
''' click the cancel trade button '''
if not self.check_trade_win_open():
logger.warning(
'Attempted to close trade winow but it is not open !')
autogui.moveTo(*config.TRADE_WIN_CANCEL_BUTTON_POS)
autogui.click()
autogui.moveTo(1, 1)
return self.check_trade_win_open()
def accept_trade(self):
''' click the accept trade button '''
if not self.check_trade_win_open():
logger.error(
'Attempted to accept trade but trade window is not open !')
raise Exception
if vh.get_trade_win_accept_status() == "ACCEPTED":
logger.info(
'Attempted to accept trade but trade is already accepted !')
return True
autogui.moveTo(*config.TRADE_WIN_ACCEPT_BUTTON_POS)
autogui.click()
autogui.moveTo(1, 1)
return True
def transfer_currency_inv_trade(self, ensure=False):
'''
Takes a list of positions in the inventory.
The items in that positions will be ctrl + clicked to the trade window.
'''
# Check we are in foreground, set otherwise
self.is_foreground(set_foreground=True)
# Check if stash is open, open it otherwise
if not self.check_trade_win_open():
logger.error(
'Attempted to transfer from inventory to trade but trade window is not open !')
raise Exception
autogui.keyDown('ctrl')
for item in inventory.iter():
pos = inventory.to_screen_position(*item)
autogui.moveTo(*pos)
autogui.click()
autogui.keyUp('ctrl')
autogui.moveTo(1, 1)
def set_currency_rates(self, rates):
''' set the rate for the requested currency '''
# Check we are in foreground, set otherwise
self.is_foreground(set_foreground=True)
# Check if stash is open, open it otherwise
self.check_stash_open(open_otherwise=True)
# Go to the currency stash tab
self.go_to_tab(listing_tab, check_stash_open=False)
for (curr, texts) in rates:
# Find all currency curr and the messages in text to each indiviudal currency
positions = vh.find_currency_in_stash_tab(
[curr], listing_tab)[curr.name]
# Assign a diferent text to each position
for pos, txt in zip(positions, texts):
autogui.moveTo(*pos)
# Double check we found the right currency
autogui.hotkey('ctrl', 'c')
found_curr, _ = self.process_item_description(
self.get_clipboard_text())
if curr != found_curr:
continue
autogui.click(button='right')
retries = 0
pos = None
while not pos and retries < 3:
autogui.press('backspace')
pos = vh.find_set_price_window()
retries += 1
if not pos:
logger.error(
'Could not find price set window for {}'.format(curr.pretty_name))
# reset mouse
autogui.moveTo(1, 1)
# click anywhere else to close the set currency window
# just in case it was left open and couldn't be detected
autogui.click()
continue
set_price_win_x = pos[0][0][0]
set_price_win_y = pos[0][0][1]
set_price_win_w = pos[0][1][0] - set_price_win_x
set_price_win_h = pos[0][1][1] - set_price_win_y
text_field_pos = [set_price_win_x + set_price_win_w * config.PRICE_SET_WINDOW_TEXT_FIELD_REL_POS[0],
set_price_win_y + set_price_win_h * config.PRICE_SET_WINDOW_TEXT_FIELD_REL_POS[1]]
autogui.moveTo(*text_field_pos)
autogui.click()
autogui.typewrite(txt)
autogui.press('enter')
# reset mouse
autogui.moveTo(1, 1)
# click anywhere else to close the set currency window
# just in case it was left open and couldn't be detected
autogui.click()
def transfer_currency_stash_inv(self, curr, ammount, from_tab):
'''
Takes a list of currencies to be transfered into the inventory.
Will take care of ammounts that are not aligned with stack size.
'''
# Check we are in foreground, set otherwise
self.is_foreground(set_foreground=True)
# Check if stash is open, open it otherwise
self.check_stash_open(open_otherwise=True)
# Go to the currency stash tab
self.go_to_tab(from_tab, check_stash_open=False)
# Find the position of this currency in stash
matches = vh.find_currency_in_stash_tab([curr], currency_tab)
matches = matches[curr.name]
# Check if it is possible
if not bank.withdraw(currency.CurrencyStack(curr, ammount)):
raise Exception
t_ammount_left = ammount
for match in matches:
# Read at that position
(found_curr, f_ammount_left) = self.process_item_description(
self.get_text_at_position(match, reset_mouse=False))
if found_curr != curr:
logger.error(
"Find mismatch ! Expected {} found {}".format(curr, found_curr))
raise Exception
# While we didnt finish or we didn't run out of currency here
while (t_ammount_left > 0) and (f_ammount_left > 0):
transfer_size = 0
# Eg 5c left here and need to transfer 6c (stack size is 10)
# Eg 11c left here and need to transfer 16c (stack size is 10) - Wouldn't work
if (f_ammount_left < t_ammount_left) and (f_ammount_left < curr.stack_size):
transfer_size = f_ammount_left
# Eg 50c left here and need to transfer 60c (stack size is 10)
elif t_ammount_left > curr.stack_size:
transfer_size = curr.stack_size
# Eg 50c left here and need to transfer 6c (stack size is 10)
elif t_ammount_left <= curr.stack_size:
transfer_size = t_ammount_left
# The transfer function will asume that if we are transfering less than a full stack
# we will do it 'by hand' and not by control + click. Since it
# This is the case most of the time. There is a corner case where we are transfering
# less than a full stack and exactly the ammount left on the stash. In that case we have to
# ctrl + click.
force_ctrl_click = (transfer_size == f_ammount_left)
(result, new_cell) = inventory.transfer(
currency.CurrencyStack(found_curr, transfer_size), force_ctrl_click=force_ctrl_click)
if not result:
print(result)
print(new_cell)
print(inventory.dump())
print(bank.contents)
raise Exception
# Can ctrl+click to transfer full stack
if transfer_size == curr.stack_size or transfer_size == f_ammount_left:
# Transfer found_ammount of found_curr into inv
autogui.moveTo(*match)
autogui.keyDown('ctrl')
autogui.click()
autogui.keyUp('ctrl')
# Need to split the stack
else:
autogui.moveTo(*match)
autogui.keyDown('shift')
autogui.click()
autogui.keyUp('shift')
time.sleep(0.2)
# Split currency
autogui.typewrite(str(transfer_size))
time.sleep(0.2)
# Confirm
autogui.press('enter')
# Find next empty slot
autogui.moveTo(
*inventory.to_screen_position(*new_cell))
# Drop the currency
autogui.click()
# Update remaining
t_ammount_left -= transfer_size
if t_ammount_left != 0:
raise Exception