diff --git a/bareket.py b/bareket.py deleted file mode 100644 index ca1e24f..0000000 --- a/bareket.py +++ /dev/null @@ -1,5 +0,0 @@ -from mahsaneiHashook import MahsaneiHashook - - -class Bareket(MahsaneiHashook): - pass diff --git a/chains/bareket.py b/chains/bareket.py new file mode 100644 index 0000000..3d8d066 --- /dev/null +++ b/chains/bareket.py @@ -0,0 +1,5 @@ +from chains.mahsaneiHashook import MahsaneiHashook + + +class Bareket(MahsaneiHashook): + pass diff --git a/cerberus_web_client.py b/chains/cerberus_web_client.py similarity index 100% rename from cerberus_web_client.py rename to chains/cerberus_web_client.py diff --git a/chains/co_op.py b/chains/co_op.py new file mode 100644 index 0000000..d03d972 --- /dev/null +++ b/chains/co_op.py @@ -0,0 +1,5 @@ +from chains.mahsaneiHashook import MahsaneiHashook + + +class CoOp(MahsaneiHashook): + pass diff --git a/dor_alon.py b/chains/dor_alon.py similarity index 81% rename from dor_alon.py rename to chains/dor_alon.py index f71b4e3..b175509 100644 --- a/dor_alon.py +++ b/chains/dor_alon.py @@ -1,4 +1,4 @@ -from cerberus_web_client import CerberusWebClient +from chains.cerberus_web_client import CerberusWebClient from supermarket_chain import SupermarketChain diff --git a/freshmarket.py b/chains/freshmarket.py similarity index 81% rename from freshmarket.py rename to chains/freshmarket.py index 8f9b9cc..dcfbdad 100644 --- a/freshmarket.py +++ b/chains/freshmarket.py @@ -1,4 +1,4 @@ -from cerberus_web_client import CerberusWebClient +from chains.cerberus_web_client import CerberusWebClient from supermarket_chain import SupermarketChain diff --git a/hazi_hinam.py b/chains/hazi_hinam.py similarity index 81% rename from hazi_hinam.py rename to chains/hazi_hinam.py index 0bfa45e..bacedff 100644 --- a/hazi_hinam.py +++ b/chains/hazi_hinam.py @@ -1,4 +1,4 @@ -from cerberus_web_client import CerberusWebClient +from chains.cerberus_web_client import CerberusWebClient from supermarket_chain import SupermarketChain diff --git a/keshet.py b/chains/keshet.py similarity index 80% rename from keshet.py rename to chains/keshet.py index 41fe4f2..99a382e 100644 --- a/keshet.py +++ b/chains/keshet.py @@ -1,4 +1,4 @@ -from cerberus_web_client import CerberusWebClient +from chains.cerberus_web_client import CerberusWebClient from supermarket_chain import SupermarketChain diff --git a/mahsaneiHashook.py b/chains/mahsaneiHashook.py similarity index 100% rename from mahsaneiHashook.py rename to chains/mahsaneiHashook.py diff --git a/osher_ad.py b/chains/osher_ad.py similarity index 81% rename from osher_ad.py rename to chains/osher_ad.py index 0f1637c..d4afc4e 100644 --- a/osher_ad.py +++ b/chains/osher_ad.py @@ -1,4 +1,4 @@ -from cerberus_web_client import CerberusWebClient +from chains.cerberus_web_client import CerberusWebClient from supermarket_chain import SupermarketChain diff --git a/rami_levi.py b/chains/rami_levi.py similarity index 81% rename from rami_levi.py rename to chains/rami_levi.py index 1f94f64..2c078c2 100644 --- a/rami_levi.py +++ b/chains/rami_levi.py @@ -1,4 +1,4 @@ -from cerberus_web_client import CerberusWebClient +from chains.cerberus_web_client import CerberusWebClient from supermarket_chain import SupermarketChain diff --git a/shufersal.py b/chains/shufersal.py similarity index 100% rename from shufersal.py rename to chains/shufersal.py diff --git a/stop_market.py b/chains/stop_market.py similarity index 81% rename from stop_market.py rename to chains/stop_market.py index d086b60..f9fcd52 100644 --- a/stop_market.py +++ b/chains/stop_market.py @@ -1,4 +1,4 @@ -from cerberus_web_client import CerberusWebClient +from chains.cerberus_web_client import CerberusWebClient from supermarket_chain import SupermarketChain diff --git a/tiv_taam.py b/chains/tiv_taam.py similarity index 77% rename from tiv_taam.py rename to chains/tiv_taam.py index 6e64cf9..ed94e23 100644 --- a/tiv_taam.py +++ b/chains/tiv_taam.py @@ -1,4 +1,4 @@ -from cerberus_web_client import CerberusWebClient +from chains.cerberus_web_client import CerberusWebClient from supermarket_chain import SupermarketChain diff --git a/chains/victory.py b/chains/victory.py new file mode 100644 index 0000000..7181876 --- /dev/null +++ b/chains/victory.py @@ -0,0 +1,5 @@ +from chains.mahsaneiHashook import MahsaneiHashook + + +class Victory(MahsaneiHashook): + pass diff --git a/yohananof.py b/chains/yohananof.py similarity index 81% rename from yohananof.py rename to chains/yohananof.py index 4a790ea..c2eac74 100644 --- a/yohananof.py +++ b/chains/yohananof.py @@ -1,4 +1,4 @@ -from cerberus_web_client import CerberusWebClient +from chains.cerberus_web_client import CerberusWebClient from supermarket_chain import SupermarketChain diff --git a/zol_vebegadol.py b/chains/zol_vebegadol.py similarity index 100% rename from zol_vebegadol.py rename to chains/zol_vebegadol.py diff --git a/co_op.py b/co_op.py deleted file mode 100644 index 8d7e7b4..0000000 --- a/co_op.py +++ /dev/null @@ -1,5 +0,0 @@ -from mahsaneiHashook import MahsaneiHashook - - -class CoOp(MahsaneiHashook): - pass diff --git a/main.py b/main.py index 38bb4cc..5640b35 100644 --- a/main.py +++ b/main.py @@ -1,37 +1,35 @@ from argparse import ArgumentParser import logging +from pathlib import Path -from bareket import Bareket -from mahsaneiHashook import MahsaneiHashook -from dor_alon import DorAlon -from freshmarket import Freshmarket -from hazi_hinam import HaziHinam -from keshet import Keshet from promotion import main_latest_promos, get_promos_by_name -from stop_market import StopMarket -from store_utils import get_all_deals, get_store_id -from tiv_taam import TivTaam +from store_utils import 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 victory import Victory -from yohananof import Yohananof -from zol_vebegadol import ZolVebegadol -from rami_levi import RamiLevi -from osher_ad import OsherAd -from pathlib import Path +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, +) # TODO: fix problem of left-to-right printing Path(RESULTS_DIRNAME).mkdir(exist_ok=True) Path(RAW_FILES_DIRNAME).mkdir(exist_ok=True) -chains = ShuferSal, CoOp, ZolVebegadol, RamiLevi, OsherAd, Freshmarket, TivTaam, HaziHinam, Keshet, StopMarket, \ - Yohananof, DorAlon, MahsaneiHashook, Bareket, Victory - -chain_dict = {repr(chain): chain() for chain in chains} - +chain_dict = {repr(chain): chain() for chain in SupermarketChain.__subclasses__()} if __name__ == '__main__': parser = ArgumentParser() diff --git a/victory.py b/victory.py deleted file mode 100644 index 7fc1134..0000000 --- a/victory.py +++ /dev/null @@ -1,5 +0,0 @@ -from mahsaneiHashook import MahsaneiHashook - - -class Victory(MahsaneiHashook): - pass