Q2ML main program test.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import Packetizer\n",
    "from Packetizer import Bonus, Tossup, Packet #necessary classes\n",
    "import numpy as np\n",
    "import pandas as pd\n",
    "import re"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "def load_csv(fname):\n",
    "    '''\n",
    "    Loads in a QEMs CSV\n",
    "    Returns two lists: one for tossups and one for bonuses\n",
    "    '''\n",
    "    count = 0\n",
    "    tu_lines = []\n",
    "    b_lines = []\n",
    "    tu_bool = True #used to trakc if in tossup part of CSV\n",
    "    done_bool = False #boolean to mark if done reading CSV\n",
    "    \n",
    "    if fname[-4:] == '.csv':\n",
    "        temp_name = fname\n",
    "    else:\n",
    "        temp_name = fname + '.csv'\n",
    "    \n",
    "    try:\n",
    "        wp = open(temp_name) #file exists\n",
    "    except FileNotFoundError:\n",
    "        raise FileNotFoundError('CSV not found. Is {} a file in the current directory?'.format(temp_name))\n",
    "        \n",
    "    for line in wp:\n",
    "        if count > 0:\n",
    "            line = line.replace('Packet','')\n",
    "        #print(line.split(',')[0])\n",
    "        if line == '\\n': #new line char\n",
    "            if tu_bool == False:\n",
    "                done_bool = True\n",
    "                break\n",
    "            else:\n",
    "                tu_bool = False\n",
    "        if tu_bool == True and done_bool == False: #in tossups\n",
    "            tu_lines.append(line)\n",
    "        elif tu_bool == False and done_bool == False: #in bonuses and not done\n",
    "            b_lines.append(line)\n",
    "        count +=1\n",
    "    wp.close()\n",
    "    b_lines.pop(0) #remove first element of bonus list\n",
    "    return tu_lines, b_lines\n",
    "\n",
    "def make_tu_b_csvs(tu_list, b_list, title, sort = True):\n",
    "    '''\n",
    "    Makes two .csvs: one for tossups and one for bonuses\n",
    "    '''\n",
    "    #write to .csv for Tossup and bonus\n",
    "    wp_t = open(title + '_t.csv', 'w')\n",
    "    wp_b = open(title + '_b.csv', 'w')\n",
    "    wp_t.write('\\n'.join(tu_list))\n",
    "    wp_b.write('\\n'.join(b_list))\n",
    "    wp_t.close()\n",
    "    wp_b.close()\n",
    "    #Tossups\n",
    "    t_df = pd.read_csv(title + '_t.csv') #load in csv\n",
    "    t_df = t_df.drop(['Comments', 'Question Number'], axis = 1 ) #useless columns\n",
    "    t_df = t_df.fillna('0') #some values are np.nan so need to have a good value for them\n",
    "    if sort == True:\n",
    "        t_df = t_df.sort_values(by = ['Packet'])\n",
    "    #Bonuses\n",
    "    b_df = pd.read_csv(title + '_b.csv')\n",
    "    b_df = b_df.drop(['Comments', 'Question Number'], axis = 1 ) #useless columns\n",
    "    b_df = b_df.fillna('0') #some values are np.nan so need to have a good value for them\n",
    "    b_df = b_df.rename(columns = {'Unnamed: 10': 'Packet'}) #used in wrangling data\n",
    "    b_df = b_df.astype({'Packet':'int32'}, errors = 'ignore') #from str to int32\n",
    "    if sort == True:\n",
    "        b_df = b_df.sort_values(by = ['Packet'])\n",
    "    \n",
    "    return t_df, b_df\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "def add_tossups(packet_list, t_df, template_bool = False):\n",
    "    '''\n",
    "    Adds tossups to packets in packet_list\n",
    "    If tmeplate_bool == True, requires distribution loaded in for categorization\n",
    "    '''\n",
    "    for index, row in t_df.iterrows():\n",
    "        text = row['Tossup Question']\n",
    "        answer = row['Answer']\n",
    "        author = row['Author']\n",
    "        packet = row['Packet']\n",
    "        t_id = row['Id']\n",
    "        temp_tu = Tossup(text, answer, author, t_id, packet)\n",
    "        category = row['Category']\n",
    "        if template_bool == True:\n",
    "            pass #for now\n",
    "        packet_list[packet - 1].add(temp_tu.clean_html()) #goes from 1 to 0 indexing\n",
    "    return packet_list\n",
    "\n",
    "def add_bonues(packet_list, b_df, template_bool = False):\n",
    "    '''\n",
    "    Adds bonues to packets in packet_list\n",
    "    If tmeplate_bool == True, requires distribution loaded in for categorization\n",
    "    '''\n",
    "    for index, row in b_df.iterrows():\n",
    "        lead_in = row['Bonus Leadin']\n",
    "        q_list = [row['Bonus Part 1'], row['Bonus Part 2'], row['Bonus Part 3']]\n",
    "        answer_list = [row['Bonus Answer 1'] , row['Bonus Answer 2'], row['Bonus Answer 3']]\n",
    "        author = row['Author']\n",
    "        b_id = row['Id']\n",
    "        packet = row['Packet']\n",
    "        temp_b = Bonus(lead_in, q_list, answer_list, author, b_id, packet)\n",
    "        category = row['Category']\n",
    "        if template_bool == True:\n",
    "            pass #for now\n",
    "        packet_list[packet - 1].add(temp_b.clean_html())\n",
    "    return packet_list\n",
    "\n",
    "#need get category helper function"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "def main():\n",
    "    '''main program to generate HTML Packet files'''\n",
    "    #TITLE = input('Title of this set or project?')\n",
    "    tu_lines, b_lines = load_csv('./Demo/demo_shw.csv')\n",
    "    t_df, b_df = make_tu_b_csvs(tu_lines, b_lines, 'demo_shw') #TITLE will go in here\n",
    "    packet_list = []\n",
    "    #num_packets = int(input)('Number of packets:?') #must be int\n",
    "    num_packets = 16\n",
    "    for i in range(num_packets):\n",
    "        temp_packet = Packet([], [], i+1)\n",
    "        packet_list.append(temp_packet)\n",
    "    packet_list = add_tossups(packet_list, t_df) #add tossups\n",
    "    packet_list = add_bonues(packet_list, b_df)\n",
    "    for pack in packet_list:\n",
    "        pack.randomize()\n",
    "        pack.to_html() #dumps in current working directory as PacketNUMBER.html\n",
    "    \n",
    "    \n",
    "    print('program done')\n",
    "    return 0"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Packet 13 full of bonuses\n",
      "Packet 16 full of bonuses\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "randomized\n",
      "program done\n"
     ]
    }
   ],
   "source": [
    "if __name__ == '__main__':\n",
    "    main()"
   ]
  }
 ],
 "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.1"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}