o jEi% @sdZddlZddlZddlmZddlZddlZddlmZddl m Z ddl m Z m Z eeZdededBfd d Zejd d d dedededefddZejdddeefddZdS)zU Celery tasks for interacting with the Amazon SP-API and managing user restrictions. N)datetime) celery_app)check_restrictions)DB_PATHget_all_user_credentials refresh_tokenreturnc Cs4tdtd}td}d}t||gstddSd|||d}z:t}|j||d }| | }Wdn1sDwY| d }|rXtd |WStd WdStj y}} ztd | j jWYd} ~ dSd} ~ wty} ztjd| ddWYd} ~ dSd} ~ ww)z Refreshes the SP-API access token using the refresh token. This is a helper function to be used within the Celery task. zBAttempting to refresh SP-API access token from within Celery task.SP_API_CLIENT_IDSP_API_CLIENT_SECRETz$https://api.amazon.com/auth/o2/tokenzSz3check_all_restrictions_for_user..Nz(No deals found in the database to check.zNo deals to check.rr4ERROR) is_restricted approval_urlr>r2z INSERT OR REPLACE INTO user_restrictions (user_id, asin, is_restricted, approval_url, last_checked_timestamp) VALUES (?, ?, ?, ?, ?) r?z%Progress: Saved restriction data for /z ASINs for user_id: z0Successfully finished restriction check for all z3Database error in check_all_restrictions_for_user: rzAAn unexpected error occurred in check_all_restrictions_for_user: Completed restriction check for z ASINs for user .)rrr*rsqlite3connectrcursorexecutefetchallwarningrangelenritemsrutcnowcommitErrorr!)selfr-r.rr auth_failedconnrErK BATCH_SIZEtotal_processedi batch_itemsresultsitemr4resultis_restricted_valr'r(r(r)check_all_restrictions_for_user=sv             "  rZz4keepa_deals.sp_api_tasks.check_restriction_for_asins)r+asinsc Cs|sdStdt|dt}|stddS|D]}|d}|d}zt|}|s:td|d Wqg}tt)}| }d d gt|} | d | d |dd| D}Wdn1smwYdd|D} |D]} | | vr| | ddq{d} tdt|| D]X} || | | }t|||}tt;}| }|D]&\} }d}|ddurd}n|ddkrd}| d|| ||dtfq|Wdn1swYqtdt|d|Wqtjy}ztjd|d|dd WYd}~qd}~wty;}ztjd!|d|dd WYd}~qd}~wwd"t|d#t|d$S)%z Celery task to check restrictions for a list of new ASINs against all connected users. Triggered when new deals are added to the database. zNo new ASINs to check.zStarting restriction check for z new ASINs.zJNo users have connected their SP-API accounts. Skipping restriction check.zNo connected users.r-rz!Could not refresh token for user z . Skipping.z, ?z1SELECT ASIN, Condition FROM deals WHERE ASIN IN ()cSr0r1r(r6r(r(r)r9r:z/check_restriction_for_asins..NcSsh|]}|dqS)r4r()r7rWr(r(r) sz.check_restriction_for_asins..r3r;rr>Tr2r<z INSERT OR REPLACE INTO user_restrictions (user_id, asin, is_restricted, approval_url, last_checked_timestamp) VALUES (?, ?, ?, ?, ?) r?z(Successfully saved restriction data for z new ASINs for user_id: z7Database error in check_restriction_for_asins for user z: rzEAn unexpected error occurred in check_restriction_for_asins for user rAz ASINs for z users.)rrrJrr*rHrCrDrrEjoinrFrGappendrIrrKrrLrMrNrr!)r[user_credentialscredsr-rrrKrQrE placeholders found_asinsr4rRrTrUrVrXrYr'r(r(r)check_restriction_for_asinssv        &&re)__doc__rCloggingrrrworkerrcelerykeepa_deals.amazon_sp_apirkeepa_deals.db_utilsrr getLogger__name__rstrr*taskrZlistrer(r(r(r)s     + X