# 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

#ExportSelectionPage.py

import string
import time

from base.templatePage import templatePage
	

#########################################
#     Export Selection Page
#########################################
class ExportSelectionPage(templatePage):

	def __init__(self, fd):

		templatePage.__init__(self, fd)

		collectionName = '%s_Select' % fd.RISet

		try:
			preSelection = fd.input_session_data[collectionName]
			preSelection = list(string.split(preSelection,','))
		except:
			preSelection = []

		for item in preSelection:
                        if not item:
                                preSelection.remove(item)

		if preSelection:
			self.content_type = 'application/txt'
			self.content_disposition = 'attachment; filename=%s' % (fd.RISet+'_export-%s.txt' % time.strftime("%y-%m-%d-%H-%M"))
			self.attachment += fd.RISet+"\n"
			for item in preSelection:
				self.attachment += item+"\n"
		else:
			heading = 'Export Collection'
                        detail = ['This collection is empty. No trait could be exported.']
                        self.error(heading=heading,detail=detail)