134 lines
5.5 KiB
Python
134 lines
5.5 KiB
Python
import os
|
|
from argparse import ArgumentParser
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
import logging
|
|
|
|
from promotion import main_latest_promos, log_promos_by_name
|
|
from store_utils import log_stores_ids
|
|
from utils import RESULTS_DIRNAME, RAW_FILES_DIRNAME, VALID_PROMOTION_FILE_EXTENSIONS, log_products_prices, \
|
|
valid_promotion_output_file
|
|
from supermarket_chain import SupermarketChain
|
|
from chains import (
|
|
bareket,
|
|
mahsaneiHashook,
|
|
dor_alon,
|
|
freshmarket,
|
|
hazi_hinam,
|
|
keshet,
|
|
stop_market,
|
|
tiv_taam,
|
|
shufersal,
|
|
co_op,
|
|
victory,
|
|
yohananof,
|
|
zol_vebegadol,
|
|
rami_levi,
|
|
osher_ad,
|
|
maayan2000,
|
|
shuk_hayir,
|
|
king_store,
|
|
shefa_birkat_hashem,
|
|
)
|
|
|
|
Path(RESULTS_DIRNAME).mkdir(exist_ok=True)
|
|
Path(RAW_FILES_DIRNAME).mkdir(exist_ok=True)
|
|
|
|
chain_dict = {repr(chain): chain() if callable(chain) else None for chain in SupermarketChain.__subclasses__()}
|
|
|
|
# TODO: change functions arguments to include all necessary parameters (e.g. chain) or split arguments
|
|
if __name__ == '__main__':
|
|
parser = ArgumentParser()
|
|
parser.add_argument('--promos',
|
|
help="generates a CSV file with all the promotions in the requested store",
|
|
metavar='store_id',
|
|
nargs=1,
|
|
type=SupermarketChain.store_id_type,
|
|
)
|
|
parser.add_argument('--find_promos_by_name',
|
|
help="prints all promos containing the given promo_name in the given store",
|
|
metavar=('store_id', 'promo_name'),
|
|
nargs=2,
|
|
# type=store_id_type, # TODO: add type-checking of first parameter
|
|
)
|
|
parser.add_argument('--price',
|
|
help='prints all products that contain the given name in the requested store',
|
|
metavar=('store_id', 'product_name'),
|
|
nargs=2,
|
|
)
|
|
parser.add_argument('--find_store_id',
|
|
help='prints all Shufersal stores in a given city. Input should be a city name in Hebrew',
|
|
metavar='city',
|
|
nargs=1,
|
|
)
|
|
parser.add_argument('--load_prices',
|
|
help='boolean flag representing whether to load an existing price XML file',
|
|
action='store_true',
|
|
)
|
|
parser.add_argument('--load_promos',
|
|
help='boolean flag representing whether to load an existing promo XML file',
|
|
action='store_true',
|
|
)
|
|
parser.add_argument('--load_stores',
|
|
help='boolean flag representing whether to load an existing stores XML file',
|
|
action='store_true',
|
|
)
|
|
parser.add_argument('--chain',
|
|
required=True,
|
|
help='The name of the requested chain',
|
|
choices=chain_dict.keys(),
|
|
)
|
|
parser.add_argument('--file_extension',
|
|
help='The extension of the promotions output file',
|
|
choices=VALID_PROMOTION_FILE_EXTENSIONS,
|
|
default='.xlsx',
|
|
)
|
|
parser.add_argument('--output_filename',
|
|
help='The path to write the promotions table to',
|
|
type=valid_promotion_output_file,
|
|
)
|
|
parser.add_argument('--only_export_to_file',
|
|
help='Boolean flag representing whether only export or also open the promotion output file',
|
|
action='store_true',
|
|
)
|
|
parser.add_argument('--debug',
|
|
help='Boolean flag representing whether to run in debug mode',
|
|
action='store_true',
|
|
)
|
|
args = parser.parse_args()
|
|
if args.debug:
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
else:
|
|
logging.basicConfig(level=logging.INFO, format='%(levelname)s:%(message)s')
|
|
|
|
chain: SupermarketChain = chain_dict[args.chain]
|
|
|
|
if args.promos:
|
|
arg_store_id = int(args.promos[0])
|
|
|
|
if args.output_filename:
|
|
output_filename = args.output_filename
|
|
directory = os.path.dirname(output_filename)
|
|
Path(directory).mkdir(parents=True, exist_ok=True)
|
|
else:
|
|
Path(RESULTS_DIRNAME).mkdir(exist_ok=True)
|
|
output_filename = f'{RESULTS_DIRNAME}/{repr(type(chain))}_promos_{arg_store_id}{args.file_extension}'
|
|
|
|
main_latest_promos(store_id=arg_store_id, output_filename=output_filename, chain=chain,
|
|
load_promos=args.load_promos, load_xml=args.load_prices)
|
|
if not args.only_export_to_file:
|
|
os.startfile(Path(output_filename))
|
|
logging.debug(f'Process finished at: {datetime.now()}')
|
|
|
|
elif args.price:
|
|
log_products_prices(chain, store_id=args.price[0], load_xml=args.load_prices, product_name=args.price[1])
|
|
|
|
elif args.find_store_id:
|
|
arg_city = args.find_store_id[0]
|
|
log_stores_ids(city=arg_city, load_xml=args.load_stores, chain=chain)
|
|
|
|
elif args.find_promos_by_name:
|
|
arg_store_id = int(args.find_promos_by_name[0])
|
|
log_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)
|