Has added Zol Vebengadol. Fixed small bug in Co-Op.

main.py

@@ -6,8 +6,10 @@ from utils import LOGS_DIRNAME, XMLS_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)
@@ -16,6 +18,7 @@ Path(XMLS_DIRNAME).mkdir(exist_ok=True)
 chain_dict = {
     'Shufersal': ShuferSal(),
     'Co-Op': CoOp(),
+    'Zol-Vebegadol': ZolVebegadol()
 }
 
 if __name__ == '__main__':
@@ -57,7 +60,7 @@ if __name__ == '__main__':
     parser.add_argument('--chain',
                         required=True,
                         help='The name of the requested chain',
-                        choices=['Shufersal', 'Co-Op'],
+                        choices=chain_dict.keys(),
                         )
     args = parser.parse_args()
 
@@ -83,3 +86,14 @@ 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]))