from datetime import datetime
from bokchoy.chain import Chain, Step
from bokchoy import (
    consumers as bokchoy_consumers,
    parsers as bokchoy_parsers
)
from tofu import (
    parsers as tofu_parsers,
    consumers as tofu_consumers
)
from tofu.sources import (
    ftp_sources as tofu_sources,
    service_sources as tofu_svc_sources,
)

logfile = '/data/shared/cron_logs/20180615_ams_coupons.log'
import logging;logging.basicConfig(
    filename=logfile,
    level=logging.DEBUG,
    format=' '.join([
        '[%(asctime)s]',
        '[%(process)d]',
        '[%(name)s]',
        '[%(levelname)s]',
        '%(message)s']),
    )


### RUN COUPON CHAIN###
ams_coupons_chain = Chain(name='ams_coupons_chain', steps=[
    Step(
            name='mix_match_upc_mapping',
            source=tofu_sources.MixMatchUpcFTPSource(day=datetime(2018,6,14)), # update date to previous day
            parser=tofu_parsers.MixMatchUpcParser(header=True),
            consumer=bokchoy_consumers.CollectionsConsumer(
                ext_product_identifier='upc')
        ),
    Step(
            name='ams_coupons',
            source=tofu_svc_sources.AmsOffersSource(),
            parser=bokchoy_parsers.IdentityParser(),
            consumer=tofu_consumers.AmsOffersConsumer()
    )])
ams_coupons_chain()