help me with the symposion schedule grid

rooms.csv

schedutils.py

#!/usr/bin/env python
import os
import argparse
import csv
from datetime import datetime
import time

#from symposion.schedule.models import Day, Room, Schedule, Slot, SlotKind, SlotRoom
#from symposion.proposals.models import ProposalBase
#from symposion.schedule.timetable import TimeTable

"""
This will create schedule objects to build a schedule grid.

In order for this to work a Schedule and associated Days must already exist.

python schedutils.py --csv <csvfile> --section <slug> --modeltype <presentations|rooms|slotkinds>

presentations csv headers

date,time_start,time_end,kind,room,proposal_id,description

rooms csv headers
section_slug,name,order

slotkinds csv headers
section_slug,slot_kind,presentation_menu

This script is like running manage.py. The DJANGO_SETTINGS_MODULE environment
variable needs to be defined, or the default value of "scipy2014.settings"
must be in the python path.


This is sloppy. It was expendient. Backup everything before doing anything.
This doesn't do things in transactions so that you can do rollbacks on failures.
if it fails, you just have to nuke it from orbit and start over.

"""

def create_presentation_slots(schedule, args):
    """ Creates a SlotRoom based on a csv file

    Room, SlotKind, Day must exist
    If a Slot does not exist it is created
    """

    data = get_csv_data(args.csv)

    for row in data:
        date, start, end, kind_label, room_name, proposal_id = parse_presentation(row)
        print row

        room = Room.objects.get(schedule=schedule, name=room_name)
        slotkind = SlotKind.objects.get(schedule=schedule, label=kind_label)
        day = Day.objects.get(schedule=schedule, date=date)

        slot, created = Slot.objects.get_or_create(
            kind=slotkind,
            day=day,
            start=start,
            end=end,
        )
        SlotRoom.objects.create(slot=slot, room=room)


def create_rooms(args):
    data = get_csv_data(args.csv)
    for row in data:
        schedule = get_schedule(row['section_slug'])
        Room.objects.create(
            schedule = schedule,
            name=row['name'],
            order=row['order'],
        )


def create_slot_kinds(args):
    data = get_csv_data(args.csv)
    for row in data:
        schedule = get_schedule(row['section_slug'])
        presentation = row['presentation_menu'] == 'TRUE'
        SlotKind.objects.create(
            schedule=schedule,
            label=row['slot_kind'],
            presentation=presentation
        )

def parse_presentation(row):
    """ parses row from presentation csv file  
    """
    # date,time_start,time_end,kind,room,proposal_id,description
    date = datetime.strptime(row['date'], "%m/%d/%Y")
    start = get_time(row['time_start'])
    end = get_time(row['time_end'])
    return date, start, end, row['kind'], row['room'], row.get('proposal_id', '')


def get_time(timestring):
    time_obj = time.strptime(timestring, '%H:%M')
    time_obj = datetime(100, 1, 1, time_obj.tm_hour, time_obj.tm_min, 00)
    return time_obj.time()


def get_csv_data(csvfile):
    reader = csv.DictReader(csvfile)
    data = [dict((k.strip(), v.strip()) for k, v in x.items()) for x in reader]
    return data


def get_schedule(section_slug):
    """ get Schedule for section
    """
    section = Section.objects.get(slug=section_slug)
    return Schedule.objects.get(section=section)


def build_parser():
    parser = argparse.ArgumentParser(description="""
    """)
    parser.add_argument('--modeltype', default='presentations')
    parser.add_argument('--csv', type=file)
    parser.add_argument('--section', default='talks-posters', help='section slug for the schedule')
    return parser



if __name__ == '__main__':
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "scipy2014.settings")
    from symposion.schedule.models import Day, Room, Schedule, Slot, SlotKind, SlotRoom
    from symposion.conference.models import Section
    parser = build_parser()
    args = parser.parse_args()

    if args.modeltype ==  'presentations':
        schedule = get_schedule(args.section)
        create_presentation_slots(schedule, args)
    elif args.modeltype == 'rooms':
        create_rooms(args)
    elif args.modeltype == 'slotkinds':
        create_slot_kinds(args)
    else:
        print("invalid modeltype")

slotkinds.csv

talksposters.csv