# Copyright (C) University of Tennessee Health Science Center, Memphis, TN.
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License
# as published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
# See the GNU Affero General Public License for more details.
#
# This program is available from Source Forge: at GeneNetwork Project
# (sourceforge.net/projects/genenetwork/).
#
# Contact Drs. Robert W. Williams and Xiaodong Zhou (2010)
# at rwilliams@uthsc.edu and xzhou15@uthsc.edu
#
#
#
# This module is used by GeneNetwork project (www.genenetwork.org)
#
# Created by GeneNetwork Core Team 2010/08/10
#
# Last updated by GeneNetwork Core Team 2010/10/20

#########################################
#convert Field storage object to Dict object
#in order to be able to saved into a session file
#########################################

class cgiData(dict):
    '''convert Field storage object to Dict object
       Filed storage object cannot be properly dumped
    '''

    def __init__(self, field_storage=None):

        if not field_storage:
            field_storage={}

        for key in field_storage.keys():
            temp = field_storage.getlist(key)
            if len(temp) > 1:
                temp = map(self.toValue, temp)
            elif len(temp) == 1:
                temp = self.toValue(temp[0])
            else:
                temp = None
            self[key]= temp

    def toValue(self, obj):
        '''fieldstorge returns different type of objects, \
                need to convert to string or None'''
        try:
            return obj.value
        except:
            return ""

    def getvalue(self, k, default= None):
        try:
            return self[k]
        except:
            return default

    getfirst = getvalue