local point crawl scrap

local_point_crawl.py

import csv
import os
import random
from itertools import chain

from graphviz import Graph


def local_point_crawl(points, point_count, connections, connection_count):
    dot = Graph(comment='Area Map')
    pairs = set()
    subset = random.sample(points, point_count)
    for thing in subset:
        for _ in range(connection_count):
            first = thing
            others = set(subset)
            others.remove(thing)
            second = random.choice(list(others))
            pair = sorted([first, second])
            pairs.add(tuple(pair))
    for pair in pairs:
        dot.edge(pair[0], pair[1], label=f"{random.choice(connections)}")
    dot.render('test-output/area_map.gv', view=True)
    print(dot.source)


if __name__ == "__main__":
    options = {}
    for root, dirs, files in os.walk(os.path.expanduser('points')):
        for file in sorted(files):
            if file.endswith("csv"):
                rows = [row[0] for row in csv.reader(open(os.path.join(root, file)))]
                name = file.split(".")[0]
                options[name] = rows
    local_point_crawl(list(chain(options["landmarks"], options['forest'])), 7,
                      options["forest_connections"], 2)