from configs.firebase import db

def saveLocation(supplierId, id, name, city, state, zip, country):
    location = dict(
        supplierid = supplierId,
        id= id,
        name=  name,
        city=city,
        state=state,
        zip=zip,
        country=country
    )
    db.collection('suppliers').document(supplierId).collection('locations').document(id).set(location, merge=True)
    return location

def getLocations(supplierId):
    locations =  db.collection('suppliers').document(supplierId).collection('locations').get()
    return [location.to_dict() for location in locations]