"""
Created on Mon May 23 15:42:32 2022
@author: Lluís Vila UPC
"""

def bikeD_gen(id_city_D, id_bike_D):
    bikeD = {}
    for idn, bike in id_bike_D.items():
        if idn in id_city_D:
            if bike not in bikeD:
                bikeD[bike] = [idn]
            else:
                bikeD[bike].append(idn)   
    return bikeD

def cityD_gen(id_city_D, id_bikes_D, brand):
    cityD = {}
    for idn, bikeL in id_bikes_D.items():
        if brand in bikeL:
            if idn in id_city_D:
                city = id_city_D[idn]
            else:
                city = 'Unknown'
            if city not in cityD:
                cityD[city] = [idn]
            else:
                cityD[city].append(idn)
    for l in cityD.values():
        l.sort()
    return cityD