saret/supermarket-scraping
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Has added tests for the promotion functions for Shufersal and CoOp. Also added minor design changes in promotion.py and item.py

Browse Source
This commit is contained in:
KorenLazar
2021-03-08 14:13:30 +02:00
parent c86fc7c1ab
commit 9f5464317d
5 changed files with 393 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
item.py
View File

@@ -3,11 +3,11 @@ class Item:
A class representing a product in some supermarket.
"""
def __init__(self, name: str, price: float, manufacturer: str, code: int):
def __init__(self, name: str, price: float, manufacturer: str, code: str):
self.name: str = name
self.price: float = price
self.manufacturer: str = manufacturer
self.code: int = code
self.code: str = code
def __repr__(self):
return str((self.name, self.price, self.manufacturer, self.code))

105
promotion.py
View File

@@ -1,13 +1,13 @@
import re
from datetime import datetime
from enum import Enum
from typing import Dict, List
from typing import Dict, List, Union
import csv
from item import Item
from utils import (
create_items_dict,
xml_file_gen,
get_float_from_tag, xml_file_gen,
create_bs_object,
)
from supermarket_chain import SupermarketChain
@@ -17,12 +17,6 @@ INVALID_OR_UNKNOWN_PROMOTION_FUNCTION = -1
PRODUCTS_TO_IGNORE = ['סירים', 'מגבות', 'מגבת', 'מפות', 'פסטיגל', 'ביגי']
# class ClubID(Enum):
# Regular = 'מבצע רגיל'
# Club = 'מועדון'
# CreditCard = 'כרטיס אשראי'
# Other = 'אחר'
class ClubID(Enum):
מבצע_רגיל = 0
מועדון = 1
@@ -48,7 +42,7 @@ class Promotion:
It contains only part of the available information in Shufersal's data.
"""
def __init__(self, content: str, start_date: datetime, end_date: datetime, update_date: datetime, item: List[Item],
def __init__(self, content: str, start_date: datetime, end_date: datetime, update_date: datetime, items: List[Item],
promo_func: callable, club_id: ClubID, promotion_id: float, max_qty: int,
allow_multiple_discounts: bool, reward_type: RewardType):
self.content: str = content
@@ -56,25 +50,17 @@ class Promotion:
self.end_date: datetime = end_date
self.update_date: datetime = update_date
self.promo_func: callable = promo_func
self.items: List[Item] = item
self.items: List[Item] = items
self.club_id: ClubID = club_id
self.max_qty: int = max_qty
self.allow_multiple_discounts = allow_multiple_discounts
self.reward_type = reward_type
self.promotion_id = promotion_id
# def __repr__(self):
# title = self.content
# dates_range = f"Between {self.start_date} and {self.end_date}"
# update_line = f"Updated at {self.update_date}"
# items = '\n'.join(str(item) for item in self.item)
# return '\n'.join([title, dates_range, update_line, items]) + '\n'
def repr_ltr(self):
title = self.content
dates_range = f"Between {self.start_date} and {self.end_date}"
update_line = f"Updated at {self.update_date}"
# items = '\n'.join(str(item) for item in self.item)
return '\n'.join([title, dates_range, update_line, str(self.items)]) + '\n'
def __eq__(self, other):
@@ -97,19 +83,22 @@ def write_promotions_to_csv(promotions: List[Promotion], output_filename: str) -
'מחיר אחרי מבצע',
'אחוז הנחה',
'סוג מבצע',
'כמות מקסימלית',
'כמות מקס',
'כפל הנחות',
'המבצע החל',
'זמן תחילת מבצע',
'זמן סיום מבצע',
'זמן עדכון אחרון',
'יצרן',
'ברקוד פריט',
'סוג מבצע',
'סוג מבצע לפי תקנות שקיפות מחירים',
])
for promo in promotions:
promos_writer.writerows(
[[promo.content,
promos_writer.writerows([get_promotion_row_in_csv(promo, item) for item in promo.items])
def get_promotion_row_in_csv(promo: Promotion, item: Item):
return [promo.content,
item.name,
item.price,
f'{promo.promo_func(item):.3f}',
@@ -117,13 +106,13 @@ def write_promotions_to_csv(promotions: List[Promotion], output_filename: str) -
promo.club_id.name.replace('_', ' '),
promo.max_qty,
promo.allow_multiple_discounts,
promo.start_date <= datetime.now(),
promo.start_date,
promo.end_date,
promo.update_date,
item.manufacturer,
item.code,
promo.reward_type.value] for item in promo.items]
)
promo.reward_type.value]
def get_available_promos(chain: SupermarketChain, store_id: int, load_prices: bool, load_promos) -> List[Promotion]:
@@ -159,18 +148,14 @@ def create_new_promo_instance(chain, items_dict, promo, promotion_id):
discounted_price = get_discounted_price(promo)
promo_description = promo.find('PromotionDescription').text
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(promo, is_discount_in_percentage)
min_qty = get_int_from_tag(promo, 'MinQty')
max_qty = get_int_from_tag(promo, 'MaxQty')
raw_discount_rate = promo.find('DiscountRate').text if promo.find('DiscountRate') else None
discount_rate = get_discount_rate(raw_discount_rate, is_discount_in_percentage)
min_qty = get_float_from_tag(promo, 'MinQty')
max_qty = get_float_from_tag(promo, 'MaxQty')
remark = promo.find("Remark")
promo_func = determine_promo_function(
reward_type=reward_type,
remark=remark,
promo_description=promo_description,
discounted_price=discounted_price,
discount_rate=discount_rate,
min_qty=min_qty,
)
promo_func = find_promo_function(reward_type=reward_type, remark=remark.text if remark else '',
promo_description=promo_description, min_qty=min_qty,
discount_rate=discount_rate, discounted_price=discounted_price)
promo_start_time = datetime.strptime(promo.find('PromotionStartDate').text + ' ' +
promo.find('PromotionStartHour').text,
chain.date_hour_format)
@@ -183,78 +168,69 @@ def create_new_promo_instance(chain, items_dict, promo, promotion_id):
multiple_discounts_allowed = bool(int(promo.find('AllowMultipleDiscounts').text))
items = chain.get_items(promo, items_dict)
if is_valid_promo(start_time=promo_start_time, end_time=promo_end_time, description=promo_description):
if is_valid_promo(end_time=promo_end_time, description=promo_description):
return Promotion(content=promo_description, start_date=promo_start_time, end_date=promo_end_time,
update_date=promo_update_time, item=items, promo_func=promo_func,
update_date=promo_update_time, items=items, promo_func=promo_func,
club_id=club_id, promotion_id=promotion_id, max_qty=max_qty,
allow_multiple_discounts=multiple_discounts_allowed, reward_type=reward_type)
def get_int_from_tag(tag, int_tag):
content = tag.find(int_tag)
return int(float(content.text)) if content else 0
def get_discounted_price(promo):
discounted_price = promo.find('DiscountedPrice')
if discounted_price:
return float(discounted_price.text)
def get_discount_rate(promo, discount_in_percentage):
discount_rate = promo.find("DiscountRate")
def get_discount_rate(discount_rate: Union[float, None], discount_in_percentage: bool):
if discount_rate:
if discount_in_percentage:
return int(discount_rate.text) * (10 ** -(len(str(discount_rate.text))))
return float(discount_rate.text)
return int(discount_rate) * (10 ** -(len(str(discount_rate))))
return float(discount_rate)
def determine_promo_function(reward_type, remark, promo_description, discounted_price, discount_rate, min_qty):
def find_promo_function(reward_type: RewardType, remark: str, promo_description: str, min_qty: float,
discount_rate: Union[float, None], discounted_price: Union[float, None]):
if reward_type == RewardType.SECOND_INSTANCE_DIFFERENT_DISCOUNT:
if not discounted_price:
return lambda item: item.price * (1 - (discount_rate / min_qty))
else:
return lambda item: (item.price * (min_qty - 1) + discounted_price) / min_qty
elif reward_type == RewardType.DISCOUNT_IN_ITEM_IF_PURCHASING_OTHER_ITEMS:
if reward_type == RewardType.DISCOUNT_IN_ITEM_IF_PURCHASING_OTHER_ITEMS:
return lambda item: item.price
elif reward_type == RewardType.SECOND_OR_THIRD_INSTANCE_FOR_FREE:
if reward_type == RewardType.SECOND_OR_THIRD_INSTANCE_FOR_FREE:
return lambda item: item.price * (1 - (1 / min_qty))
elif reward_type == RewardType.DISCOUNT_IN_PERCENTAGE:
if reward_type == RewardType.DISCOUNT_IN_PERCENTAGE:
return lambda item: item.price * (1 - discount_rate / (2 if "השני ב" in promo_description else 1))
elif reward_type == RewardType.SECOND_INSTANCE_SAME_DISCOUNT:
if reward_type == RewardType.SECOND_INSTANCE_SAME_DISCOUNT:
if "השני ב" in promo_description:
return lambda item: (item.price + discounted_price) / 2
else:
return lambda item: discounted_price / min_qty
elif reward_type == RewardType.DISCOUNT_BY_THRESHOLD:
if reward_type == RewardType.DISCOUNT_BY_THRESHOLD:
return lambda item: item.price - discount_rate
elif remark and 'מחיר המבצע הינו המחיר לק"ג' in remark.text:
if 'מחיר המבצע הינו המחיר לק"ג' in remark:
return lambda item: discounted_price
elif discounted_price and min_qty:
if discounted_price and min_qty:
return lambda item: discounted_price / min_qty
return lambda item: INVALID_OR_UNKNOWN_PROMOTION_FUNCTION
def is_valid_promo(start_time: datetime, end_time: datetime, description):
def is_valid_promo(end_time: datetime, description) -> bool:
"""
This function returns whether a given Promotion object is currently valid.
"""
today_date: datetime = datetime.now()
not_expired: bool = end_time >= today_date
has_started: bool = start_time <= today_date
not_expired: bool = end_time >= datetime.now()
in_promo_ignore_list: bool = any(product in description for product in PRODUCTS_TO_IGNORE)
return not_expired and has_started and not in_promo_ignore_list
return not_expired and not in_promo_ignore_list
def main_latest_promos(store_id: int, load_xml: bool, chain: SupermarketChain, load_promos: bool):
def main_latest_promos(store_id: int, load_xml: bool, chain: SupermarketChain, load_promos: bool) -> None:
"""
This function writes to a CSV file the available promotions in a store with a given id sorted by their update date.
@@ -286,7 +262,8 @@ def get_promos_by_name(store_id: int, chain: SupermarketChain, promo_name: str,
print(promo.repr_ltr())
def get_all_null_items_in_promos(chain, store_id):
# TODO: change to returning list of Items
def get_all_null_items_in_promos(chain, store_id) -> List[str]:
"""
This function finds all items appearing in the chain's promotions file but not in the chain's prices file.
"""

331
tests/test_promotions_parsing.py Normal file
View File

@@ -0,0 +1,331 @@
from item import Item
from promotion import RewardType, find_promo_function, get_discount_rate
# TODO: create a test for Shufersal promo type 3
def test_shufersal_promo_type_1():
reward_type = RewardType(1)
discounted_price = 100.00
orig_discount_rate = None
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark=' מחיר המבצע הינו המחיר לק"ג ',
promo_description='300ב30 פטה פיראוס 20% במשקל',
min_qty=0.3,
discount_rate=discount_rate,
discounted_price=discounted_price,
)
item = Item('פטה פיראוס 20%', 113, '', '')
assert promo_func(item) == 100
def test_shufersal_promo_type_2():
reward_type = RewardType(2)
discounted_price = None
orig_discount_rate = 2000
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark='',
promo_description='20%הנחה גרנולה פנינה רוזנבלום500',
min_qty=1,
discount_rate=discount_rate,
discounted_price=discounted_price,
)
item = Item('חגיגת גרנולה פ.יבשים500ג', 26.9, '', '')
assert promo_func(item) == 21.52
def test_shufersal_promo_type_6_1():
reward_type = RewardType(6)
discounted_price = 0.00
orig_discount_rate = None
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark=' מחיר המבצע הינו המחיר לק"ג ',
promo_description='ב-קנה350גרם נקניק במעדניה קבל קופסת מתנה',
min_qty=0.35,
discount_rate=discount_rate,
discounted_price=discounted_price,
)
item = Item('פסטרמה מקסיקנית במשקל', 89, '', '')
assert promo_func(item) == 89
def test_shufersal_promo_type_6_2():
reward_type = RewardType(6)
discounted_price = 0.00
orig_discount_rate = None
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark='',
promo_description='מכונת קפה לוואצה גולי2-חב קפסולות',
min_qty=1.00,
discount_rate=discount_rate,
discounted_price=discounted_price,
)
item = Item('מכונת לוואצה ג\'ולי אדומה', 449, '', '')
assert promo_func(item) == 449
def test_shufersal_promo_type_7_1():
reward_type = RewardType(7)
discounted_price = None
orig_discount_rate = 10000
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark='',
promo_description='1+1הזול מוצרי קולקשיין שופרסל',
min_qty=2.00,
discount_rate=discount_rate,
discounted_price=discounted_price,
)
item = Item('פינצטה 2011 שחורה/כסופה', 14.9, '', '')
assert promo_func(item) == 7.45
def test_shufersal_promo_type_7_2():
reward_type = RewardType(7)
discounted_price = None
orig_discount_rate = 10000
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark='',
promo_description='3+1 יוגורט עיזים ביו 150 גרם',
min_qty=4.00,
discount_rate=discount_rate,
discounted_price=discounted_price,
)
item = Item('יוגורט עיזים 500 גרם', 12.9, '', '')
assert promo_func(item) == 12.9 * 0.75
def test_shufersal_promo_type_9_1():
reward_type = RewardType(9)
discounted_price = None
orig_discount_rate = 5000
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark='',
promo_description='שני ב%50הנחה מוצרי מותג קבוצת יבנה',
min_qty=2.00,
discount_rate=discount_rate,
discounted_price=discounted_price,
)
item = Item('זיתים מבוקעים פיקנטי540ג', 9.3, '', '')
assert promo_func(item) == 9.3 * 0.75
def test_shufersal_promo_type_9_2():
reward_type = RewardType(9)
discounted_price = 10.00
orig_discount_rate = None
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark='',
promo_description='ב-שני ב10 ירקות קפואים שופרסל',
min_qty=2.00,
discount_rate=discount_rate,
discounted_price=discounted_price,
)
item = Item('שעועית לבנה שופרסל 800גר', 18.9, '', '')
assert promo_func(item) == (18.9 + 10) / 2
def test_shufersal_promo_type_9_3():
reward_type = RewardType(9)
discounted_price = None
orig_discount_rate = 5000
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark='',
promo_description='השני ב50% הזול אביזרי שיער BE NOW',
min_qty=2.00,
discount_rate=discount_rate,
discounted_price=discounted_price,
)
item = Item('גומיות שחורות 12 יח', 9.9, '', '')
assert promo_func(item) == 9.9 * 0.75
def test_shufersal_promo_type_10_1():
reward_type = RewardType(10)
discounted_price = 10
orig_discount_rate = None
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark='',
promo_description='2ב10משקה סויה מועשר בחלבון 250 מ"ל',
min_qty=2,
discount_rate=discount_rate,
discounted_price=discounted_price
)
item = Item('טופו טעם טבעי 300 גרם', 10.9, '', '7296073345763')
assert promo_func(item) == 5
def test_shufersal_promo_type_10_2():
reward_type = RewardType(10)
discounted_price = 14
orig_discount_rate = None
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark='',
promo_description='2ב14טופו טבעי/רך מועשר בסידן 300 גרם',
min_qty=2,
discount_rate=discount_rate,
discounted_price=discounted_price
)
item = Item('טופו טעם טבעי 300 גרם', 10.9, 'כפרי בריא משק ויילר', '7296073345763')
assert promo_func(item) == 7
def assert_discount(discounted_price, item_barcode, item_manufacturer, item_name, min_qty, orig_discount_rate,
orig_price, price_after_discount, promo_description, reward_type, remark):
is_discount_in_percentage = reward_type == RewardType.DISCOUNT_IN_PERCENTAGE or not discounted_price
discount_rate = get_discount_rate(orig_discount_rate, is_discount_in_percentage)
promo_func = find_promo_function(
reward_type=reward_type,
remark=remark,
promo_description=promo_description,
min_qty=min_qty,
discount_rate=discount_rate,
discounted_price=discounted_price
)
item = Item(item_name, orig_price, item_manufacturer, item_barcode)
assert abs(promo_func(item) - price_after_discount) <= 1e-5, promo_description
def test_coop_promo_type_2():
reward_type = RewardType(2)
discounted_price = 6.95
orig_discount_rate = 50
promo_description = 'ק/עגבני. השני ב50% הנחה רסק/תרכיז/המשתתפים'
orig_price = 13.9
price_after_discount = 13.9 * 0.75
item_name = 'מחית עגבניות גרנד איטליה 700גר'
item_manufacturer = 'תה ויסוצקי (ישראל)בעמ'
item_barcode = '7290015150088'
min_qty = 2
remark = ''
assert_discount(discounted_price, item_barcode, item_manufacturer, item_name, min_qty, orig_discount_rate,
orig_price, price_after_discount, promo_description, reward_type, remark)
def test_coop_promo_type_3_1():
reward_type = RewardType(3)
discounted_price = 19.90
orig_discount_rate = 20
promo_description = '*ק/מקס/בירה ב19.90ש עד2יח סטלה 1/6 330מ"ל'
orig_price = 39.90
price_after_discount = 19.90
item_name = 'בירה סטלה מארז שישיה 330 מ"ל'
item_manufacturer = 'קומפלקס כימיקלסבעמ ~~~'
item_barcode = '7290002814016'
min_qty = 1
remark = ''
assert_discount(discounted_price, item_barcode, item_manufacturer, item_name, min_qty, orig_discount_rate,
orig_price, price_after_discount, promo_description, reward_type, remark)
def test_coop_promo_type_3_2():
reward_type = RewardType(3)
discounted_price = 14.90
orig_discount_rate = 2.50
promo_description = 'ק/אמנטל ב14.90 נעם 30% 200גר'
orig_price = 17.4
price_after_discount = 14.90
item_name = 'אמנטל נעם 200ג'
item_manufacturer = '*חברה המרכזית להפצתמשקאותבעמ*'
item_barcode = '7290102397730'
min_qty = 1
remark = ''
assert_discount(discounted_price, item_barcode, item_manufacturer, item_name, min_qty, orig_discount_rate,
orig_price, price_after_discount, promo_description, reward_type, remark)
def test_coop_promo_type_7_1():
reward_type = RewardType(7)
discounted_price = None
orig_discount_rate = 100
promo_description = 'ק/מגוון מוצרי סנפרוסט 3+1 הזול מבניהם ללא לקטים'
orig_price = 23.9
price_after_discount = orig_price * 0.75
item_name = 'עדשים מבושלים 1ק"ג'
item_manufacturer = 'תנובה בשר'
item_barcode = '104072'
min_qty = 4
remark = ''
assert_discount(discounted_price, item_barcode, item_manufacturer, item_name, min_qty, orig_discount_rate,
orig_price, price_after_discount, promo_description, reward_type, remark)
def test_coop_promo_type_7_2():
reward_type = RewardType(7)
discounted_price = 0
orig_discount_rate = 100
promo_description = 'ק/2+1ספיד סטיק הזול מבניהם המשתתפים'
orig_price = 26.9
price_after_discount = orig_price * 2 / 3
item_name = '##ספיד סטיק ג ל-כחול'
item_manufacturer = 'ש.שסטוביץ בעמ'
item_barcode = '22200956932'
min_qty = 3
remark = ''
assert_discount(discounted_price, item_barcode, item_manufacturer, item_name, min_qty, orig_discount_rate,
orig_price, price_after_discount, promo_description, reward_type, remark)
def test_coop_promo_type_10():
reward_type = RewardType(10)
discounted_price = 25
orig_discount_rate = 2.57
promo_description = 'ק/סנו סושי 3ב25 מטליות זיגזג 1/3/כריות הפלא/9מטליו'
orig_price = 10.90
price_after_discount = 25 / 3
item_name = 'סנו סושי מטלית הפלא לרצפה Decor'
item_manufacturer = 'החב הדרומית לשיווק'
item_barcode = '7290108353686'
min_qty = 3
remark = ''
assert_discount(discounted_price, item_barcode, item_manufacturer, item_name, min_qty, orig_discount_rate,
orig_price, price_after_discount, promo_description, reward_type, remark)

0
tests/test_scraping.py Normal file
View File

5
utils.py
View File

@@ -118,3 +118,8 @@ def get_products_prices(chain: SupermarketChain, store_id: int, load_xml: bool,
prod.find('ItemPrice').text
)
)
def get_float_from_tag(tag, int_tag) -> int:
content = tag.find(int_tag)
return float(content.text) if content else 0