Пример #1
0
def get():
    result = None
    biz_dict = {}
    loc_dict = {}
    result = sql_adapter.fetchall(GET_ALL_QUERY)

    if result:
        for row in result:
            if row.id not in biz_dict.keys():
                # biz is not in dict yet
                biz_dict[row.id] = Business(
                    business_id=row.id,
                    name=row.name,
                    desc=row.description,
                    price_level=row.price_level
                )

            loc = Location(
                loc_id=row.loc_id,
                lat=row.lat,
                lon=row.lon,
                address=row.address,
                neighborhood=row.neighborhood,
                website=row.website,
                google_rating=row.google_rating,
                phone=row.phone,
                active=row.active,
                perm_closed=row.perm_closed
            )
            biz_dict[row.id].add_location(loc)

            # lets build a location dict for the hours
            # for easy lookup
            if row.loc_id not in loc_dict:
                loc_dict[row.loc_id] = []

            loc_dict[row.loc_id].append(Hours(
                weekday=row.weekday,
                open_time=row.open_time,
                close_time=row.close_time,
                hour_id=row.hour_id
            ))

            coup = Coupon(
                coupon_id=row.coupon_id,
                desc=row.coup_desc,
                restrictions=row.restrictions,
                coup_type=row.coup_type
            )

            biz_dict[row.id].add_coupon(coup)

        for k,v in biz_dict.items():
            for loc in v.locations:
                if loc.loc_id in loc_dict.keys():
                    loc.add_hours(loc_dict[loc.loc_id])


        biz_list = [b.serialize() for b in biz_dict.values()]
    return jsonify(biz_list)
Пример #2
0
def update_biz():
    result = {}
    try:
        req = request.json
        bus = Business(
            business_id=req.get('business_id'),
            name=req.get('name'),
            desc=req.get('desc'),
            price_level=req.get('price_level')
        )
        bus.validate()

        bus = bus.serialize()
        sql_adapter.update(UPDATE_BUS_QUERY % bus)
        result['success'] = True
    except Exception as ex:
        print("Failed to update business object: %s" % ex)
        result['success'] = False
        result['error'] = str(ex)
    return jsonify(result)