from functions.Suppliers.Auth import getClient, getSupplierCredentials
from configs.firebase import db
from zeep import helpers

def getBlankImages(blankProduct:dict):
    supplierId = blankProduct.get('supplierId')
    client, version = getClient(supplierId, 'media')
    if blankProduct:
        id = blankProduct.get("blankProductId")
        creds = getSupplierCredentials(supplierId)
        imagesResponse = helpers.serialize_object(
            client.service.getMediaContent(
                wsVersion=version,
                id = creds.get('username'),
                password = creds.get('password'),
                mediaType='Image',
                productId=id,
                # partId=variant
            ), dict
        )
        images = imagesResponse.get("MediaContentArray", {}).get("MediaContent", [])
        images = [image for image in images if image.get("mediaType") == "Image"]
        return images
    return []


    
def getBlankProduct(id:str):
    ref = db.collection("blankProducts").document(id).get()
    if ref.exists:
        return ref.to_dict()
    return None