Updated README.md with latest changes and directory names to be more meaningful
This commit is contained in:
25
main.py
25
main.py
@@ -1,19 +1,18 @@
|
||||
from argparse import ArgumentParser
|
||||
import logging
|
||||
from promotion import main_latest_promos, get_promos_by_name
|
||||
from store_utils import get_store_id
|
||||
from utils import LOGS_DIRNAME, XMLS_DIRNAME, get_products_prices
|
||||
from store_utils import get_all_deals, get_store_id
|
||||
from utils import RESULTS_DIRNAME, RAW_FILES_DIRNAME, get_products_prices
|
||||
from supermarket_chain import SupermarketChain
|
||||
from shufersal import ShuferSal
|
||||
from co_op import CoOp
|
||||
from zol_vebegadol import ZolVebegadol
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
# TODO: fix problem of left-to-right printing
|
||||
|
||||
Path(LOGS_DIRNAME).mkdir(exist_ok=True)
|
||||
Path(XMLS_DIRNAME).mkdir(exist_ok=True)
|
||||
Path(RESULTS_DIRNAME).mkdir(exist_ok=True)
|
||||
Path(RAW_FILES_DIRNAME).mkdir(exist_ok=True)
|
||||
|
||||
chain_dict = {
|
||||
'Shufersal': ShuferSal(),
|
||||
@@ -45,6 +44,9 @@ if __name__ == '__main__':
|
||||
metavar='city',
|
||||
nargs=1,
|
||||
)
|
||||
# parser.add_argument('--all_deals',
|
||||
# action='store_true',
|
||||
# )
|
||||
parser.add_argument('--load_prices',
|
||||
help='boolean flag representing whether to load an existing price XML file',
|
||||
action='store_true',
|
||||
@@ -70,7 +72,7 @@ if __name__ == '__main__':
|
||||
|
||||
logger = logging.getLogger()
|
||||
logger.setLevel(logging.INFO)
|
||||
handler = logging.FileHandler(filename=f'logs/{args.chain}_promos_{arg_store_id}.log', mode='w',
|
||||
handler = logging.FileHandler(filename=f'{RESULTS_DIRNAME}/{args.chain}_promos_{arg_store_id}.log', mode='w',
|
||||
encoding='utf-8')
|
||||
logger.addHandler(handler)
|
||||
main_latest_promos(store_id=arg_store_id, load_xml=args.load_prices, logger=logger, chain=chain)
|
||||
@@ -86,14 +88,3 @@ if __name__ == '__main__':
|
||||
arg_store_id = int(args.find_promos_by_name[0])
|
||||
get_promos_by_name(store_id=arg_store_id, chain=chain, promo_name=args.find_promos_by_name[1],
|
||||
load_prices=args.load_prices, load_promos=args.load_promos)
|
||||
|
||||
|
||||
# Script for Shufersal:
|
||||
# store_ids = get_all_deals(chain)
|
||||
# print(store_ids)
|
||||
# # store_ids = [133, 234, 73, 62, 607, 610, 111, 219, 81, 606, 609, 295, 349, 496, 611, 812, 608, 300]
|
||||
# null_items_lists = list()
|
||||
# for store_id in store_ids[::-1]:
|
||||
# print(store_id)
|
||||
# null_items_lists.append(get_all_null_items_in_promos(chain, store_id))
|
||||
# print(setintersection(*[set(list) for list in null_items_lists]))
|
||||
|
Reference in New Issue
Block a user