"""User settings/configuration loading""" # Copyright (C) 2006-2007, Red Hat, Inc. # # This library is free software; you can redistribute it and/or # modify it under the terms of the GNU Lesser General Public # License as published by the Free Software Foundation; either # version 2 of the License, or (at your option) any later version. # # This library 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 # Lesser General Public License for more details. # # You should have received a copy of the GNU Lesser General Public # License along with this library; if not, write to the # Free Software Foundation, Inc., 59 Temple Place - Suite 330, # Boston, MA 02111-1307, USA. import os import logging from ConfigParser import ConfigParser from sugar import env from sugar import util from sugar.graphics.xocolor import XoColor class _Profile(object): """Local user's current options/profile information User settings are stored in an INI-style configuration file. This object uses the ConfigParser module to load the settings. (We only very rarely set keys, so we don't keep the ConfigParser around between calls.) The profile is also responsible for loading the user's public and private ssh keys from disk. Attributes: name -- child's name color -- XoColor for the child's icon server -- school server with which the child is associated server_registered -- whether the child has registered with the school server or not backup1 -- temporary backup info key for Trial-2 pubkey -- public ssh key privkey_hash -- SHA has of the child's public key """ def __init__(self): self.valid = True self.name = None self.color = None self.pubkey = None self.privkey_hash = None self.server = None self.server_registered = False self.backup1 = None self._config_path = os.path.join(env.get_profile_path(), 'config') self._load() def update(self): self._load() def _load(self): self._load_config() self._load_pubkey() self._hash_private_key() def _load_config(self): cp = ConfigParser() parsed = cp.read([self._config_path]) if cp.has_option('Buddy', 'NickName'): name = cp.get('Buddy', 'NickName') # decode nickname from ascii-safe chars to unicode self.name = name.decode("utf-8") else: self.valid = False if cp.has_option('Buddy', 'Color'): self.color = XoColor(cp.get('Buddy', 'Color')) if cp.has_option('Server', 'Server'): self.server = cp.get('Server', 'Server') if cp.has_option('Server', 'Registered'): registered = cp.get('Server', 'Registered') if registered.lower() == "true": self.server_registered = True if cp.has_option('Server', 'Backup1'): self.backup1 = cp.get('Server', 'Backup1') del cp def _load_pubkey(self): self.pubkey = None key_path = os.path.join(env.get_profile_path(), 'owner.key.pub') try: f = open(key_path, "r") lines = f.readlines() f.close() except IOError, e: self.valid = False logging.error("Error reading public key: %s" % e) return magic = "ssh-dss " for l in lines: l = l.strip() if not l.startswith(magic): continue self.pubkey = l[len(magic):] break if not self.pubkey: logging.error("Error parsing public key.") def _hash_private_key(self): self.privkey_hash = None key_path = os.path.join(env.get_profile_path(), 'owner.key') try: f = open(key_path, "r") lines = f.readlines() f.close() except IOError, e: logging.error("Error reading private key: %s" % e) return key = "" for l in lines: l = l.strip() if l.startswith("-----BEGIN DSA PRIVATE KEY-----"): continue if l.startswith("-----END DSA PRIVATE KEY-----"): continue key += l if not len(key): logging.error("Error parsing public key.") # hash it key_hash = util._sha_data(key) self.privkey_hash = util.printable_hash(key_hash) def set_key(self, section, key, value): cp = ConfigParser() parsed = cp.read([self._config_path]) if not cp.has_section(section): cp.add_section(section) cp.set(section, key, value) f = open(self._config_path, 'w') cp.write(f) f.close() del cp self._load_config() def is_valid(): return _profile.valid def get_nick_name(): return _profile.name def get_color(): return _profile.color def get_pubkey(): return _profile.pubkey def get_private_key_hash(): return _profile.privkey_hash def get_server(): return _profile.server def set_server(server): _profile.set_key('Server', 'server', server) def get_trial2_backup(): return _profile.backup1 def set_trial2_backup(backup_info): _profile.set_key('Server', 'backup1', backup_info) def get_server_registered(): return _profile.server_registered def set_server_registered(): _profile.set_key('Server', 'Registered', True) def update(): _profile.update() _profile = _Profile()