Making_Example_DB.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Making an example NEO4J db for the talk"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 41,
   "metadata": {},
   "outputs": [],
   "source": [
    "import random"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Creating the nodes data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "metadata": {},
   "outputs": [],
   "source": [
    "resource_codes = []\n",
    "scale = 1\n",
    "for i in range(0,50*scale):\n",
    "    resource_codes.append(\"Res_\" + str(i))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 43,
   "metadata": {},
   "outputs": [],
   "source": [
    "group_codes = []\n",
    "for i in range(0,250*scale):\n",
    "    group_codes.append(\"G_\" + str(i))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 44,
   "metadata": {},
   "outputs": [],
   "source": [
    "person_codes = []\n",
    "for i in range(0,1500*scale):\n",
    "    person_codes.append(\"P_\" + str(i))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Creating the edges randomly"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 45,
   "metadata": {},
   "outputs": [],
   "source": [
    "person_is_in_group = []\n",
    "\n",
    "\n",
    "# Assign every person to a group\n",
    "for person in person_codes:\n",
    "    person_is_in_group.append([person,random.choice(group_codes)])\n",
    "    \n",
    "# Assign a 50% of people to a random set of groups:\n",
    "\n",
    "for person in random.sample(person_codes, int(round(len(person_codes)/2))):\n",
    "    for extra_group in range(0,random.randint(1, 5)):\n",
    "        person_is_in_group.append([person,random.choice(group_codes)])\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 46,
   "metadata": {},
   "outputs": [],
   "source": [
    "group_child_of_group = []\n",
    "\n",
    "for i in range(0, len(group_codes)):\n",
    "    group_child_of_group.append([random.choice(group_codes),random.choice(group_codes)])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 47,
   "metadata": {},
   "outputs": [],
   "source": [
    "group_has_access_resource = []\n",
    "\n",
    "for resource in resource_codes:\n",
    "    group_has_access_resource.append([random.choice(group_codes),resource])\n",
    "\n",
    "# Assign a 5% to an extra group to simulate reality :\n",
    "\n",
    "for resource in random.sample(resource_codes, int(round(len(resource_codes)/20))):\n",
    "    group_has_access_resource.append([random.choice(group_codes),resource])"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Insert the data into Neo4j"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 48,
   "metadata": {},
   "outputs": [],
   "source": [
    "from neo4j import GraphDatabase\n",
    "\n",
    "uri = \"bolt://localhost:7687\"\n",
    "driver = GraphDatabase.driver(uri, auth=(\"neo4j\", \"q1w2\"))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Importing the nodes into NEO4J"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 49,
   "metadata": {},
   "outputs": [],
   "source": [
    "with driver.session() as session:\n",
    "    tx = session.begin_transaction()\n",
    "    for resource in resource_codes:\n",
    "        tx.run(\"CREATE (r:Resource {code:'\"+resource+\"'}) RETURN r.name\")\n",
    "    tx.commit()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 50,
   "metadata": {},
   "outputs": [],
   "source": [
    "with driver.session() as session:\n",
    "    tx = session.begin_transaction()\n",
    "    for group in group_codes:\n",
    "        tx.run(\"CREATE (g:Group {code:'\"+group+\"'}) RETURN g.name\")\n",
    "    tx.commit()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 51,
   "metadata": {},
   "outputs": [],
   "source": [
    "with driver.session() as session:\n",
    "    tx = session.begin_transaction()\n",
    "    for person in person_codes:\n",
    "        tx.run(\"CREATE (p:Person {code:'\"+person+\"'}) RETURN p.name\")\n",
    "    tx.commit()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Importing the edges into NEO4J"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 52,
   "metadata": {},
   "outputs": [],
   "source": [
    "with driver.session() as session:\n",
    "    tx = session.begin_transaction()\n",
    "    for edge in person_is_in_group:\n",
    "        tx.run(\n",
    "            \"\"\"\n",
    "            MATCH (p:Person {code:'\"\"\"+edge[0]+\"\"\"'})\n",
    "            MATCH (g:Group {code:'\"\"\"+edge[1]+\"\"\"'})\n",
    "            CREATE (p)-[:IS_IN]->(g) RETURN p,g\n",
    "            \"\"\"\n",
    "            \n",
    "        )\n",
    "    tx.commit()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 53,
   "metadata": {},
   "outputs": [],
   "source": [
    "with driver.session() as session:\n",
    "    tx = session.begin_transaction()\n",
    "    for edge in group_child_of_group:\n",
    "        tx.run(\n",
    "            \"\"\"\n",
    "            MATCH (g1:Group {code:'\"\"\"+edge[0]+\"\"\"'})\n",
    "            MATCH (g2:Group {code:'\"\"\"+edge[1]+\"\"\"'})\n",
    "            CREATE (g1)-[:CHILD_OF]->(g2) RETURN g1,g2\n",
    "            \"\"\"\n",
    "            \n",
    "        )\n",
    "    tx.commit()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 54,
   "metadata": {},
   "outputs": [],
   "source": [
    "with driver.session() as session:\n",
    "    tx = session.begin_transaction()\n",
    "    for edge in group_has_access_resource:\n",
    "        tx.run(\n",
    "            \"\"\"\n",
    "            MATCH (g:Group {code:'\"\"\"+edge[0]+\"\"\"'})\n",
    "            MATCH (r:Resource {code:'\"\"\"+edge[1]+\"\"\"'})\n",
    "            CREATE (g)-[:HAS_ACCESS]->(r) RETURN g,r\n",
    "            \"\"\"\n",
    "            \n",
    "        )\n",
    "    tx.commit()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Find person with the most power\n",
    "match (p:Person)--(:Group)--(r:Resource) return p,count(r) as rescount order by rescount desc limit 1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}