o Zh& @s2ddlmZddlmZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZdd lmZdd lmZmZdd lmZdd lmZdd lmZddlmZdefddZddlmZd!defddZd"dedefddZdefddZ ddlmZgddfdededeededef dd Z!dS)#)datetime)ListN) PROJECT_ID)db) API_Error) apiVersion createHeader)Product)Variant)ShopifyGraphQLClientvalidate_shopify_id)getShopifyProductVariants)Shop) Enterprise)Paramsparamsc Cs|jd}t|}t|j}td|td|jtdt|jdtd}tj||d}|j dkr[| d g}td |D]}td |d d |dd|dqC|Std|j dS)NshopIdheaderszshop.urlrz /admin/api/z/locations.json)r locationsu📍 Available Locations:z- namez | ID: idz | Active: activeu❌ Failed to fetch locations: ) argsgetrr accessTokenprinturlrrequests status_codejsontext)rrshoprrresrlocr%S/var/www/html/riverr-enterprise-integrations-main/V2/functions/Shopify/Locations.pygetShopLocationss       (r'clientc Cs|s t|jjj}td|j}|j r=| }|d}|r=d}d|i}| ||}|didi}|r=|St ||dS)aB Retrieves the fulfillment service for a given shop and validates if the location exists in Shopify. If the fulfillment service does not exist, it creates a new one. :param shop: ShopifyGraphQLClient instance :param shopId: The shop's ID :return: Fulfillment service data if valid, otherwise None fulfillmentServicesGraphQLraV query FulfillmentServiceShow($id: ID!) { fulfillmentService(id: $id) { id serviceName callbackUrl location { legacyResourceId id } } } datafulfillmentService)r(enterpriseName) rrr" enterpriseIdrr collectiondocumentrexiststo_dictexecutecreateFulfillmentService) r(r,reffulfillment_datafulfillment_idquery variablesresultr+r%r%r&getFulfillmentServiceOs    r:PODr,c Cs|d}dtd}d}||d}|||}|dididg}|r4td |tt|d |ddd }td |j  ||S) z Creates a new fulfillment service in Shopify via GraphQL and stores it in Firestore. :param shop: Shopify shop data z - Fulfillment Servicez+https://riverr-enterprise-integrations-dot-z%.uc.r.appspot.com/shopify/fulfillmenta1 mutation fulfillmentServiceCreate($name: String!, $callbackUrl: URL!) { fulfillmentServiceCreate(name: $name, callbackUrl: $callbackUrl, trackingSupport: true, inventoryManagement: true) { fulfillmentService { id serviceName callbackUrl location { id } } userErrors { field message } } } )r callbackUrlr*fulfillmentServiceCreate userErrorszFULFILLMENT SERVICE ERRORS: ir+r)) rr2rrrstrrr.r/rset) r(r,fulfillment_name callback_urlmutationr8r9errorsfulfillment_servicer%r%r&r3zs    r3cCs@|jd}t|}t|}t|j}|rt||jdSdS)Nr)r,)rrrr rr-r3r)rrr"r( enterpriser%r%r&createFulfillmentServiceRoutes   rG productIdvariants locationIdplatformProductIdc Cs |st|}|st|dd}td||s t|j}td|t||}g}t|D]\}}d|} | d| d|dd |d q0d d |d } | | } td| t|D]&\}}d|} | di| idg} | rtd|dd| qctddS)a Updates the fulfillment location of product variants in Shopify using the inventoryAdjustQuantities mutation. :param client: Shopify GraphQL client :param productId: The product ID :param variants: The product variants (optional) :param locationId: The location ID to update the inventory :param platformProductId: The Shopify platform product ID (optional) locationrz locationId:-zplatformProductId:-inventoryActivatez z&: inventoryActivate(inventoryItemId: "inventoryItemIdz", locationId: "z") { inventoryLevel { id } userErrors { field message } } z%mutation inventoryActivateMultiple {  z }zRESULT: r*r>u$❌ ERROR Activating Inventory Item z: u@✅ Fulfillment locations updated successfully for all variants!N) r getAllr:rrr rKr enumerateappendjoinr2) r(rHrIrJrKshopify_variantsmutation_partsidxvariantaliasrCr9rDr%r%r&updateFulfillmentLocationss:           rY)N)r;)"rtypingrr configs.envrconfigs.firebaserfunctions.Responserfunctions.Shopify.AuthrrV2.functions.Products.mainr V2.functions.Products.Variantsr V2.functions.Shopify.mainr r V2.functions.Shopify.Productsr V2.functions.Shops.mainrV2.functions.Users.mainr V2.Paramsrr'r:r?r3rGrYr%r%r%r&s*           9+: .