From 6c32293d4c41d688fd16195ef96f1e893e7c4ef2 Mon Sep 17 00:00:00 2001 From: Jeff Johnson Date: Thu, 5 Sep 2013 19:25:31 -0400 Subject: [PATCH] use dict instead of set as set cannot be encoded as JSON (used by flask-login secure cookies) --- flask_split/core.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/flask_split/core.py b/flask_split/core.py index 12ad22e..67564fc 100644 --- a/flask_split/core.py +++ b/flask_split/core.py @@ -137,7 +137,7 @@ def finished(experiment_name, reset_choice=True, reset_finished_tracking=True): alternative_name = _get_session().get(experiment.key) if alternative_name: if 'split_finished' not in session: - session['split_finished'] = set() + session['split_finished'] = {} if experiment.key not in session['split_finished']: alternative = Alternative( redis, alternative_name, experiment_name) @@ -146,12 +146,12 @@ def finished(experiment_name, reset_choice=True, reset_finished_tracking=True): _get_session().pop(experiment.key, None) if reset_finished_tracking: try: - session['split_finished'].remove(experiment.key) + del session['split_finished'][experiment.key] except KeyError: pass session.modified = True else: - session['split_finished'].add(experiment.key) + session['split_finished'][experiment.key] = True session.modified = True except ConnectionError: if not current_app.config['SPLIT_DB_FAILOVER']: