dummy_data_generator.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 53,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import random\n",
    "import string\n",
    "from datetime import datetime, timedelta\n",
    "\n",
    "import numpy as np"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 54,
   "metadata": {},
   "outputs": [],
   "source": [
    "def generate_random_address():\n",
    "    \"\"\"Generate a random address-like string.\"\"\"\n",
    "    return \"\".join(random.choices(string.ascii_letters + string.digits, k=10))\n",
    "\n",
    "\n",
    "def generate_random_datetime(start, end):\n",
    "    \"\"\"Generate a random datetime between 'start' and 'end'.\"\"\"\n",
    "    return start + timedelta(\n",
    "        seconds=random.randint(0, int((end - start).total_seconds()))\n",
    "    )\n",
    "\n",
    "\n",
    "def generate_votes(max_projects, max_votes):\n",
    "    \"\"\"Generate a random votes array.\"\"\"\n",
    "    num_votes = random.randint(1, max_votes)\n",
    "    projects = random.sample(range(max_projects), num_votes)\n",
    "    return [\n",
    "        {\"amount\": str(random.randint(1000, 1000000)), \"projectId\": f\"proj{proj_id}\"}\n",
    "        for proj_id in projects\n",
    "    ]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 55,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Set parameters\n",
    "num_rows = 20\n",
    "max_projects_in_ballot = 2\n",
    "max_votes = 1\n",
    "start_date = datetime(2023, 9, 1)\n",
    "end_date = datetime(2023, 12, 1)\n",
    "\n",
    "# set seed\n",
    "random.seed(42)\n",
    "\n",
    "# Generate data\n",
    "data = []\n",
    "for _ in range(num_rows):\n",
    "    has_published = random.choice([True, False])\n",
    "    has_voted = True if has_published else random.choice([True, False])\n",
    "    created_at = generate_random_datetime(start_date, end_date)\n",
    "    updated_at = generate_random_datetime(created_at, end_date)\n",
    "    published_at = (\n",
    "        generate_random_datetime(updated_at, end_date) if has_published else None\n",
    "    )\n",
    "\n",
    "    votes = generate_votes(max_projects_in_ballot, max_votes) if has_published else []\n",
    "\n",
    "    projects_in_ballot = (\n",
    "        len(votes)\n",
    "        # random.randint(1, max_projects_in_ballot) if has_published else None\n",
    "    )\n",
    "\n",
    "    row = {\n",
    "        \"Address\": generate_random_address(),\n",
    "        \"Has voted\": has_voted,\n",
    "        \"Has published\": has_published,\n",
    "        \"Published at\": published_at,\n",
    "        \"Created at\": created_at,\n",
    "        \"Updated at\": updated_at,\n",
    "        \"Projects in ballot\": projects_in_ballot,\n",
    "        \"Votes\": votes,\n",
    "    }\n",
    "    data.append(row)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 56,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Address</th>\n",
       "      <th>Has voted</th>\n",
       "      <th>Has published</th>\n",
       "      <th>Published at</th>\n",
       "      <th>Created at</th>\n",
       "      <th>Updated at</th>\n",
       "      <th>Projects in ballot</th>\n",
       "      <th>Votes</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>TP3fAbnFbm</td>\n",
       "      <td>True</td>\n",
       "      <td>True</td>\n",
       "      <td>2023-11-21 02:25:59</td>\n",
       "      <td>2023-09-03 10:16:45</td>\n",
       "      <td>2023-11-14 10:13:01</td>\n",
       "      <td>1</td>\n",
       "      <td>[{'amount': '147316', 'projectId': 'proj0'}]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>r1VjArnVgx</td>\n",
       "      <td>True</td>\n",
       "      <td>False</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2023-10-14 14:43:58</td>\n",
       "      <td>2023-11-12 05:16:37</td>\n",
       "      <td>0</td>\n",
       "      <td>[]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>YTH8xIZM1J</td>\n",
       "      <td>False</td>\n",
       "      <td>False</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2023-10-29 14:47:01</td>\n",
       "      <td>2023-11-11 10:58:21</td>\n",
       "      <td>0</td>\n",
       "      <td>[]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>ogrNwwmq6O</td>\n",
       "      <td>True</td>\n",
       "      <td>True</td>\n",
       "      <td>2023-11-14 04:23:37</td>\n",
       "      <td>2023-09-05 10:46:42</td>\n",
       "      <td>2023-11-08 15:38:00</td>\n",
       "      <td>1</td>\n",
       "      <td>[{'amount': '897865', 'projectId': 'proj0'}]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>q5QnuVdYXy</td>\n",
       "      <td>True</td>\n",
       "      <td>True</td>\n",
       "      <td>2023-11-28 17:47:00</td>\n",
       "      <td>2023-10-22 20:38:10</td>\n",
       "      <td>2023-11-27 06:07:49</td>\n",
       "      <td>1</td>\n",
       "      <td>[{'amount': '398887', 'projectId': 'proj0'}]</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "      Address  Has voted  Has published        Published at  \\\n",
       "0  TP3fAbnFbm       True           True 2023-11-21 02:25:59   \n",
       "1  r1VjArnVgx       True          False                 NaT   \n",
       "2  YTH8xIZM1J      False          False                 NaT   \n",
       "3  ogrNwwmq6O       True           True 2023-11-14 04:23:37   \n",
       "4  q5QnuVdYXy       True           True 2023-11-28 17:47:00   \n",
       "\n",
       "           Created at          Updated at  Projects in ballot  \\\n",
       "0 2023-09-03 10:16:45 2023-11-14 10:13:01                   1   \n",
       "1 2023-10-14 14:43:58 2023-11-12 05:16:37                   0   \n",
       "2 2023-10-29 14:47:01 2023-11-11 10:58:21                   0   \n",
       "3 2023-09-05 10:46:42 2023-11-08 15:38:00                   1   \n",
       "4 2023-10-22 20:38:10 2023-11-27 06:07:49                   1   \n",
       "\n",
       "                                          Votes  \n",
       "0  [{'amount': '147316', 'projectId': 'proj0'}]  \n",
       "1                                            []  \n",
       "2                                            []  \n",
       "3  [{'amount': '897865', 'projectId': 'proj0'}]  \n",
       "4  [{'amount': '398887', 'projectId': 'proj0'}]  "
      ]
     },
     "execution_count": 56,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Create DataFrame\n",
    "df = pd.DataFrame(data)\n",
    "df.head()  # Display the first few rows of the DataFrame"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 57,
   "metadata": {},
   "outputs": [],
   "source": [
    "df.to_csv(\"data/dummy_data_rpgf3.csv\", index=False)  # Save the DataFrame to a CSV file"
   ]
  }
 ],
 "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.9.15"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}