示例#1
0
def organization_presenters_create(auth_user, organization):
    with WriteSession() as session:
        with session.no_autoflush:
            org_presenter = Presenter.from_request(request.json)
            if session.query(Presenter).filter(Presenter.organization == organization, Presenter.slug == org_presenter.slug).first():
                abort(400, {'code': 400, 'message': 'That presenter already exists on this organization', 'data': {'presenter': org_presenter.to_json()}})
            if org_presenter.user and session.query(Presenter).filter(Presenter.organization == organization, Presenter.user == org_presenter.user).first():
                abort(400, {'code': 400, 'message': 'A presenter with that user already exists on this organization', 'data': {'presenter': org_presenter.to_json()}})
            org_presenter.organization = organization
            session.add(org_presenter)
            session.commit()
            return organization.to_json(with_relationships={'presenters': {'user': None}})['presenters']
示例#2
0
def refresh_data(sched_info_fname, convention_info_fname=None):
    # Keep track of the number of errors and warnings.
    num_errors = 0
    num_warnings = 0

    # Delete records from tables of interest.
    events = Event.query.all()
    for event in events:
        event.rooms = []
        event.presenters = []
        event.timeslots = []
    db.session.commit()

    DataLoadError.query.delete()
    Convention.query.delete()
    Timeslot.query.delete()
    Track.query.delete()
    Event.query.delete()
    Presenter.query.delete()
    Room.query.delete()
    RoomGroup.query.delete()

    # Define the convention.

    if not convention_info_fname:
        script_dir = os.path.dirname(__file__)
        convention_info_fname = os.path.join(script_dir, CONVENTION_INFO_FNAME)

    convention = Convention()
    with open(convention_info_fname, 'rb') as csvfile:
        csvreader = csv.reader(csvfile, delimiter=',', quotechar='"')
        first_row = True
        for row in csvreader:
            if first_row:
                first_row = False
            else:
                convention.name = row[0]
                convention.description = row[1]
                convention.date_format = row[5]
                convention.datetime_format = row[6]
                convention.start_dt = datetime.datetime.strptime(
                    row[2], convention.datetime_format)
                convention.end_dt = datetime.datetime.strptime(
                    row[3], convention.datetime_format)
                convention.timeslot_duration = datetime.timedelta(
                    0,
                    int(row[4]) * 60)  # Minutes converted to seconds.
                convention.url = row[7]
                convention.active = True

                # There is only one row of convention data.
                break
    db.session.add(convention)
    db.session.commit()

    # Commit the data to the database.
    db.session.commit()

    # Create timeslots.
    timeslot_count = int(
        (convention.end_dt - convention.start_dt).total_seconds() /
        convention.timeslot_duration.total_seconds())
    for n in range(timeslot_count):
        timeslot = Timeslot(n)
        timeslot.active = True
        db.session.add(timeslot)

    # Commit the data to the database.
    db.session.commit()

    # Add tracks.

    # The track name and the email address for each CREM track.
    track_infos = (
        ('Literature', '*****@*****.**'),
        ('Tech', '*****@*****.**'),
        ('After Dark', '*****@*****.**'),
        ('Action Adventure', '*****@*****.**'),
        ('Costuming', '*****@*****.**'),
        ('Comics', '*****@*****.**'),
        ('Gaming', '*****@*****.**'),
        ('DIY', '*****@*****.**'),
        ('Food', '*****@*****.**'),
        ('Science', '*****@*****.**'),
        ('Media', '*****@*****.**'),
        ('Mayhem', '*****@*****.**'),
        ('Anime', '*****@*****.**'),
        ('Penguicon', '*****@*****.**'),
        ('Life', '*****@*****.**'),
    )

    # Create tracks and save database objects in dictionary for later reference.
    tracks = {}
    for track_info in track_infos:
        track = Track(track_info[0], track_info[1])
        tracks[track_info[0]] = track
        db.session.add(track)

    # Commit the data to the database.
    db.session.commit()

    # Add room groups.

    room_group_names = (
        'Algonquin',
        'Charlevoix',
        'Lobby',
    )

    for room_group_name in room_group_names:
        room_group = RoomGroup(room_group_name)
        db.session.add(room_group)

    # Commit the data to the database.
    db.session.commit()

    # Add rooms.

    # For each room, the name, square feet, capacity and room group it belongs to.
    room_infos = (
        ('Algonquin A', 1207, 100, 'Algonquin'),
        ('Algonquin B', 1207, 100, 'Algonquin'),
        ('Algonquin C', 1207, 100, 'Algonquin'),
        ('Algonquin D', 1207, 100, 'Algonquin'),
        ('Algonquin Foyer', 3000, 450, None),
        ('Charlevoix A', 756, 64, 'Charlevoix'),
        ('Charlevoix B', 756, 64, 'Charlevoix'),
        ('Charlevoix C', 756, 64, 'Charlevoix'),
        ('Portage Auditorium', 1439, 68, None),
        ('Windover', 1475, 40, None),
        ('TC Linguinis', 1930, 40, None),
        ('Baldwin Board Room', 431, 12, None),
        ('Board of Directors', 511, 15, None),
        ('Board of Governors', 391, 5, None),
        ('Board of Regents', 439, 15, None),
        ('Board of Trustees', 534, 40, None),
        ('Hamlin', 360, 25, None),
        ('Montcalm', 665, 50, None),
        ('Nicolet', 667, 50, None),
        ('Game Table A', 20, 10, 'Lobby'),
        ('Game Table B', 20, 10, 'Lobby'),
        ('Game Table C', 20, 10, 'Lobby'),
        ('Game Table D', 20, 10, 'Lobby'),
    )

    # Create rooms and save database objects in dictionary for later reference.
    rooms = {}
    for room_info in room_infos:
        room = Room()
        room.room_name = room_info[0]
        room.room_sq_ft = room_info[1]
        room.room_capacity = room_info[2]
        if room_info[3]:
            room.room_group = db.session.query(RoomGroup).\
                filter(RoomGroup.room_group_name == room_info[3]).first()
        rooms[room.room_name] = room
        db.session.add(room)

    # Commit the data to the database.
    db.session.commit()

    # Keep track of presenters.
    presenters = {}

    # Read events from file.
    with open(sched_info_fname, 'rb') as csvfile:
        csvreader = csv.reader(csvfile, delimiter=',', quotechar='"')
        first_row = True
        for row in csvreader:
            if first_row:
                first_row = False
                continue
            if len(row) < 11:
                load_error = DataLoadError()
                load_error.error_level = 'Error'
                load_error.destination_table = 'event'
                load_error.line_num = csvreader.line_num
                load_error.error_msg = 'Not enough columns in row %d' % csvreader.line_num
                load_error.error_dt = datetime.datetime.now()
                db.session.add(load_error)
                num_errors += 1
                continue

            trackname = row[5].split(',')[0].strip()
            if trackname not in tracks:
                # There is no corresponding track, so add it.
                email = '-'.join(
                    trackname.lower().split()) + '*****@*****.**'
                track = Track(trackname, email)
                tracks[trackname] = track
                db.session.add(track)

                load_error = DataLoadError()
                load_error.error_level = 'Warning'
                load_error.destination_table = 'event'
                load_error.line_num = csvreader.line_num
                load_error.error_msg = '%s is not a defined track; adding it' % trackname
                load_error.error_dt = datetime.datetime.now()
                db.session.add(load_error)
                num_errors += 1
                continue
            event = Event()
            event.title = row[6]
            event.description = row[8]
            event.track = tracks[trackname]

            # Add timeslots and duration.
            try:
                timeslots = get_timeslots(row[0], row[1], row[9], convention,
                                          Timeslot)
                event.timeslots = timeslots
                event.duration = len(timeslots)
            except Exception as e:
                load_error = DataLoadError()
                load_error.error_level = 'Error'
                load_error.destination_table = 'event'
                load_error.line_num = csvreader.line_num
                load_error.error_msg = str(e)
                load_error.error_dt = datetime.datetime.now()
                db.session.add(load_error)
                num_errors += 1
                continue

            event.facilityRequest = row[10]
            event.convention = convention

            # Add room to the event.
            if row[4].strip():
                if row[4] not in rooms:
                    # This is not a predefined room, so add it.
                    load_error = DataLoadError()
                    load_error.error_level = 'Warning'
                    load_error.destination_table = 'event'
                    load_error.line_num = csvreader.line_num
                    load_error.error_msg = '%s is not a pre-defined room; adding this room' % row[
                        4]
                    load_error.error_dt = datetime.datetime.now()
                    num_warnings += 1
                    db.session.add(load_error)

                    room = Room()
                    room.room_name = row[4]
                    room.room_sq_ft = 0
                    room.room_capacity = 0
                    rooms[row[4]] = room
                    db.session.add(room)
                else:
                    room = rooms[row[4]]
                event.rooms.append(room)

            # Add presenters.
            if row[7].strip():
                presenter_names = row[7].split(',')
                presenter_names = [s.strip() for s in presenter_names]
                for presenter_name in presenter_names:
                    if presenter_name in presenters:
                        presenter = presenters[presenter_name]
                    else:
                        last_name = presenter_name.split(' ')[-1].strip()
                        first_name = ' '.join(
                            presenter_name.split(' ')[0:-1]).strip()
                        presenter = Presenter(first_name, last_name)
                        presenters[presenter_name] = presenter
                        db.session.add(presenter)
                    event.presenters.append(presenter)

            db.session.add(event)

    # Commit the data to the database.
    db.session.commit()

    # Return the number of errors and warnings.
    return num_errors, num_warnings
示例#3
0
    ('Sarah', 'Elkins', '*****@*****.**', '301-613-4393'),
    ('Jen', 'Talley', '*****@*****.**', '7347319771'),
    ('cassinator', '', '*****@*****.**', ''),
    ('mwlauthor', '', '', ''),
    ('Leah', 'Rapp', '*****@*****.**', ''),
    ('Bob', 'Trembley', '*****@*****.**', ''),
    ('Kent', 'Newland', '*****@*****.**', ''),
    ('Angela', 'Rush', '*****@*****.**', '248-505-1551'),
    ('Brittany', 'Burke', '*****@*****.**', '248-259-5122'),
    ('Stu', 'Chisholm', '*****@*****.**', '(586) 773-6182'),
    ('Joshua', 'DeBonis', '*****@*****.**', '203.470.7264'),
    ('Nikita', 'Mikros', '*****@*****.**', ''),
)

for presenter_info in presenter_infos:
    presenter = Presenter(presenter_info[0], presenter_info[1])
    presenter.email = presenter_info[2]
    presenter.phone = presenter_info[3]
    db.session.add(presenter)

# Commit the test data to the database.
db.session.commit()

# Assign a random set of presenters to each event.
events = Event.query.all()
presenters = Presenter.query.all()
for event in events:
    # Randomly select from 0 to 4 presenters for this event.
    random_presenters = random.sample(presenters, random.randrange(5))
    event.presenters = random_presenters