planner.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "dfcb71de-de58-4429-ae7c-14dae3d7ab0f",
   "metadata": {},
   "source": [
    "# Planner calculations"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 57,
   "id": "67f28ada-01ad-4ce9-8892-0bba53f0bd1c",
   "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>WBS</th>\n",
       "      <th>ID</th>\n",
       "      <th>Name</th>\n",
       "      <th>Duration (Workdays)</th>\n",
       "      <th>Start</th>\n",
       "      <th>End</th>\n",
       "      <th>Milestone</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>KO</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2025-04-01</td>\n",
       "      <td>2025-04-01</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "      <td>Analysis and Design</td>\n",
       "      <td>80.0</td>\n",
       "      <td>2025-04-01</td>\n",
       "      <td>2025-07-21</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2.1</td>\n",
       "      <td>3</td>\n",
       "      <td>Review of current FM (TRL9)</td>\n",
       "      <td>20.0</td>\n",
       "      <td>2025-04-01</td>\n",
       "      <td>2025-04-28</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2.2</td>\n",
       "      <td>4</td>\n",
       "      <td>Mechanical and thermal redesign</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-04-28</td>\n",
       "      <td>2025-06-09</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2.3</td>\n",
       "      <td>5</td>\n",
       "      <td>Electrical redesign</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-04-28</td>\n",
       "      <td>2025-06-09</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>2.4</td>\n",
       "      <td>6</td>\n",
       "      <td>Wearing design</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-06-09</td>\n",
       "      <td>2025-07-21</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>2.5</td>\n",
       "      <td>7</td>\n",
       "      <td>SRR (System Requirement Review)</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2025-06-09</td>\n",
       "      <td>2025-06-09</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>3</td>\n",
       "      <td>8</td>\n",
       "      <td>Manufacturing and Testing EM</td>\n",
       "      <td>90.0</td>\n",
       "      <td>2025-06-09</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>3.1</td>\n",
       "      <td>9</td>\n",
       "      <td>EM manufacturing (TRL7)</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-06-09</td>\n",
       "      <td>2025-07-21</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>3.2</td>\n",
       "      <td>10</td>\n",
       "      <td>Battery module EQM manufacturing (TRL6)</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-06-09</td>\n",
       "      <td>2025-07-21</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>3.3</td>\n",
       "      <td>11</td>\n",
       "      <td>FW development</td>\n",
       "      <td>60.0</td>\n",
       "      <td>2025-07-21</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>3.4</td>\n",
       "      <td>12</td>\n",
       "      <td>Testing</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-07-21</td>\n",
       "      <td>2025-09-01</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>3.5</td>\n",
       "      <td>13</td>\n",
       "      <td>MRR (Manufacturing Readiness Reviev)</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2025-09-01</td>\n",
       "      <td>2025-09-01</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>4</td>\n",
       "      <td>14</td>\n",
       "      <td>Manufacturing and Testing PFM</td>\n",
       "      <td>180.0</td>\n",
       "      <td>2025-07-21</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>4.1</td>\n",
       "      <td>15</td>\n",
       "      <td>PFM manufacturing</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-09-01</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>4.2</td>\n",
       "      <td>16</td>\n",
       "      <td>GM/ TrM</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>4.3</td>\n",
       "      <td>17</td>\n",
       "      <td>Wearing accessories manufacturing</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-07-21</td>\n",
       "      <td>2025-09-01</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>4.4</td>\n",
       "      <td>18</td>\n",
       "      <td>Transport package design</td>\n",
       "      <td>14.0</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>2025-10-31</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>4.5</td>\n",
       "      <td>19</td>\n",
       "      <td>Transport package manufacturing</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-10-31</td>\n",
       "      <td>2025-12-12</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>4.6</td>\n",
       "      <td>20</td>\n",
       "      <td>Calibration</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-11-24</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>4.7</td>\n",
       "      <td>21</td>\n",
       "      <td>Testing</td>\n",
       "      <td>57.0</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>2026-03-26</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>4.8</td>\n",
       "      <td>22</td>\n",
       "      <td>Qualification documentation</td>\n",
       "      <td>60.0</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>4.9</td>\n",
       "      <td>23</td>\n",
       "      <td>QAR (Qualification and Acceptance Review)</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>4.10</td>\n",
       "      <td>24</td>\n",
       "      <td>FM</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>5</td>\n",
       "      <td>25</td>\n",
       "      <td>Dosimetry Display Unit (DDU) demonstrator</td>\n",
       "      <td>260.0</td>\n",
       "      <td>2025-04-01</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>5.1</td>\n",
       "      <td>26</td>\n",
       "      <td>Review of current prototype (TRL6)</td>\n",
       "      <td>20.0</td>\n",
       "      <td>2025-04-01</td>\n",
       "      <td>2025-04-28</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>5.2</td>\n",
       "      <td>27</td>\n",
       "      <td>DDU electrical redesign &amp; Dosimeter extra PCB ...</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-09-01</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>5.3</td>\n",
       "      <td>28</td>\n",
       "      <td>DDU mechanical redesign</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-09-01</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>5.4</td>\n",
       "      <td>29</td>\n",
       "      <td>DDU FW redesign</td>\n",
       "      <td>60.0</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>5.5</td>\n",
       "      <td>30</td>\n",
       "      <td>DDU EQM Manufacturing</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-10-13</td>\n",
       "      <td>2025-11-24</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>5.6</td>\n",
       "      <td>31</td>\n",
       "      <td>DDU EQM</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>5.7</td>\n",
       "      <td>32</td>\n",
       "      <td>DDU Testing</td>\n",
       "      <td>60.0</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>5.8</td>\n",
       "      <td>33</td>\n",
       "      <td>DDU FM</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>6</td>\n",
       "      <td>34</td>\n",
       "      <td>Operations Preparation</td>\n",
       "      <td>258.0</td>\n",
       "      <td>2025-04-28</td>\n",
       "      <td>2026-04-23</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34</th>\n",
       "      <td>6.1</td>\n",
       "      <td>35</td>\n",
       "      <td>Detector response simulation</td>\n",
       "      <td>60.0</td>\n",
       "      <td>2025-04-28</td>\n",
       "      <td>2025-07-21</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>6.2</td>\n",
       "      <td>36</td>\n",
       "      <td>Development of Hp calculation method</td>\n",
       "      <td>60.0</td>\n",
       "      <td>2025-09-01</td>\n",
       "      <td>2025-11-24</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>6.3</td>\n",
       "      <td>37</td>\n",
       "      <td>Hp calculation method verification</td>\n",
       "      <td>50.0</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>2026-03-16</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>37</th>\n",
       "      <td>6.4</td>\n",
       "      <td>38</td>\n",
       "      <td>Operating manual &amp; Worksheets</td>\n",
       "      <td>40.0</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>2026-03-02</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>38</th>\n",
       "      <td>6.5</td>\n",
       "      <td>39</td>\n",
       "      <td>Support of crew training</td>\n",
       "      <td>28.0</td>\n",
       "      <td>2026-03-02</td>\n",
       "      <td>2026-04-09</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>39</th>\n",
       "      <td>6.6</td>\n",
       "      <td>40</td>\n",
       "      <td>Payload integration</td>\n",
       "      <td>50.0</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>2026-03-16</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>40</th>\n",
       "      <td>6.7</td>\n",
       "      <td>41</td>\n",
       "      <td>Support to real time works</td>\n",
       "      <td>10.0</td>\n",
       "      <td>2026-04-09</td>\n",
       "      <td>2026-04-23</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>41</th>\n",
       "      <td>6.8</td>\n",
       "      <td>42</td>\n",
       "      <td>CoFR (Certificate of Flight Readiness)</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>2026-03-30</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>42</th>\n",
       "      <td>7</td>\n",
       "      <td>43</td>\n",
       "      <td>Outreach</td>\n",
       "      <td>281.0</td>\n",
       "      <td>2025-04-01</td>\n",
       "      <td>2026-04-29</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>43</th>\n",
       "      <td>7.1</td>\n",
       "      <td>44</td>\n",
       "      <td>Outreach planning</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-04-01</td>\n",
       "      <td>2025-05-12</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>44</th>\n",
       "      <td>7.2</td>\n",
       "      <td>45</td>\n",
       "      <td>Studens lectures/experiments</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-07-21</td>\n",
       "      <td>2025-09-01</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45</th>\n",
       "      <td>7.3</td>\n",
       "      <td>46</td>\n",
       "      <td>Outreach activities</td>\n",
       "      <td>221.0</td>\n",
       "      <td>2025-06-23</td>\n",
       "      <td>2026-04-29</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>46</th>\n",
       "      <td>7.4</td>\n",
       "      <td>47</td>\n",
       "      <td>Colaboration with ESERO</td>\n",
       "      <td>220.0</td>\n",
       "      <td>2025-06-23</td>\n",
       "      <td>2026-04-27</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>47</th>\n",
       "      <td>7.5</td>\n",
       "      <td>48</td>\n",
       "      <td>Support of PR materials creation</td>\n",
       "      <td>50.0</td>\n",
       "      <td>2026-01-05</td>\n",
       "      <td>2026-03-16</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>48</th>\n",
       "      <td>7.6</td>\n",
       "      <td>49</td>\n",
       "      <td>Web presentation</td>\n",
       "      <td>30.0</td>\n",
       "      <td>2025-05-12</td>\n",
       "      <td>2025-06-23</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>49</th>\n",
       "      <td>8</td>\n",
       "      <td>50</td>\n",
       "      <td>Project management</td>\n",
       "      <td>280.0</td>\n",
       "      <td>2025-04-01</td>\n",
       "      <td>2026-04-27</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>50</th>\n",
       "      <td>8.1</td>\n",
       "      <td>51</td>\n",
       "      <td>Management</td>\n",
       "      <td>280.0</td>\n",
       "      <td>2025-04-01</td>\n",
       "      <td>2026-04-27</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "     WBS  ID                                               Name  \\\n",
       "0      1   1                                                 KO   \n",
       "1      2   2                                Analysis and Design   \n",
       "2    2.1   3                        Review of current FM (TRL9)   \n",
       "3    2.2   4                    Mechanical and thermal redesign   \n",
       "4    2.3   5                                Electrical redesign   \n",
       "5    2.4   6                                     Wearing design   \n",
       "6    2.5   7                    SRR (System Requirement Review)   \n",
       "7      3   8                       Manufacturing and Testing EM   \n",
       "8    3.1   9                            EM manufacturing (TRL7)   \n",
       "9    3.2  10            Battery module EQM manufacturing (TRL6)   \n",
       "10   3.3  11                                     FW development   \n",
       "11   3.4  12                                            Testing   \n",
       "12   3.5  13               MRR (Manufacturing Readiness Reviev)   \n",
       "13     4  14                      Manufacturing and Testing PFM   \n",
       "14   4.1  15                                  PFM manufacturing   \n",
       "15   4.2  16                                            GM/ TrM   \n",
       "16   4.3  17                  Wearing accessories manufacturing   \n",
       "17   4.4  18                           Transport package design   \n",
       "18   4.5  19                    Transport package manufacturing   \n",
       "19   4.6  20                                        Calibration   \n",
       "20   4.7  21                                            Testing   \n",
       "21   4.8  22                        Qualification documentation   \n",
       "22   4.9  23          QAR (Qualification and Acceptance Review)   \n",
       "23  4.10  24                                                 FM   \n",
       "24     5  25          Dosimetry Display Unit (DDU) demonstrator   \n",
       "25   5.1  26                 Review of current prototype (TRL6)   \n",
       "26   5.2  27  DDU electrical redesign & Dosimeter extra PCB ...   \n",
       "27   5.3  28                            DDU mechanical redesign   \n",
       "28   5.4  29                                    DDU FW redesign   \n",
       "29   5.5  30                              DDU EQM Manufacturing   \n",
       "30   5.6  31                                            DDU EQM   \n",
       "31   5.7  32                                        DDU Testing   \n",
       "32   5.8  33                                             DDU FM   \n",
       "33     6  34                             Operations Preparation   \n",
       "34   6.1  35                       Detector response simulation   \n",
       "35   6.2  36               Development of Hp calculation method   \n",
       "36   6.3  37                 Hp calculation method verification   \n",
       "37   6.4  38                      Operating manual & Worksheets   \n",
       "38   6.5  39                           Support of crew training   \n",
       "39   6.6  40                                Payload integration   \n",
       "40   6.7  41                         Support to real time works   \n",
       "41   6.8  42             CoFR (Certificate of Flight Readiness)   \n",
       "42     7  43                                           Outreach   \n",
       "43   7.1  44                                  Outreach planning   \n",
       "44   7.2  45                       Studens lectures/experiments   \n",
       "45   7.3  46                                Outreach activities   \n",
       "46   7.4  47                            Colaboration with ESERO   \n",
       "47   7.5  48                   Support of PR materials creation   \n",
       "48   7.6  49                                   Web presentation   \n",
       "49     8  50                                 Project management   \n",
       "50   8.1  51                                         Management   \n",
       "\n",
       "    Duration (Workdays)      Start        End  Milestone  \n",
       "0                   0.0 2025-04-01 2025-04-01       True  \n",
       "1                  80.0 2025-04-01 2025-07-21      False  \n",
       "2                  20.0 2025-04-01 2025-04-28      False  \n",
       "3                  30.0 2025-04-28 2025-06-09      False  \n",
       "4                  30.0 2025-04-28 2025-06-09      False  \n",
       "5                  30.0 2025-06-09 2025-07-21      False  \n",
       "6                   0.0 2025-06-09 2025-06-09       True  \n",
       "7                  90.0 2025-06-09 2025-10-13      False  \n",
       "8                  30.0 2025-06-09 2025-07-21      False  \n",
       "9                  30.0 2025-06-09 2025-07-21      False  \n",
       "10                 60.0 2025-07-21 2025-10-13      False  \n",
       "11                 30.0 2025-07-21 2025-09-01      False  \n",
       "12                  0.0 2025-09-01 2025-09-01       True  \n",
       "13                180.0 2025-07-21 2026-03-30      False  \n",
       "14                 30.0 2025-09-01 2025-10-13      False  \n",
       "15                  0.0 2025-10-13 2025-10-13       True  \n",
       "16                 30.0 2025-07-21 2025-09-01      False  \n",
       "17                 14.0 2025-10-13 2025-10-31      False  \n",
       "18                 30.0 2025-10-31 2025-12-12      False  \n",
       "19                 30.0 2025-11-24 2026-01-05      False  \n",
       "20                 57.0 2026-01-05 2026-03-26      False  \n",
       "21                 60.0 2025-10-13 2026-01-05      False  \n",
       "22                  0.0 2026-03-30 2026-03-30       True  \n",
       "23                  0.0 2026-03-30 2026-03-30       True  \n",
       "24                260.0 2025-04-01 2026-03-30      False  \n",
       "25                 20.0 2025-04-01 2025-04-28      False  \n",
       "26                 30.0 2025-09-01 2025-10-13      False  \n",
       "27                 30.0 2025-09-01 2025-10-13      False  \n",
       "28                 60.0 2025-10-13 2026-01-05      False  \n",
       "29                 30.0 2025-10-13 2025-11-24      False  \n",
       "30                  0.0 2026-01-05 2026-01-05       True  \n",
       "31                 60.0 2026-01-05 2026-03-30      False  \n",
       "32                  0.0 2026-03-30 2026-03-30       True  \n",
       "33                258.0 2025-04-28 2026-04-23      False  \n",
       "34                 60.0 2025-04-28 2025-07-21      False  \n",
       "35                 60.0 2025-09-01 2025-11-24      False  \n",
       "36                 50.0 2026-01-05 2026-03-16      False  \n",
       "37                 40.0 2026-01-05 2026-03-02      False  \n",
       "38                 28.0 2026-03-02 2026-04-09      False  \n",
       "39                 50.0 2026-01-05 2026-03-16      False  \n",
       "40                 10.0 2026-04-09 2026-04-23      False  \n",
       "41                  0.0 2026-03-30 2026-03-30       True  \n",
       "42                281.0 2025-04-01 2026-04-29      False  \n",
       "43                 30.0 2025-04-01 2025-05-12      False  \n",
       "44                 30.0 2025-07-21 2025-09-01      False  \n",
       "45                221.0 2025-06-23 2026-04-29      False  \n",
       "46                220.0 2025-06-23 2026-04-27      False  \n",
       "47                 50.0 2026-01-05 2026-03-16      False  \n",
       "48                 30.0 2025-05-12 2025-06-23      False  \n",
       "49                280.0 2025-04-01 2026-04-27      False  \n",
       "50                280.0 2025-04-01 2026-04-27      False  "
      ]
     },
     "execution_count": 57,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Import necessary libraries\n",
    "import xml.etree.ElementTree as ET\n",
    "import pandas as pd\n",
    "\n",
    "# Function to parse tasks with WBS\n",
    "def parse_tasks_no_duplicates(tasks, parent_wbs=\"\", top_level_counter=[1], processed_ids=set()):\n",
    "    wbs_list = []\n",
    "    local_counter = 1\n",
    "\n",
    "    for task in tasks:\n",
    "        task_id = task.get(\"id\", \"\")\n",
    "        if task_id in processed_ids:  # Skip duplicates\n",
    "            continue\n",
    "        processed_ids.add(task_id)\n",
    "\n",
    "        name = task.get(\"name\", \"\")\n",
    "        duration = task.get(\"duration\", \"0\")\n",
    "        start = task.get(\"start\", \"\")[:8]  # Trimming to YYYYMMDD format\n",
    "        end = task.get(\"end\", \"\")[:8]     # Trimming to YYYYMMDD format\n",
    "\n",
    "        # Generate WBS\n",
    "        if parent_wbs:\n",
    "            current_wbs = f\"{parent_wbs}.{local_counter}\"\n",
    "            local_counter += 1\n",
    "        else:  # Top-level task\n",
    "            current_wbs = str(top_level_counter[0])\n",
    "            top_level_counter[0] += 1\n",
    "\n",
    "        # Append task data\n",
    "        wbs_list.append({\n",
    "            \"WBS\": current_wbs,\n",
    "            \"ID\": task_id,\n",
    "            \"Name\": name,\n",
    "            \"Duration (Workdays)\": round(int(duration) / (8 * 3600),0),\n",
    "            \"Start\": start,\n",
    "            \"End\": end,\n",
    "            \"Milestone\": int(duration) == 0\n",
    "        })\n",
    "\n",
    "        # Recurse into child tasks\n",
    "        child_tasks = task.findall(\"./task\")\n",
    "        if child_tasks:\n",
    "            wbs_list.extend(parse_tasks_no_duplicates(\n",
    "                child_tasks, parent_wbs=current_wbs, top_level_counter=top_level_counter, processed_ids=processed_ids\n",
    "            ))\n",
    "\n",
    "    return wbs_list\n",
    "\n",
    "# Load XML file\n",
    "xml_file_path = \"CZPAD.planner\"  # Replace with the path to your file\n",
    "tree = ET.parse(xml_file_path)\n",
    "root = tree.getroot()\n",
    "\n",
    "# Parse tasks and create a DataFrame\n",
    "root_tasks = root.findall(\".//task\")\n",
    "parsed_tasks = parse_tasks_no_duplicates(root_tasks)\n",
    "tasks_df = pd.DataFrame(parsed_tasks)\n",
    "\n",
    "# Convert Start and End to datetime\n",
    "tasks_df[\"Start\"] = pd.to_datetime(tasks_df[\"Start\"], format=\"%Y%m%d\")\n",
    "tasks_df[\"End\"] = pd.to_datetime(tasks_df[\"End\"], format=\"%Y%m%d\")\n",
    "\n",
    "# Display the resulting DataFrame (optional)\n",
    "tasks_df\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 58,
   "id": "3e270921-4e6f-4546-8ccf-8f7588ae2a4d",
   "metadata": {},
   "outputs": [],
   "source": [
    "tasks_df.to_csv('tasks.csv')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 48,
   "id": "8175dec5-430a-42c4-881c-54ab7e4b827a",
   "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>WBS</th>\n",
       "      <th>EE</th>\n",
       "      <th>PROJ</th>\n",
       "      <th>UST</th>\n",
       "      <th>PR</th>\n",
       "      <th>ME</th>\n",
       "      <th>SCI</th>\n",
       "      <th>FW</th>\n",
       "      <th>SUP</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>task-id</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2.1</td>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2.2</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>2.3</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>2.4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>3.1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>3.2</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>3.3</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>60</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>3.4</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>4.1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>4.3</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>4.4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>14</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>4.5</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>4.6</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>4.7</td>\n",
       "      <td>46</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>46</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>4.8</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>60</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>5.1</td>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>5.2</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>5.3</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>5.4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>60</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>5.5</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>5.7</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>6.1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>60</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>6.2</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>60</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>37</th>\n",
       "      <td>6.3</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>40</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>38</th>\n",
       "      <td>6.4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>20</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>39</th>\n",
       "      <td>6.5</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>14</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>40</th>\n",
       "      <td>6.6</td>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>41</th>\n",
       "      <td>6.7</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>44</th>\n",
       "      <td>7.1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45</th>\n",
       "      <td>7.2</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>46</th>\n",
       "      <td>7.3</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>199</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>47</th>\n",
       "      <td>7.4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>22</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>48</th>\n",
       "      <td>7.5</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>49</th>\n",
       "      <td>7.6</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>30</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>51</th>\n",
       "      <td>8.1</td>\n",
       "      <td>0</td>\n",
       "      <td>280</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "         WBS  EE  PROJ  UST   PR  ME  SCI  FW  SUP\n",
       "task-id                                           \n",
       "3        2.1  10     0    0    0  10   10   0    0\n",
       "4        2.2   0     0    0    0  30    0   0    0\n",
       "5        2.3  30     0    0    0   0    0   0    0\n",
       "6        2.4   0     0    0    0  30    0   0    0\n",
       "9        3.1   0     0   30    0   0    0   0    0\n",
       "10       3.2   0     0   30    0   0    0   0    0\n",
       "11       3.3   0     0    0    0   0    0  60    0\n",
       "12       3.4  30     0    0    0  30    0   0    0\n",
       "15       4.1   0     0   30    0   0    0   0    0\n",
       "17       4.3   0     0   30    0   0    0   0    0\n",
       "18       4.4   0     0    0    0  14    0   0    0\n",
       "19       4.5   0     0   30    0   0    0   0    0\n",
       "20       4.6   0     0    0    0   0   30   0    0\n",
       "21       4.7  46     0    0    0  46    0   0    0\n",
       "22       4.8   0     0    0    0   0    0   0   60\n",
       "26       5.1  10     0    0    0  10   10   0    0\n",
       "27       5.2  30     0    0    0   0    0   0    0\n",
       "28       5.3   0     0    0    0  30    0   0    0\n",
       "29       5.4   0     0    0    0   0    0  60    0\n",
       "30       5.5   0     0   30    0   0    0   0    0\n",
       "32       5.7   0     0    0    0   0    0   0   30\n",
       "35       6.1   0     0    0    0   0   60   0    0\n",
       "36       6.2   0     0    0    0   0   60   0    0\n",
       "37       6.3   0     0    0    0   0   40   0    0\n",
       "38       6.4   0     0    0    0   0    0   0   20\n",
       "39       6.5   0     0    0    0   0    0   0   14\n",
       "40       6.6  10     0    0    0  10    0   0    0\n",
       "41       6.7   0     0    0    0   0   10   0   10\n",
       "44       7.1   0     0    0   30   0    0   0    0\n",
       "45       7.2   0     0    0    0   0   30   0    0\n",
       "46       7.3   0     0    0  199   0    0   0    0\n",
       "47       7.4   0     0    0   22   0    0   0    0\n",
       "48       7.5   0     0    0    0   0   10   0    0\n",
       "49       7.6   0     0    0   30   0    0   0    0\n",
       "51       8.1   0   280    0    0   0    0   0    0"
      ]
     },
     "execution_count": 48,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import xml.etree.ElementTree as ET\n",
    "import pandas as pd\n",
    "\n",
    "# Load the XML file\n",
    "xml_file_path = \"CZPAD.planner\"\n",
    "tree = ET.parse(xml_file_path)\n",
    "root = tree.getroot()\n",
    "\n",
    "# Parse tasks from XML\n",
    "def parse_tasks_with_wbs(tasks, parent_wbs=\"\", top_level_counter=[1], processed_ids=set()):\n",
    "    wbs_list = []\n",
    "    local_counter = 1\n",
    "\n",
    "    for task in tasks:\n",
    "        task_id = task.get(\"id\", \"\")\n",
    "        if task_id in processed_ids:  # Skip duplicates\n",
    "            continue\n",
    "        processed_ids.add(task_id)\n",
    "\n",
    "        name = task.get(\"name\", \"\")\n",
    "        duration = task.get(\"duration\", \"0\")\n",
    "        start = task.get(\"start\", \"\")[:8]  # Trimming to YYYYMMDD format\n",
    "        end = task.get(\"end\", \"\")[:8]     # Trimming to YYYYMMDD format\n",
    "\n",
    "        # Generate WBS\n",
    "        if parent_wbs:\n",
    "            current_wbs = f\"{parent_wbs}.{local_counter}\"\n",
    "            local_counter += 1\n",
    "        else:  # Top-level task\n",
    "            current_wbs = str(top_level_counter[0])\n",
    "            top_level_counter[0] += 1\n",
    "\n",
    "        # Append task data\n",
    "        wbs_list.append({\n",
    "            \"task-id\": task_id,\n",
    "            \"name\": name,\n",
    "            \"duration\": duration,\n",
    "            \"start\": start,\n",
    "            \"end\": end,\n",
    "            \"WBS\": current_wbs\n",
    "        })\n",
    "\n",
    "        # Recurse into child tasks\n",
    "        child_tasks = task.findall(\"./task\")\n",
    "        if child_tasks:\n",
    "            wbs_list.extend(parse_tasks_with_wbs(\n",
    "                child_tasks, parent_wbs=current_wbs, top_level_counter=top_level_counter, processed_ids=processed_ids\n",
    "            ))\n",
    "\n",
    "    return wbs_list\n",
    "\n",
    "# Parse tasks\n",
    "root_tasks = root.findall(\".//task\")\n",
    "tasks_data = parse_tasks_with_wbs(root_tasks)\n",
    "tasks_with_wbs_df = pd.DataFrame(tasks_data)\n",
    "\n",
    "# Parse allocations and calculate workdays\n",
    "def parse_allocations_with_durations(root, tasks_df):\n",
    "    task_duration_map = tasks_df.set_index(\"task-id\")[\"duration\"].astype(float) / (8 * 3600)  # Convert seconds to workdays\n",
    "    \n",
    "    allocations = []\n",
    "    for allocation in root.findall(\".//allocation\"):\n",
    "        task_id = allocation.get(\"task-id\")\n",
    "        resource_id = allocation.get(\"resource-id\")\n",
    "        units = float(allocation.get(\"units\", \"0\")) / 100.0  # Convert percentage to fraction\n",
    "        \n",
    "        if task_id in task_duration_map.index:\n",
    "            allocations.append({\n",
    "                \"task-id\": task_id,\n",
    "                \"resource-id\": resource_id,\n",
    "                \"units\": units,\n",
    "                \"workdays\": round(units * task_duration_map[task_id],0)  # Calculate resource workdays\n",
    "            })\n",
    "\n",
    "    return pd.DataFrame(allocations)\n",
    "\n",
    "# Parse allocations with durations\n",
    "allocations_with_workdays_df = parse_allocations_with_durations(root, tasks_with_wbs_df)\n",
    "\n",
    "# Pivot the table to have resource-ids as columns\n",
    "allocations_pivot = allocations_with_workdays_df.pivot_table(\n",
    "    index=\"task-id\", \n",
    "    columns=\"resource-id\", \n",
    "    values=\"workdays\", \n",
    "    fill_value=0\n",
    ")\n",
    "\n",
    "# Rename columns based on short-name\n",
    "resources_df = pd.DataFrame([\n",
    "    {\"resource-id\": res.get(\"id\"), \"short-name\": res.get(\"short-name\")}\n",
    "    for res in root.findall(\".//resource\")\n",
    "])\n",
    "resource_map = resources_df.set_index(\"resource-id\")[\"short-name\"].to_dict()\n",
    "allocations_pivot.columns = [resource_map.get(col, f\"Resource-{col}\") for col in allocations_pivot.columns]\n",
    "\n",
    "# Add WBS to the table\n",
    "allocations_pivot[\"WBS\"] = tasks_with_wbs_df.set_index(\"task-id\")[\"WBS\"]\n",
    "\n",
    "# Reorder columns to place WBS at the beginning\n",
    "columns = [\"WBS\"] + [col for col in allocations_pivot.columns if col != \"WBS\"]\n",
    "allocations_df = allocations_pivot[columns]\n",
    "\n",
    "# Ensure task-id is numeric and sort\n",
    "allocations_df = allocations_df.reset_index()\n",
    "allocations_df[\"task-id\"] = allocations_df[\"task-id\"].astype(int)\n",
    "allocations_df = allocations_df.sort_values(by=\"task-id\").set_index(\"task-id\")\n",
    "\n",
    "# The resulting dataframe is named `allocations_df`\n",
    "allocations_df\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2dbfe1e0-5cf7-46a2-abdc-e606df552454",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 49,
   "id": "806a9c71-f749-4be5-879f-e0482131648a",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "           EE  PROJ  UST   PR  ME  SCI  FW  SUP\n",
      "WBS_Group                                      \n",
      "2          40     0    0    0  70   10   0    0\n",
      "3          30     0   60    0  30    0  60    0\n",
      "4          46     0   90    0  60   30   0   60\n",
      "5          40     0   30    0  40   10  60   30\n",
      "6          10     0    0    0  10  170   0   44\n",
      "7           0     0    0  281   0   40   0    0\n",
      "8           0   280    0    0   0    0   0    0\n"
     ]
    }
   ],
   "source": [
    "# Extract the first digit of the WBS and create a grouping column\n",
    "allocations_df[\"WBS_Group\"] = allocations_df[\"WBS\"].str.split('.').str[0]\n",
    "\n",
    "# Group by the first digit of the WBS and sum rows for each group\n",
    "grouped_allocations = allocations_df.groupby(\"WBS_Group\").sum(numeric_only=True)\n",
    "\n",
    "# Drop the grouping column if not needed in the final output\n",
    "grouped_allocations = grouped_allocations.drop(columns=[\"WBS_Group\"], errors=\"ignore\")\n",
    "\n",
    "# Display the grouped DataFrame\n",
    "print(grouped_allocations)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 55,
   "id": "0f5a6f5f-d1da-4cd5-bcc1-927e0201b615",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "    team    team-leader           team-member  FTE Company\n",
      "0     EE   Jakub Kakona          Jakub Kakona  1.0     UST\n",
      "1     EE   Jakub Kakona  Electrical Engineer2  1.0     UST\n",
      "2     ME   Roman Dvorak          Roman Dvorak  1.0     UST\n",
      "3     ME   Roman Dvorak  Mechanical Engineer2  0.5     UST\n",
      "4    SCI    Ondrej Ploc           Ondrej Ploc  0.5     NPI\n",
      "5    SCI    Ondrej Ploc         Iva Ambrozova  0.5     NPI\n",
      "6    SCI    Ondrej Ploc         Martin Kakona  0.8     NPI\n",
      "7    SCI    Ondrej Ploc           Jakub Slegl  0.5     NPI\n",
      "8     FW   Vit Hanousek          Vit Hanousek  1.0     UST\n",
      "9     FW   Vit Hanousek          Programator2  0.5     UST\n",
      "10   SUP  Iva Ambrozova           Technician1  0.5     NPI\n",
      "11   SUP  Iva Ambrozova           Technician2  0.5     NPI\n",
      "12  PROJ  Iva Ambrozova         Iva Ambrozova  0.3     NPI\n",
      "13  PROJ  Iva Ambrozova                Roxana  0.5     NPI\n",
      "14   UST   Jakub Kakona                   NaN  NaN     NaN\n",
      "15    PR  Iva Ambrozova            PR manager  0.2     NPI\n"
     ]
    }
   ],
   "source": [
    "import pandas as pd\n",
    "\n",
    "# Load the CSV file into a DataFrame\n",
    "file_path = \"team_structure.csv\"\n",
    "team = pd.read_csv(file_path)\n",
    "\n",
    "# Display the DataFrame for verification\n",
    "print(team)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "f5def4b2-7ce8-4090-a47f-484cc36726f8",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "8b04583e-ec1d-4d89-a168-4d82cbcc6a9b",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 51,
   "id": "152309bd-43a6-4577-a807-8247f3064498",
   "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>WBS_Group</th>\n",
       "      <th>Team</th>\n",
       "      <th>Team Member</th>\n",
       "      <th>Workdays</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2</td>\n",
       "      <td>EE</td>\n",
       "      <td>Jakub Kakona</td>\n",
       "      <td>40.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2</td>\n",
       "      <td>EE</td>\n",
       "      <td>Electrical Engineer2</td>\n",
       "      <td>40.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2</td>\n",
       "      <td>ME</td>\n",
       "      <td>Roman Dvorak</td>\n",
       "      <td>70.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2</td>\n",
       "      <td>ME</td>\n",
       "      <td>Mechanical Engineer2</td>\n",
       "      <td>35.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Ondrej Ploc</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>2</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Iva Ambrozova</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>2</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Martin Kakona</td>\n",
       "      <td>8.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>2</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Jakub Slegl</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>3</td>\n",
       "      <td>EE</td>\n",
       "      <td>Jakub Kakona</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>3</td>\n",
       "      <td>EE</td>\n",
       "      <td>Electrical Engineer2</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>3</td>\n",
       "      <td>UST</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>3</td>\n",
       "      <td>ME</td>\n",
       "      <td>Roman Dvorak</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>3</td>\n",
       "      <td>ME</td>\n",
       "      <td>Mechanical Engineer2</td>\n",
       "      <td>15.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>3</td>\n",
       "      <td>FW</td>\n",
       "      <td>Vit Hanousek</td>\n",
       "      <td>60.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>3</td>\n",
       "      <td>FW</td>\n",
       "      <td>Programator2</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>4</td>\n",
       "      <td>EE</td>\n",
       "      <td>Jakub Kakona</td>\n",
       "      <td>46.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>4</td>\n",
       "      <td>EE</td>\n",
       "      <td>Electrical Engineer2</td>\n",
       "      <td>46.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>4</td>\n",
       "      <td>UST</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>4</td>\n",
       "      <td>ME</td>\n",
       "      <td>Roman Dvorak</td>\n",
       "      <td>60.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>4</td>\n",
       "      <td>ME</td>\n",
       "      <td>Mechanical Engineer2</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>4</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Ondrej Ploc</td>\n",
       "      <td>15.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>4</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Iva Ambrozova</td>\n",
       "      <td>15.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>4</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Martin Kakona</td>\n",
       "      <td>24.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>4</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Jakub Slegl</td>\n",
       "      <td>15.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>4</td>\n",
       "      <td>SUP</td>\n",
       "      <td>Technician1</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>4</td>\n",
       "      <td>SUP</td>\n",
       "      <td>Technician2</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>5</td>\n",
       "      <td>EE</td>\n",
       "      <td>Jakub Kakona</td>\n",
       "      <td>40.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>5</td>\n",
       "      <td>EE</td>\n",
       "      <td>Electrical Engineer2</td>\n",
       "      <td>40.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>5</td>\n",
       "      <td>UST</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>5</td>\n",
       "      <td>ME</td>\n",
       "      <td>Roman Dvorak</td>\n",
       "      <td>40.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>5</td>\n",
       "      <td>ME</td>\n",
       "      <td>Mechanical Engineer2</td>\n",
       "      <td>20.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>5</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Ondrej Ploc</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>5</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Iva Ambrozova</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>5</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Martin Kakona</td>\n",
       "      <td>8.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34</th>\n",
       "      <td>5</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Jakub Slegl</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>5</td>\n",
       "      <td>FW</td>\n",
       "      <td>Vit Hanousek</td>\n",
       "      <td>60.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>5</td>\n",
       "      <td>FW</td>\n",
       "      <td>Programator2</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>37</th>\n",
       "      <td>5</td>\n",
       "      <td>SUP</td>\n",
       "      <td>Technician1</td>\n",
       "      <td>15.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>38</th>\n",
       "      <td>5</td>\n",
       "      <td>SUP</td>\n",
       "      <td>Technician2</td>\n",
       "      <td>15.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>39</th>\n",
       "      <td>6</td>\n",
       "      <td>EE</td>\n",
       "      <td>Jakub Kakona</td>\n",
       "      <td>10.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>40</th>\n",
       "      <td>6</td>\n",
       "      <td>EE</td>\n",
       "      <td>Electrical Engineer2</td>\n",
       "      <td>10.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>41</th>\n",
       "      <td>6</td>\n",
       "      <td>ME</td>\n",
       "      <td>Roman Dvorak</td>\n",
       "      <td>10.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>42</th>\n",
       "      <td>6</td>\n",
       "      <td>ME</td>\n",
       "      <td>Mechanical Engineer2</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>43</th>\n",
       "      <td>6</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Ondrej Ploc</td>\n",
       "      <td>85.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>44</th>\n",
       "      <td>6</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Iva Ambrozova</td>\n",
       "      <td>85.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45</th>\n",
       "      <td>6</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Martin Kakona</td>\n",
       "      <td>136.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>46</th>\n",
       "      <td>6</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Jakub Slegl</td>\n",
       "      <td>85.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>47</th>\n",
       "      <td>6</td>\n",
       "      <td>SUP</td>\n",
       "      <td>Technician1</td>\n",
       "      <td>22.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>48</th>\n",
       "      <td>6</td>\n",
       "      <td>SUP</td>\n",
       "      <td>Technician2</td>\n",
       "      <td>22.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>49</th>\n",
       "      <td>7</td>\n",
       "      <td>PR</td>\n",
       "      <td>PR manager</td>\n",
       "      <td>56.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>50</th>\n",
       "      <td>7</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Ondrej Ploc</td>\n",
       "      <td>20.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>51</th>\n",
       "      <td>7</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Iva Ambrozova</td>\n",
       "      <td>20.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>52</th>\n",
       "      <td>7</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Martin Kakona</td>\n",
       "      <td>32.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>53</th>\n",
       "      <td>7</td>\n",
       "      <td>SCI</td>\n",
       "      <td>Jakub Slegl</td>\n",
       "      <td>20.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>54</th>\n",
       "      <td>8</td>\n",
       "      <td>PROJ</td>\n",
       "      <td>Iva Ambrozova</td>\n",
       "      <td>84.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>55</th>\n",
       "      <td>8</td>\n",
       "      <td>PROJ</td>\n",
       "      <td>Roxana</td>\n",
       "      <td>140.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   WBS_Group  Team           Team Member  Workdays\n",
       "0          2    EE          Jakub Kakona      40.0\n",
       "1          2    EE  Electrical Engineer2      40.0\n",
       "2          2    ME          Roman Dvorak      70.0\n",
       "3          2    ME  Mechanical Engineer2      35.0\n",
       "4          2   SCI           Ondrej Ploc       5.0\n",
       "5          2   SCI         Iva Ambrozova       5.0\n",
       "6          2   SCI         Martin Kakona       8.0\n",
       "7          2   SCI           Jakub Slegl       5.0\n",
       "8          3    EE          Jakub Kakona      30.0\n",
       "9          3    EE  Electrical Engineer2      30.0\n",
       "10         3   UST                   NaN       NaN\n",
       "11         3    ME          Roman Dvorak      30.0\n",
       "12         3    ME  Mechanical Engineer2      15.0\n",
       "13         3    FW          Vit Hanousek      60.0\n",
       "14         3    FW          Programator2      30.0\n",
       "15         4    EE          Jakub Kakona      46.0\n",
       "16         4    EE  Electrical Engineer2      46.0\n",
       "17         4   UST                   NaN       NaN\n",
       "18         4    ME          Roman Dvorak      60.0\n",
       "19         4    ME  Mechanical Engineer2      30.0\n",
       "20         4   SCI           Ondrej Ploc      15.0\n",
       "21         4   SCI         Iva Ambrozova      15.0\n",
       "22         4   SCI         Martin Kakona      24.0\n",
       "23         4   SCI           Jakub Slegl      15.0\n",
       "24         4   SUP           Technician1      30.0\n",
       "25         4   SUP           Technician2      30.0\n",
       "26         5    EE          Jakub Kakona      40.0\n",
       "27         5    EE  Electrical Engineer2      40.0\n",
       "28         5   UST                   NaN       NaN\n",
       "29         5    ME          Roman Dvorak      40.0\n",
       "30         5    ME  Mechanical Engineer2      20.0\n",
       "31         5   SCI           Ondrej Ploc       5.0\n",
       "32         5   SCI         Iva Ambrozova       5.0\n",
       "33         5   SCI         Martin Kakona       8.0\n",
       "34         5   SCI           Jakub Slegl       5.0\n",
       "35         5    FW          Vit Hanousek      60.0\n",
       "36         5    FW          Programator2      30.0\n",
       "37         5   SUP           Technician1      15.0\n",
       "38         5   SUP           Technician2      15.0\n",
       "39         6    EE          Jakub Kakona      10.0\n",
       "40         6    EE  Electrical Engineer2      10.0\n",
       "41         6    ME          Roman Dvorak      10.0\n",
       "42         6    ME  Mechanical Engineer2       5.0\n",
       "43         6   SCI           Ondrej Ploc      85.0\n",
       "44         6   SCI         Iva Ambrozova      85.0\n",
       "45         6   SCI         Martin Kakona     136.0\n",
       "46         6   SCI           Jakub Slegl      85.0\n",
       "47         6   SUP           Technician1      22.0\n",
       "48         6   SUP           Technician2      22.0\n",
       "49         7    PR            PR manager      56.0\n",
       "50         7   SCI           Ondrej Ploc      20.0\n",
       "51         7   SCI         Iva Ambrozova      20.0\n",
       "52         7   SCI         Martin Kakona      32.0\n",
       "53         7   SCI           Jakub Slegl      20.0\n",
       "54         8  PROJ         Iva Ambrozova      84.0\n",
       "55         8  PROJ                Roxana     140.0"
      ]
     },
     "execution_count": 51,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Initialize a list to store results\n",
    "workdays_per_member = []\n",
    "\n",
    "# Iterate over each WBS_Group\n",
    "for wbs_group, row in grouped_allocations.iterrows():\n",
    "    # Iterate over each team\n",
    "    for team_name, total_workdays in row.items():\n",
    "        if total_workdays > 0:  # Check if there are workdays assigned\n",
    "            # Get the members of the current team\n",
    "            team_members = team[team[\"team\"] == team_name]\n",
    "            for _, member in team_members.iterrows():\n",
    "                # Calculate workdays for each member\n",
    "                member_workdays = round(total_workdays * member[\"FTE\"],0)\n",
    "                workdays_per_member.append({\n",
    "                    \"WBS_Group\": wbs_group,\n",
    "                    \"Team\": team_name,\n",
    "                    \"Team Member\": member[\"team-member\"],\n",
    "                    \"Workdays\": member_workdays\n",
    "                })\n",
    "\n",
    "# Convert to DataFrame\n",
    "workdays_per_member_df = pd.DataFrame(workdays_per_member)\n",
    "\n",
    "# Display specific results for Jakub Kakona in WBS_Group 2\n",
    "jakub_kakona_results = workdays_per_member_df[\n",
    "    (workdays_per_member_df[\"Team Member\"] == \"Jakub Kakona\") & \n",
    "    (workdays_per_member_df[\"WBS_Group\"] == \"2\")\n",
    "]\n",
    "\n",
    "workdays_per_member_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 52,
   "id": "8abb42f9-0495-4d39-b7d5-6e2904f3567c",
   "metadata": {},
   "outputs": [],
   "source": [
    "workdays_per_member_df.to_csv('workdays.csv')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 53,
   "id": "e6b8b909-7038-4ca9-8baf-57cea67fa6a4",
   "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>Team Member</th>\n",
       "      <th>Total Workdays</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Electrical Engineer2</td>\n",
       "      <td>166.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Iva Ambrozova</td>\n",
       "      <td>214.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Jakub Kakona</td>\n",
       "      <td>166.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Jakub Slegl</td>\n",
       "      <td>130.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Martin Kakona</td>\n",
       "      <td>208.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>Mechanical Engineer2</td>\n",
       "      <td>105.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>Ondrej Ploc</td>\n",
       "      <td>130.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>PR manager</td>\n",
       "      <td>56.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>Programator2</td>\n",
       "      <td>60.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>Roman Dvorak</td>\n",
       "      <td>210.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>Roxana</td>\n",
       "      <td>140.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>Technician1</td>\n",
       "      <td>67.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>Technician2</td>\n",
       "      <td>67.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>Vit Hanousek</td>\n",
       "      <td>120.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             Team Member  Total Workdays\n",
       "0   Electrical Engineer2           166.0\n",
       "1          Iva Ambrozova           214.0\n",
       "2           Jakub Kakona           166.0\n",
       "3            Jakub Slegl           130.0\n",
       "4          Martin Kakona           208.0\n",
       "5   Mechanical Engineer2           105.0\n",
       "6            Ondrej Ploc           130.0\n",
       "7             PR manager            56.0\n",
       "8           Programator2            60.0\n",
       "9           Roman Dvorak           210.0\n",
       "10                Roxana           140.0\n",
       "11           Technician1            67.0\n",
       "12           Technician2            67.0\n",
       "13          Vit Hanousek           120.0"
      ]
     },
     "execution_count": 53,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Group by team member and sum their workdays\n",
    "total_workdays_per_member = workdays_per_member_df.groupby(\"Team Member\")[\"Workdays\"].sum().reset_index()\n",
    "\n",
    "# Rename columns for clarity\n",
    "total_workdays_per_member.columns = [\"Team Member\", \"Total Workdays\"]\n",
    "\n",
    "# Display the resulting DataFrame\n",
    "total_workdays_per_member"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 54,
   "id": "fe1ad7d6-6093-4547-847c-2827311f4198",
   "metadata": {},
   "outputs": [],
   "source": [
    "total_workdays_per_member.to_csv('total_workdays.csv')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c8992cca-f809-4a68-b348-8922225b824a",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "768cb7ce-df43-4dec-8302-daf64603627b",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "ipython_py3"
  },
  "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.8.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}