diff options
Diffstat (limited to 'coupons.py')
-rwxr-xr-x | coupons.py | 172 |
1 files changed, 172 insertions, 0 deletions
diff --git a/coupons.py b/coupons.py new file mode 100755 index 0000000..061f998 --- /dev/null +++ b/coupons.py @@ -0,0 +1,172 @@ +#!/usr/bin/env python3 +# File: coupons.py +# Location: . +# Author: bgstack15 +# Startdate: 2022-08-31 +# Title: Json Parser of Publix sales +# Project: coupons +# Purpose: Parse json for coupons that match lowercase string +# Usage: +# called from check-sales.sh +# Search with a lower-case string, against the lowercase values of coupon titles. +# <input.json ./coupons.py --stdin --search 'candy' +# History: +# I attempted to write similar logic with jq, but python is way easier +# Reference: +# [internal] rod2/rod.py +# Improve: +# Documentation: README.md +import sys, json, requests, os, datetime, re, textwrap + +store_urls = { + "publix": "https://clipmunk.southernsavers.com/stores/4f823db2090c841ce0000013/widget.json?callback=jQuery111106644488051860198_1661993569105&_=1661993569106", + "ingles": "https://clipmunk.southernsavers.com/stores/4f823db2090c841ce000000f/widget.json?callback=jQuery11110011370202243518035_1662043118344&_=1662043118345" +} +coupons_version = "2022-09-01a" + +def fetch(store, force = False, date = None): + """ Given a store name, visit the url and clean the json. If force, then update cached response.""" + # Reference: + # curl 'https://clipmunk.southernsavers.com/stores/4f823db2090c841ce0000013/widget.json?callback=jQuery111106644488051860198_1661993569105&_=1661993569106' | LANG=C sed -r -e 's/\\\\[uU]003[eE]/>/g;' -e 's/\\\\[uU]003[cC]/</g;' -e 's/^.*lists: \[/\[/;' -e 's/\],\\n.*$/\]/;' -e 's/\\\\"/\\"/g;' -e 's/\\"/"/g;' | ./coupons.py 'candy' | jq + if store is None or store not in [f for f in store_urls]: + print(f"ERROR (fetch): store {store} not a valid option.",file=sys.stderr) + return -1 + # try to use cache at first + contents = None + if not force: + contents = None + contents = get_cached_contents(store, date) # it is safe to return None + # So if force == True, or the cache failed + if contents is None or "" == contents: + print(f"INFO (fetch): no cached content, so visiting url",file=sys.stderr) + try: + url = store_urls[store.lower()] + except: + print(f"ERROR (fetch): no url saved for store {store}",file=sys.stderr) + r = requests.get(url) + contents = r.text + # try to save to cache, but it is not a blocker + try: + set_cached_contents(store, date, contents) + except: + pass + return contents + +def clean(contents): + """ Clean the javascript from southernsavers.com widget.json response. """ + # Reference: + # curl 'https://clipmunk.southernsavers.com/stores/4f823db2090c841ce0000013/widget.json?callback=jQuery111106644488051860198_1661993569105&_=1661993569106' | LANG=C sed -r -e 's/\\\\[uU]003[eE]/>/g;' -e 's/\\\\[uU]003[cC]/</g;' -e 's/^.*lists: \[/\[/;' -e 's/\],\\n.*$/\]/;' -e 's/\\\\"/\\"/g;' -e 's/\\"/"/g;' | ./coupons.py 'candy' | jq + a = re.sub("^.*lists: \\[","[",contents) + a = re.sub("\],\\\\n.*$","]",a) + a = re.sub("\\\\\\\\[uU]003[eE]",">",a) + a = re.sub("\\\\\\\\[uU]003[cC]","<",a) + contents = re.sub('\\\\"','"',re.sub('\\\\\\\\"','\\\\"',a)) + return contents + +def get_cached_name(store, date = None): + """ + Given store name, return cache filename regardless of existence or contents. + """ + USE_CACHE = True + store = store.lower() + cache_dir = os.environ.get("XDG_CACHE_DIR") # defauls to ~/.cache + if "" == cache_dir or cache_dir is None: + cache_dir = os.path.join(os.environ.get("HOME"),".cache") + # use an app-specific dir underneath it + cache_dir = os.path.join(cache_dir, "coupons") + if not os.path.isdir(cache_dir): + try: + os.mkdir(cache_dir) + except: + # caching is not available; but this should not stop the program + USE_CACHE = False + if USE_CACHE: + if date is None: + date = datetime.datetime.today().strftime("%F") + cache_file = os.path.join(cache_dir,"_".join([store,date]) + ".json") + return cache_file + +def get_cached_contents(store, date = None): + """ + Given store name, get cached contents + Also, use today's date if not given a specific one. + """ + cache_file = get_cached_name(store, date) + if os.path.exists(cache_file): + try: + print(f"INFO(get_cached_contents): using cache {cache_file}",file=sys.stderr) + return open(cache_file,"r").read() + except: + print(f"INFO(get_cached_contents): unable to open existing cache file {cache_file}",file=sys.stderr) + return None + +def set_cached_contents(store, date = None, contents = None): + if contents is None or "" == contents: + return True # cache nothing so short-circuit + if date is None: + date = datetime.datetime.today().strftime("%F") + store = store.lower() + cache_file = get_cached_name(store, date) + open(cache_file,"w").write(contents) + +def parse_coupons(inputobject, searchstring = None): + """ + Main logic to simplify the json down as well as return only results that match searchstring which should be lowercase. + """ + a = inputobject + b = json.loads(a) + if searchstring is None: + searchstring = "" + response = {} + for group in b: + rgroup = group["name"] + #response[rgroup] = {} + for c in group["categories"]: + rcat = c["name"] + #response[rgroup][rcat] = [] + for i in c["items"]: + text = i["html"] + if searchstring in text.lower(): + # only make this group and category if we have a match + if rgroup not in response: + response[rgroup] = {} + if rcat not in response[rgroup]: + response[rgroup][rcat] = [] + response[rgroup][rcat].append(text) + return(json.dumps(response)) + +def fetch_and_search(store, force = False, date = None, searchstring = None): + """ Main usage of the whole library. """ + a = clean(fetch(store, force, date)) + return parse_coupons(a, searchstring) + +if "__main__" == __name__: + import argparse + parser = argparse.ArgumentParser(prog = sys.argv[0], description = "Search currently listed sales/coupons on SouthernSavers.com", formatter_class=argparse.RawDescriptionHelpFormatter, epilog=textwrap.dedent("""To use the cache file as standard input, run: + <~/.cache/coupons/publix_2022-09-01.json ./coupons.py --stdin --clean --search "candy" --pretty + +Basic usage: + ./coupons.py --store "publix" --search "candy" +""")) + parser.add_argument("-n","--nocache","--nc","--no-cache", action = "store_true", help = "Skip the cache and always visit site.") + parser.add_argument("-d", "--date", help = "Use different YYYY-MM-DD than today, for cache purposes. Does not affect visiting the website") + parser.add_argument("-s","--search", help = "Search for items that match this, when converted to lowercase. Can leave blank to display all items") + parser.add_argument("--store", help = f"Select sales from this store.", choices = [f for f in store_urls]) + parser.add_argument("--stdin", action = "store_true", help = "Pipe stdin to parse_coupons. Can still use --search") + parser.add_argument("--clean", action = "store_true", help = "If using --stdin, also clean the whole javascript input into just the useful json part.") + parser.add_argument("-p","--pretty","--prettyprint","--pretty-print", action = "store_true", help = "Pretty-print json output") + parser.add_argument("-V|--version", action = "version", version = coupons_version) + args = parser.parse_args() + #print(args,file=sys.stderr) + a = None + if args.stdin: + if args.clean: + a = parse_coupons(clean(sys.stdin.read()),args.search) + else: + a = parse_coupons(sys.stdin.read(),args.search) + else: + a = fetch_and_search(args.store,args.nocache,args.date,args.search) + if args.pretty: + print(json.dumps(json.loads(a),indent=3)) + else: + print(a) |