yelp_fieldlists.py

BUSINESS    = "business"
REVIEW      = "review"
TIP         = "tip"
CHECKIN     = "checkin"
USER        = "user"

def getHeaders(_type):
    
    if(_type == BUSINESS):
        return ["name", "city", "review_count", "hours", "neighborhood", "categories", "latitude","stars","attributes","longitude","address","postal_code", "state", "is_open", "business_id", "type"];
    elif (_type == REVIEW):
        return ["review_id", "user_id", "date", "useful", "text", "cool", "stars", "business_id", "type", "funny"];
    elif (_type == TIP):
        return ["business_id", "user_id", "date", "text", "likes", "type"];
    elif (_type == CHECKIN):
        return ["business_id", "time", "type"];
    else:
        return ["user_id", "name", "review_count", "yelping_since", "friends", "useful", "funny", "cool", "fans", "elite", "average_stars", "compliment_hot", "compliment_more", "compliment_profile", "compliment_cute", "compliment_list", "compliment_note", "compliment_plain", "compliment_cool", "compliment_funny", "compliment_writer", "compliment_photos", "type"];
    
def getData(_type, _data):
    
    data_arr = []
    for i in getHeaders(_type):
        data_arr.append(_data['{}'.format(i)])
    
    return data_arr