You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.4 KiB
66 lines
2.4 KiB
4 months ago
|
|
||
|
|
from changedetectionio.model.Watch import model as BaseWatch
|
||
|
|
import re
|
||
|
|
from babel.numbers import parse_decimal
|
||
|
|
|
||
|
|
class Restock(dict):
|
||
|
|
|
||
|
|
def parse_currency(self, raw_value: str) -> float:
|
||
|
|
# Clean and standardize the value (ie 1,400.00 should be 1400.00), even better would be store the whole thing as an integer.
|
||
|
|
standardized_value = raw_value
|
||
|
|
|
||
|
|
if ',' in standardized_value and '.' in standardized_value:
|
||
|
|
# Identify the correct decimal separator
|
||
|
|
if standardized_value.rfind('.') > standardized_value.rfind(','):
|
||
|
|
standardized_value = standardized_value.replace(',', '')
|
||
|
|
else:
|
||
|
|
standardized_value = standardized_value.replace('.', '').replace(',', '.')
|
||
|
|
else:
|
||
|
|
standardized_value = standardized_value.replace(',', '.')
|
||
|
|
|
||
|
|
# Remove any non-numeric characters except for the decimal point
|
||
|
|
standardized_value = re.sub(r'[^\d.-]', '', standardized_value)
|
||
|
|
|
||
|
|
# Convert to float
|
||
|
|
return float(parse_decimal(standardized_value, locale='en'))
|
||
|
|
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
# Define default values
|
||
|
|
default_values = {
|
||
|
|
'in_stock': None,
|
||
|
|
'price': None,
|
||
|
|
'currency': None,
|
||
|
|
'original_price': None
|
||
|
|
}
|
||
|
|
|
||
|
|
# Initialize the dictionary with default values
|
||
|
|
super().__init__(default_values)
|
||
|
|
|
||
|
|
# Update with any provided positional arguments (dictionaries)
|
||
|
|
if args:
|
||
|
|
if len(args) == 1 and isinstance(args[0], dict):
|
||
|
|
self.update(args[0])
|
||
|
|
else:
|
||
|
|
raise ValueError("Only one positional argument of type 'dict' is allowed")
|
||
|
|
|
||
|
|
def __setitem__(self, key, value):
|
||
|
|
# Custom logic to handle setting price and original_price
|
||
|
|
if key == 'price':
|
||
|
|
if isinstance(value, str):
|
||
|
|
value = self.parse_currency(raw_value=value)
|
||
|
|
|
||
|
|
if value and not self.get('original_price'):
|
||
|
|
self['original_price'] = value
|
||
|
|
|
||
|
|
super().__setitem__(key, value)
|
||
|
|
|
||
|
|
class Watch(BaseWatch):
|
||
|
|
def __init__(self, *arg, **kw):
|
||
|
|
super().__init__(*arg, **kw)
|
||
|
|
self['restock'] = Restock(kw['default']['restock']) if kw.get('default') and kw['default'].get('restock') else Restock()
|
||
|
|
|
||
|
|
def clear_watch(self):
|
||
|
|
super().clear_watch()
|
||
|
|
self.update({'restock': Restock()})
|
||
|
|
|