summaryrefslogtreecommitdiff
path: root/portato/session.py
blob: 2a049d14bb7f9580662f70340f39e2c1a8d0762e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
# -*- coding: utf-8 -*-
#
# File: portato/session.py
# This file is part of the Portato-Project, a graphical portage-frontend.
#
# Copyright (C) 2007 René 'Necoro' Neumann
# This is free software.  You may redistribute copies of it under the terms of
# the GNU General Public License version 2.
# There is NO WARRANTY, to the extent permitted by law.
#
# Written by René 'Necoro' Neumann <necoro@necoro.net>

from __future__ import absolute_import, with_statement

import os, os.path

from .config_parser import ConfigParser
from .constants import SESSION_DIR
from .helper import debug

class Session (object):
	"""
	A small class allowing to save certain states of a program.
	This class works in a quite abstract manner, as it works with handlers, which
	define what options to use out of the config file and how to apply them to the program.

	Note: This class currently does not work with boolean config options. If you
	want to define boolean values, use 0 and 1. This is future proof.
	"""

	def __init__ (self, file):
		"""
		Initialize a session with a certain file inside L{SESSION_DIR.}

		@param file: the file in L{SESSION_DIR}, where the options will be saved.
		"""

		self._cfg = None
		self._handlers = []

		if not (os.path.exists(SESSION_DIR) and os.path.isdir(SESSION_DIR)):
			os.mkdir(SESSION_DIR)
		self._cfg = ConfigParser(os.path.join(SESSION_DIR, file))
		try:
			self._cfg.parse()
		except IOError, e:
			if e.errno == 2: pass
			else: raise

	def add_handler (self, (options, load_fn, save_fn)):
		"""
		Adds a handler to this session. A handler is a three-tuple consisting of:
			- a list of (key,section) values
			- a function getting number of option arguments and applying them to the program
			- a function returning the number of option return values - getting them out of the program
		"""
		self._handlers.append((options, load_fn, save_fn))

	def load (self):
		"""
		Loads and applies all values of the session.
		"""
		for options, lfn, sfn in self._handlers:
			try:
				loaded = [self._cfg.get(*x) for x in options]
			except KeyError: # does not exist -> ignore
				debug("No values for %s.", options)
			else:
				debug("Loading %s with values %s.", options, loaded)
				lfn(*loaded)

	def save (self):
		"""
		Saves all options into the file.
		"""

		for options, lfn, sfn in self._handlers:
			vals = sfn()
			
			# map into tuple if necessairy
			if not hasattr(vals, "__iter__"):
				vals = (vals,)
			debug("Saving %s with values %s", options, vals)

			for value, (option, section) in zip(vals, options):
				self._cfg.add_section(section)
				self._cfg.add(option, str(value), section = section, with_blankline = False)
		
		self._cfg.write()