o Q?h5@sddlZddlmZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z ddlmZmZdd lmZdd lmZdd lmZd d ZddZddZddZddZdefddZddZddZddZdd Z d!d"Z!dS)#N)datetime) PROJECT_ID)db)getAppCredentials)getEnterpriseById) API_ErrorpriorityLogger) apiVersion createHeader) getShopById)PLANS)ShopcCs|d}|d}|d}|d|d}}t|}|dd}|d}t|tvr2d |nd } t|} | r=| S|sDt|d }tt|d tt d d dt dd dd | rxt| d| d| d| d| d|dndd} t |d} d| d<|ddtd} tj| tt| dtt| d| d}|jd kr|d!} td"|| | St|j|j|)#NshopId currentUsershopuid enterpriseIdnamePODplanId22zNAlxdaZG6hKf6vexv6ljqHyP8i1Tz - Fulfillment Service Fz+https://riverr-enterprise-integrations-dot-z%.uc.r.appspot.com/shopify/fulfillmentjsonaddress1address2cityzip country_code)rrrrrr)rinventory_managementtracking_support callback_urlrequires_shipping_methodformatfulfillment_orders_opt_inlocation accessTokenzapplication/jsonz Content-Typeurl /admin/api/z/fulfillment_services.json)fulfillment_service)datarheadersr(fulfillmentServices)getrrr getFulfillmentServicer dictstrintrnow timestamprr r requestspostrdumps status_coder collectiondocumentsetrtext)r)rrrrr enterpriserrappfulfillmentServiceservicer*r&resrAR/var/www/html/riverr-enterprise-integrations-main/functions/Shopify/Fulfillment.pycreateFulfillmentServicesN      * rCcCs&td|}|jr|SdSNr,)rr8r9r-existsto_dictrrefrArArBr.9sr.cCs6|dtd}tj||d}|jr|dSgS)Nr'z//fulfillment_services.json?scope=current_clientr*fulfillment_services)r r4r-okr)r&r*r@rArArBgetFulfillmentServices>srLcCsR|dtd}t||dd}tj|t||d}|jdkr!dSt|j|dS)Nr'z/inventory_levels/connect.jsonT) location_idinventory_item_idrelocate_if_necessary)r)r*r+F) r r/r4r5rr6r7printr;)r&r*itemId locationIdr)r@rArArBconnectionInventoryLocationDs  rScCsXt|d}|ddtd}t|||d}tj|||d}|jdkr*|SdS)Nr%r&r'z/inventory_levels.json)rNrM available)rr*r+)r r-r r/r4r5r7r)rrMrQquantityr*r&r)r@rArArB addInventoryLs rVrc Cst|j}t|j||}d}t|dkr|dd}nt|jdd|td|t j |jdt d|t t t ||dd d |igd d d }|j dkrWtd||Stdd|j||jjdt|jdS)NridsubmitShopifyShipmentzNo fulfillment order found zNo fulfillment order foundr'z/fulfillments.json)numbercompanyTfulfillment_order_id) tracking_infonotify_customerline_items_by_fulfillment_order) fulfillmentr*rr+zShipment submitted to ShopifyNone)platformOrderIdr))r r%getFulfillmentOrdersr&lenr-rrrPr4r5r r/r7rr;requestbody)rrb carrierNametrackingNumberr*fulfillmentOrdersfulfillmentOrderIdr@rArArBrXUs4      rXcCsRtj|dtd|d|d}|jdkr|dStdd|j|||d dS) Nr'z/orders/z/fulfillment_orders.jsonrIfulfillment_ordersrarc)r&r*rb)r4r-r r7rrr;)r&r*rbr@rArArBrcs rccCs.td|}|jr|j|jSdSrD)rr8r9r-rE referencedeleterW)shoIdrHrArArBdeleteFulfillmentServices  rpcCs&td|}|jr|SdS)NzfulfillmentServices/)rr9r-rErFrGrArArBgetFulfillmentLocationsrqc st|}t|}|s dS|dt|j}t|j|d}fdd|D}g}|D]}|jd|dd}tj||d} | | j q+|S)NrWrIcsg|] }|dkr|qS)rW)r-).0lcurrentLocationIdrArB sz3removeWrongFulfillmentLocations..z(/admin/api/2024-04/fulfillment_services/z.json) r r-rqr r%rLr&r4rnappendr;) rrcurrentLocationr* locationslocationsToRemove responseslocationToRemover&r@rArtrBremoveWrongFulfillmentLocationss   r}c Csztddd|}t|}td||s!td|dtdt||s5td|dWd St|d }|d }|d }t|}|D]p}| }|d } |dt d} t| d| d} t j | |d| id} | j dkrtd| d| jqJ| didid} | dkrtd| dqJd| d}t j | |d|id} | j dkrtd| d| jqJ| dididg}|std | dqJ|d!d"}d#|d$|d"d%}t j | |d|id}|j dkr td&|d|jqJtd'|d"||did(id)g}|r0td*|d+|qJtd,|d-|d"dtd.d/d|j}|sXtd0|jdqJ|d! }d1| d2|d3|d4d5|d6d7|d"d8 }t j | |d|id}|j dkrtd9| d|jqJ|did:id)g}|rtd;| d+|qJtd<| dqJWd Sty}ztd=d }~ww)>z Use Shopify's GraphQL API to check order shipment status, move fulfillment orders, and submit shipment details if not shipped. ordersrz==fulfillment_locationzNo orders found for shop .zOrders: z'No fulfillment location found for shop NTr&r%rbr'z /graphql.jsonz> { order(id: "gid://shopify/Order/zW") { fulfillmentStatus } } queryr`rkz*Failed to fetch shipment status for order z . Response: r)orderfulfillmentStatusshippedzOrder z is already shipped.z") { fulfillmentOrders { id } } } z-Failed to fetch fulfillment orders for order riz&No fulfillment orders found for order rrWzW mutation { moveFulfillmentOrder( id: "z>", newLocationId: "gid://shopify/Location/a" ) { movedFulfillmentOrder { id } userErrors { field message } } } z!Failed to move fulfillment order z$Orders moved to fulfillment locationmoveFulfillmentOrder userErrorsz&Errors while moving fulfillment order z: z%Successfully moved fulfillment order z to location shipmentsorderIdz$No shipment details found for order zw mutation { fulfillmentCreateV2(input: { orderId: "gid://shopify/Order/zd", lineItemsByFulfillmentOrder: [{ fulfillmentOrderId: "z", fulfillmentLineItems: [] }], trackingInfo: { number: "rhz%", company: "rgzQ" }, locationId: "gid://shopify/Location/a " }) { fulfillment { id } userErrors { field message } } } z$Failed to submit shipment for order fulfillmentCreateV2z+Errors while submitting shipment for order z*Successfully submitted shipment for order z/Failed to move orders to fulfillment locations.)rr8wherer-rqrPrdr r rFr r4r5r7r;rrW Exceptionr)rr~rrshop_url access_tokenr*r order_dataplatform_order_id graphql_urlquery_order_shipmentresponseshipment_statusquery_fulfillment_ordersrlr[move_fulfillment_order_mutation move_response user_errorsshipment shipment_datasubmit_shipment_mutationshipment_responseerArArB#move_orders_to_fulfillment_locations                  {r)"rrr4 configs.envrconfigs.firebaserfunctions.Applicationsrfunctions.Enterprisesrfunctions.Responserrfunctions.Shopify.Authr r functions.Shopsr V2.functions.Shops.Authr V2.functions.Shops.mainr rCr.rLrSrVrXrcrprqr}rrArArArBs.        ' -