Creating a random set of Tinder users and pairing them up in dates

dates.csv

fake_tinder_user_data.py

from faker import Faker
import random

user_file = open("output/users.csv", "w")
user_file.write("id,name,date_of_birth")

dates_file = open("output/dates.csv", "w")
dates_file.write("a,b")


# Generate users
fake = Faker()

users = [] # Store users in this list
for _id in range(1, 101):
    is_male = random.random() > 0.5
    name = fake.name_male() if is_male else fake.name_female()
    date_of_birth = fake.date_of_birth(minimum_age=18, maximum_age=65)
    users.append({'id': _id, "name": name, "date_of_birth": date_of_birth})

random.shuffle(users)

remaining_users = users.copy()

while len(remaining_users) > 0:
    # Get the first user in the list
    user = remaining_users.pop(0)

    # Store the first user into the CSV
    user_file.write("\n")
    user_file.write(f"{user['id']},{user['name']},{user['date_of_birth']}")

    for i in range(random.randint(0, 2)):
        # Get a random user from the list
        random_user = random.choice(users)
        if random_user == user:
            continue

        # Store the relationship between the user and the random user
        dates_file.write("\n")
        dates_file.write(f"{user['id']},{random_user['id']}")


user_file.close()
dates_file.close()

load_csv_scripts.cypher

LOAD CSV WITH HEADERS FROM 'file:///users.csv' AS line
MERGE (p:Person {id: line.id, name: line.name, date_of_birth: line.date_of_birth})
RETURN p;


LOAD CSV WITH HEADERS FROM 'file:///dates.csv' AS line
MATCH (p1:Person {id: line.a})
MATCH (p2:Person {id: line.b})
CREATE (p1)-[:DATED]->(p2);

users.csv