def update_stock_prices_to_trend(self, api):
''' This function updates all prices in the user's stock to TREND. '''
uploadable_json = []
if os.path.isfile(PRICE_CHANGES_FILE):
if PyMkmHelper.prompt_bool(
"Found existing changes. Upload [y] or discard [n]?"
) == True:
with open(PRICE_CHANGES_FILE, 'r') as changes:
uploadable_json = json.load(changes)
else:
os.remove(PRICE_CHANGES_FILE)
self.update_stock_prices_to_trend(api=self.api)
else:
uploadable_json = self.calculate_new_prices_for_stock(api=self.api)
if len(uploadable_json) > 0:
self.display_price_changes_table(uploadable_json)
if PyMkmHelper.prompt_bool(
"Do you want to update these prices?") == True:
# Update articles on MKM
api.set_stock(uploadable_json)
print('Prices updated.')
else:
with open(PRICE_CHANGES_FILE, 'w') as outfile:
json.dump(uploadable_json, outfile)
print('Prices not updated. Changes saved.')
else:
print('No prices to update.')
class TestPyMkmHelperFunctions(unittest.TestCase):
def setUp(self):
self.helper = PyMkmHelper()
def test_calculate_average(self):
table = [['Yxskaft', 'SE', 'NM', 1, 1.21],
['Frazze11', 'SE', 'NM', 3, 1.3],
['andli826', 'SE', 'NM', 2, 1.82]]
self.assertEqual(self.helper.calculate_average(table, 3, 4), 1.46)
def test_calculate_median(self):
table = [['Yxskaft', 'SE', 'NM', 1, 1.21],
['Frazze11', 'SE', 'NM', 3, 1.3],
['andli826', 'SE', 'NM', 2, 1.82]]
self.assertEqual(self.helper.calculate_median(table, 3, 4), 1.3)
self.assertEqual(self.helper.calculate_average(table, 3, 4), 1.46)
def test_calculate_lowest(self):
table = [['Yxskaft', 'SE', 'NM', 1, 1.21],
['Frazze11', 'SE', 'NM', 3, 1.3],
['andli826', 'SE', 'NM', 2, 1.82]]
self.assertEqual(self.helper.calculate_lowest(table, 4), 1.21)
def test_round_up_to_limit(self):
self.assertEqual(self.helper.round_up_to_limit(0.25, 0.99), 1)
self.assertEqual(self.helper.round_up_to_limit(0.25, 0), 0)
self.assertEqual(self.helper.round_up_to_limit(0.25, 0.1), 0.25)
self.assertEqual(self.helper.round_up_to_limit(0.1, 0.99), 1)
self.assertEqual(self.helper.round_up_to_limit(0.01, 0.011), 0.02)
self.assertEqual(self.helper.round_up_to_limit(0.01, 1), 1)
self.assertEqual(self.helper.round_up_to_limit(1, 0.1), 1)
def test_round_down_to_limit(self):
self.assertEqual(self.helper.round_down_to_limit(0.25, 0.99), 0.75)
self.assertEqual(self.helper.round_down_to_limit(0.25, 1.01), 1)
self.assertEqual(self.helper.round_down_to_limit(0.25, 0.1), 0)
self.assertEqual(self.helper.round_down_to_limit(0.1, 0.99), 0.9)
self.assertEqual(self.helper.round_down_to_limit(0.01, 0.011), 0.01)
self.assertEqual(self.helper.round_down_to_limit(0.01, 1), 1)
self.assertEqual(self.helper.round_down_to_limit(1, 0.1), 0)
@patch('sys.stdout', new_callable=io.StringIO)
@patch('builtins.input', side_effect=['y', 'n', 'p', 'n'])
def test_prompt_bool(self, mock_input, mock_stdout):
self.assertTrue(self.helper.prompt_bool('test_y'))
self.assertFalse(self.helper.prompt_bool('test_n'))
self.helper.prompt_bool('test_error')
self.assertRegex(mock_stdout.getvalue(),
r'\nPlease answer with y\/n\n')
@patch('builtins.input', side_effect=['y'])
def test_prompt_string(self, mock_input):
self.assertEqual(self.helper.prompt_string('test'), 'y')
def update_product_to_trend(self, api):
''' This function updates one product in the user's stock to TREND. '''
self.report("update product price to trend")
search_string = PyMkmHelper.prompt_string('Search product name')
try:
articles = api.find_stock_article(search_string, 1)
except Exception as err:
print(err)
if len(articles) > 1:
article = self.select_from_list_of_articles(articles)
else:
article = articles[0]
found_string = f"Found: {article['product']['enName']}"
if article['product'].get('expansion'):
found_string += f"[{article['product'].get('expansion')}]."
else:
found_string += '.'
print(found_string)
undercut_local_market = PyMkmHelper.prompt_bool(
'Try to undercut local market? (slower, more requests)')
r = self.get_article_with_updated_price(article,
undercut_local_market,
api=self.api)
if r:
self.draw_price_changes_table([r])
print('\nTotal price difference: {}.'.format(
str(
round(
sum(item['price_diff'] * item['count']
for item in [r]), 2))))
if PyMkmHelper.prompt_bool(
"Do you want to update these prices?") == True:
# Update articles on MKM
print('Updating prices...')
api.set_stock([r])
print('Price updated.')
else:
print('Prices not updated.')
else:
print('No prices to update.')
def update_product_to_trend(self, search_string, api):
''' This function updates one product in the user's stock to TREND. '''
try:
articles = api.find_stock_article(search_string, 1)
except Exception as err:
print(err)
if len(articles) > 1:
article = self.select_from_list_of_articles(articles)
else:
article = articles[0]
print('Found: {} [{}].'.format(article['product']['enName'],
article['product']['expansion']))
r = self.get_price_for_single_article(article, api=self.api)
if r:
self.draw_price_changes_table([r])
print('\nTotal price difference: {}.'.format(
str(
round(
sum(item['price_diff'] * item['count']
for item in [r]), 2))))
if PyMkmHelper.prompt_bool(
"Do you want to update these prices?") == True:
# Update articles on MKM
api.set_stock([r])
print('Price updated.')
else:
print('Prices not updated.')
else:
print('No prices to update.')
def list_competition_for_product(self, api):
self.report("list competition for product")
search_string = PyMkmHelper.prompt_string('Search card name')
is_foil = PyMkmHelper.prompt_bool("Foil?")
result = api.find_product(search_string, **{
# 'exact ': 'true',
'idGame': 1,
'idLanguage': 1,
# TODO: Add Partial Content support
# TODO: Add language support
})
if (result):
products = result['product']
stock_list_products = [x['idProduct']
for x in self.get_stock_as_array(api=self.api)]
products = [x for x in products if x['idProduct']
in stock_list_products]
if len(products) == 0:
print('No matching cards in stock.')
else:
if len(products) > 1:
product = self.select_from_list_of_products(
[i for i in products if i['categoryName'] == 'Magic Single'])
elif len(products) == 1:
product = products[0]
self.show_competition_for_product(
product['idProduct'], product['enName'], is_foil, api=self.api)
else:
print('No results found.')
def update_stock_prices_to_trend(self, api):
''' This function updates all prices in the user's stock to TREND. '''
self.report("update stock price to trend")
undercut_local_market = PyMkmHelper.prompt_bool(
'Try to undercut local market? (slower, more requests)')
uploadable_json = self.calculate_new_prices_for_stock(
undercut_local_market, api=self.api)
if len(uploadable_json) > 0:
self.display_price_changes_table(uploadable_json)
if PyMkmHelper.prompt_bool("Do you want to update these prices?") == True:
# Update articles on MKM
api.set_stock(uploadable_json)
print('Prices updated.')
else:
print('Prices not updated.')
else:
print('No prices to update.')
def start(self):
while True:
menu_items = [
"Update stock prices", "Update price for a card",
"List competition for a card",
"Show top 20 expensive items in stock", "Show account info",
"Clear entire stock (WARNING)", "Import stock from .\list.csv"
]
self.print_menu(menu_items, f"PyMKM {__version__}")
choice = input("Action number: ")
try:
if choice == "1":
self.update_stock_prices_to_trend(api=self.api)
elif choice == "2":
search_string = PyMkmHelper.prompt_string(
'Search card name')
self.update_product_to_trend(search_string, api=self.api)
elif choice == "3":
is_foil = False
search_string = PyMkmHelper.prompt_string(
'Search card name')
if PyMkmHelper.prompt_bool("Foil?") == True:
is_foil = True
self.list_competition_for_product(search_string,
is_foil,
api=self.api)
elif choice == "4":
self.show_top_expensive_articles_in_stock(20, api=self.api)
elif choice == "5":
self.show_account_info(api=self.api)
elif choice == "6":
self.clear_entire_stock(api=self.api)
elif choice == "7":
self.import_from_csv(api=self.api)
elif choice == "0":
break
else:
print("Not a valid choice, try again.")
except ConnectionError as err:
print(err)
def clear_entire_stock(self, api):
self.report("clear entire stock")
stock_list = self.get_stock_as_array(api=self.api)
if PyMkmHelper.prompt_bool("Do you REALLY want to clear your entire stock ({} items)?".format(len(stock_list))) == True:
# for article in stock_list:
# article['count'] = 0
delete_list = [{'count': x['count'], 'idArticle': x['idArticle']}
for x in stock_list]
api.delete_stock(delete_list)
print('Stock cleared.')
else:
print('Aborted.')
