Exemple #1
0
def maps():
    data = request.get_json()
    nodes = data['nodes']
    nodes_count = len(nodes)
    graph = [[-1 for j in range(nodes_count)] for i in range(nodes_count)]
    mapping = get_mapping(nodes)
    for street in data['streets']:
        graph[mapping[street['from']]][mapping[street['to']]] = 1
    for i in range(nodes_count):
        graph[i][i] = 0
    intermediary_info = compute_routes(graph)
    model_data = {
        'name': data['name'],
        'mapping': json.dumps(mapping),
        'graph': json.dumps(data['streets']),
        'intermediary_info': json.dumps(intermediary_info)
    }
    try:
        name = data['name']
        city = City.get(City.name == name)
        for k, v in model_data.items():
            setattr(city, k, v)
        city.save()
    except DoesNotExist:
        City.create(**model_data)
    return 'ok'
Exemple #2
0
from models import City


def chunks(l, n):
    """Yield successive n-sized chunks from l."""
    for i in range(0, len(l), n):
        yield l[i:i + n]


with open('locations.txt', 'r') as myfile:
    data = chunks(myfile.read().split('\r')[5:], 5)

for row in data:
    print row[0]
    City.create(rank=int(row[0]),
                city=row[1],
                country=row[2],
                tourists_millions=float(row[3]),
                ig_link=row[4])