gramps/gramps2/src/GrampsIniKeys.py
Alex Roitman 3b2ac1ef49 * src/GrampsGconfKeys.py: Add get/set methods for welcome key.
* src/GrampsIniKeys.py: Add get/set methods for welcome key.


svn: r4533
2005-05-10 14:57:11 +00:00

678 lines
21 KiB
Python

#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2005 Donald N. Allingham
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
# $Id$
"""
Low-level handling of .ini keys.
"""
import os
import time
import ConfigParser
NL = "\n" # FIX: newlines on Mac/Windows, if different?
# the following table was generated from data/gramps.schemas
_ini_schema = {
'behavior/autoload' : '0',
'behavior/betawarn' : '0',
'behavior/welcome' : '100',
'preferences/date-format' : '0',
'preferences/dont-ask' : '0',
'interface/defaultview' : '0',
'interface/familyview' : '0',
'interface/filter' : '0',
'preferences/fprefix' : 'F%04d',
'preferences/eprefix' : 'E%04d',
'preferences/iprefix' : 'I%04d',
'preferences/oprefix' : 'O%04d',
'preferences/pprefix' : 'P%04d',
'preferences/sprefix' : 'S%04d',
'preferences/goutput-preference' : 'No default format',
'preferences/output-preference' : 'No default format',
'preferences/paper-preference' : 'Letter',
'index-visible' : '0',
'interface/index-visible' : '0',
'paths/recent-file' : '',
'paths/recent-import-dir' : '',
'paths/recent-export-dir' : '',
'behavior/make-reference' : '1',
'behavior/media-global' : '1',
'behavior/media-local' : '1',
'preferences/name-format' : '0',
'paths/report-directory' : './',
'researcher/researcher-addr' : '',
'researcher/researcher-city' : '',
'researcher/researcher-country' : '',
'researcher/researcher-email' : '',
'researcher/researcher-name' : '',
'researcher/researcher-phone' : '',
'researcher/researcher-postal' : '',
'researcher/researcher-state' : '',
'behavior/show-calendar' : '0',
'behavior/startup' : '0',
'interface/size-checked' : '0',
'interface/statusbar' : '1',
'behavior/surname-guessing' : '0',
'interface/toolbar' : '5',
'interface/toolbar-on' : '1',
'behavior/use-lds' : '0',
'behavior/use-tips' : '0',
'behavior/pop-plugin-status' : '0',
'interface/view' : '1',
'paths/website-directory' : './',
}
def make_bool(val):
""" Function to turn strings into booleans. """
# these are the possible strings that should be considered False
if val.lower() in ["0", "false", "none", ""]:
return False
else:
return True
class IniKeyClient:
""" Class to emulate gconf's client """
def __init__(self, filename = None):
""" Constructor takes an optional filename """
self.data = {}
self.callbacks = {}
self.filename = filename
if self.filename and os.path.exists(filename):
self.data = self.load_ini(self.filename)
def notify_add(self, path, func):
"""
I think that these are callbacks that get called when the keys
are set.
"""
parts = path.split("/") # FIX: make path-sep independent
# /apps/gramps/section/key
section = parts[-2]
key = parts[-1]
if section not in self.callbacks:
self.callbacks[section] = {}
self.callbacks[section][key] = func
def load_ini(self, filename):
""" Load .ini into dict of dicts, which it returns """
cp = ConfigParser.ConfigParser()
cp.read(filename)
data = {}
for sec in cp.sections():
name = sec.lower()
if not data.has_key(name):
data[name] = {}
for opt in cp.options(sec):
data[name][opt.lower()] = cp.get(sec, opt).strip()
return data
def save_ini(self, filename = None):
"""
Saves the current section/keys to a .ini file. Optional filename
will override the default filename, if one.
"""
if not filename:
filename = self.filename
if filename:
fp = open(filename, "w")
fp.write(";; Gramps key file" + NL)
fp.write((";; Automatically created at %s" % time.strftime("%Y/%m/%d %H:%M:%S")) + NL + NL)
sections = self.data.keys()
sections.sort()
for section in sections:
fp.write(("[%s]" + NL) % section)
keys = self.data[section].keys()
keys.sort()
for key in keys:
fp.write(("%s=%s" + NL)% (key, self.data[section][key]))
fp.write(NL)
fp.close()
# else, no filename given
def get_bool(self, section, key):
""" Emulates gconf's client method """
return make_bool(self.data[section][key])
def get_string(self, section, key):
""" Emulates gconf's client method """
return self.data[section][key]
def get_int(self, section, key):
""" Emulates gconf's client method """
if self.data[section][key].isdigit():
return int(self.data[section][key])
elif self.data[section][key].lower() in ["true"]:
return 1
else:
return 0
def set_bool(self, section, key, val):
""" Emulates gconf's client method """
if section not in self.data:
self.data[section] = {}
self.data[section][key] = str(val)
if section in self.callbacks and key in self.callbacks[section]:
self.callbacks[section][key](self,0,self.data[section][key],None)
def set_string(self, section, key, val):
""" Emulates gconf's client method """
if section not in self.data:
self.data[section] = {}
self.data[section][key] = val
if section in self.callbacks and key in self.callbacks[section]:
self.callbacks[section][key](self,0,self.data[section][key],None)
def set_int(self, section, key, val):
""" Emulates gconf's client method """
if section not in self.data:
self.data[section] = {}
self.data[section][key] = str(val)
if section in self.callbacks and key in self.callbacks[section]:
self.callbacks[section][key](self,0,self.data[section][key],None)
def get_default_from_schema(self, section, key):
try:
return _ini_schema["%s/%s" % (section,key)]
except:
print "get_default_from_schema:", section, key
return "0" # FIX: where does this get its defaults?
def suggest_sync(self):
self.save_ini() # save back to default file, if named
# FIX: not OS-path independent:
client = IniKeyClient(os.path.expanduser("~/.gramps/keys.ini"))
#-------------------------------------------------------------------------
#
# Functions to obtain values from .ini keys
# and store values into .ini keys
#
# All gramps keys should be accessed through these functions!
#
#-------------------------------------------------------------------------
# interface keys
def get_default_view():
return get_int("interface", "defaultview",(0,1))
def save_default_view(val):
set_int("interface", "defaultview",val,(0,1))
def get_family_view():
return get_int("interface", "familyview", (0,1))
def save_family_view(val):
set_int("interface", "familyview",val, (0,1))
def get_filter():
return get_bool("interface", "filter")
def save_filter(val):
set_bool("interface", "filter",val)
def get_dont_ask():
return get_bool("interface", "dont-ask")
def save_dont_ask(val):
set_bool("interface", "dont-ask",val)
def get_index_visible():
return get_bool("interface", "index-visible")
def save_index_visible(val):
set_bool("interface", "index-visible",val)
def get_statusbar():
return get_int("interface", "statusbar",(0,1,2))
def save_statusbar(val):
set_int("interface", "statusbar",val,(0,1,2))
def get_toolbar():
return get_int("interface", "toolbar",(0,1,2,3,5))
def save_toolbar(val):
set_int("interface", "toolbar",val,(0,1,2,3,5))
def get_toolbar_on():
return get_bool("interface", "toolbar-on")
def save_toolbar_on(val):
set_bool("interface", "toolbar-on",val)
def get_view():
return get_bool("interface", "view")
def save_view(val):
set_bool("interface", "view",val)
# paths keys
def get_lastfile():
return get_string("paths", "recent-file")
def save_last_file(val):
set_string("paths", "recent-file",val)
def get_last_import_dir():
return get_string("paths", "recent-import-dir")
def save_last_import_dir(val):
set_string_as_path("paths", "recent-import-dir",val)
def get_last_export_dir():
return get_string("paths", "recent-export-dir")
def save_last_export_dir(val):
set_string_as_path("paths", "recent-export-dir",val)
def get_report_dir():
return get_string("paths", "report-directory")
def save_report_dir(val):
set_string_as_path("paths", "report-directory",val)
def get_web_dir():
return get_string("paths", "website-directory")
def save_web_dir(val):
set_string_as_path("paths", "website-directory",val)
# behavior keys
def get_startup():
return get_int("behavior", "startup",(0,1))
def save_startup(val):
set_int("behavior", "startup",val,(0,1))
def get_screen_size_checked():
return get_bool("interface", "size-checked")
def save_screen_size_checked(val):
set_bool("interface", "size-checked",val)
def get_autoload():
return get_bool("behavior", "autoload")
def save_autoload(val):
set_bool("behavior", "autoload",val)
def get_betawarn():
return get_bool("behavior", "betawarn")
def save_betawarn(val):
set_bool("behavior", "betawarn",val)
def get_welcome():
return get_bool("behavior", "welcome")
def save_welcome(val):
set_bool("behavior", "welcome",val)
def get_media_reference():
return get_bool("behavior", "make-reference")
def save_media_reference(val):
set_bool("behavior", "make-reference",val)
def get_media_global():
return get_bool("behavior", "media-global")
def save_media_global(val):
set_bool("behavior", "media-global",val)
def get_media_local():
return get_bool("behavior", "media-local")
def save_media_local(val):
set_bool("behavior", "media-local",val)
def get_lastnamegen(_surname_styles=[]):
return get_int("behavior", "surname-guessing",
range(len(_surname_styles)))
def save_lastnamegen(val,_surname_styles=[]):
set_int("behavior", "surname-guessing",val,
range(len(_surname_styles)))
def get_uselds():
return get_bool("behavior", "use-lds")
def save_uselds(val):
set_bool("behavior", "use-lds",val)
def get_usetips():
return get_bool("behavior", "use-tips")
def save_usetips(val):
set_bool("behavior", "use-tips",val)
def get_pop_plugin_status():
return get_bool("behavior", "pop-plugin-status")
def save_pop_plugin_status(val):
set_bool("behavior", "pop-plugin-status",val)
# preferences keys
def get_person_id_prefix():
return get_string("preferences", "iprefix")
def get_event_id_prefix():
return get_string("preferences", "eprefix")
def save_iprefix(val):
set_string_as_id_prefix("preferences", "iprefix",val)
def get_object_id_prefix():
return get_string("preferences", "oprefix")
def save_oprefix(val):
set_string_as_id_prefix("preferences", "oprefix",val)
def get_source_id_prefix():
return get_string("preferences", "sprefix")
def save_sprefix(val):
set_string_as_id_prefix("preferences", "sprefix",val)
def save_eprefix(val):
set_string_as_id_prefix("preferences", "eprefix",val)
def get_place_id_prefix():
return get_string("preferences", "pprefix")
def save_pprefix(val):
set_string_as_id_prefix("preferences", "pprefix",val)
def get_family_id_prefix():
return get_string("preferences", "fprefix")
def save_fprefix(val):
set_string_as_id_prefix("preferences", "fprefix",val)
def get_paper_preference():
return get_string("preferences", "paper-preference")
def save_paper_preference(val):
set_string("preferences", "paper-preference",val)
def get_output_preference():
return get_string("preferences", "output-preference")
def save_output_preference(val):
set_string("preferences", "output-preference",val)
def get_goutput_preference():
return get_string("preferences", "goutput-preference")
def save_goutput_preference(val):
set_string("preferences", "goutput-preference",val)
def get_use_tips():
return get_bool("preferences", "use-tips")
def save_use_tips(val):
set_bool("preferences", "use-tips",val)
def get_date_format(_date_format_list=[]):
return get_int("preferences", "date-format",
range(len(_date_format_list)))
def save_date_format(val,_date_format_list=[]):
set_int("preferences", "date-format",val,
range(len(_date_format_list)))
def get_name_format(_name_format_list):
return get_int("preferences", "name-format",
range(len(_name_format_list)))
def save_name_format(val,_name_format_list):
set_int("preferences", "name-format",val,
range(len(_name_format_list)))
# researcher keys
def get_researcher_name():
return get_string("researcher", "researcher-name")
def save_researcher_name(val):
set_string("researcher", "researcher-name",val)
def get_researcher_addr():
return get_string("researcher", "researcher-addr")
def save_researcher_addr(val):
set_string("researcher", "researcher-addr",val)
def get_researcher_city():
return get_string("researcher", "researcher-city")
def save_researcher_city(val):
set_string("researcher", "researcher-city",val)
def get_researcher_state():
return get_string("researcher", "researcher-state")
def save_researcher_state(val):
set_string("researcher", "researcher-state",val)
def get_researcher_country():
return get_string("researcher", "researcher-country")
def save_researcher_country(val):
set_string("researcher", "researcher-country",val)
def get_researcher_postal():
return get_string("researcher", "researcher-postal")
def save_researcher_postal(val):
set_string("researcher", "researcher-postal",val)
def get_researcher_phone():
return get_string("researcher", "researcher-phone")
def save_researcher_phone(val):
set_string("researcher", "researcher-phone",val)
def get_researcher_email():
return get_string("researcher", "researcher-email")
def save_researcher_email(val):
set_string("researcher", "researcher-email",val)
#-------------------------------------------------------------------------
#
# Low-level grabbing and saving keys with error checking.
#
#-------------------------------------------------------------------------
def get_bool(section, key):
try:
val = client.get_bool(section, key)
except KeyError:
val = None
if val in (True,False):
return val
else:
return make_bool(client.get_default_from_schema(section, key))
def set_bool(section, key,val):
if val in (True,False):
client.set_bool(section, key,val)
def get_int(section, key,correct_tuple=None):
try:
val = client.get_int(section, key)
except KeyError:
val = None
if not correct_tuple or val in correct_tuple:
return val
else:
return int(client.get_default_from_schema(section, key))
def set_int(section, key,val,correct_tuple=None):
if not correct_tuple or val in correct_tuple:
client.set_int(section, key,val)
def get_string(section, key,test_func=None):
try:
val = client.get_string(section, key)
except KeyError:
val = ""
if not test_func or test_func(val):
return val
else:
return client.get_default_from_schema(section, key)
def set_string(section, key,val,test_func=None):
if not test_func or test_func(val):
client.set_string(section, key,val)
def set_string_as_path(section, key,val):
if not val:
val = client.get_default_from_schema(section, key)
else:
val = os.path.normpath(val) + os.sep
client.set_string(section, key,val)
def set_string_as_id_prefix(section, key,val):
if not val:
val = client.get_default_from_schema(section, key)
else:
try:
junk = val % 1
except:
val = client.get_default_from_schema(section, key)
client.set_string(section, key,val)
def sync():
client.suggest_sync()
if __name__ == "__main__":
print "Testing..."
client.filename = "test1.ini"
val = "test"
print "get_default_view()", get_default_view()
save_default_view(val)
print "get_family_view()", get_family_view()
save_family_view(val)
print "get_filter()", get_filter()
save_filter(val)
print "get_dont_ask()", get_dont_ask()
save_dont_ask(val)
print "get_index_visible()", get_index_visible()
save_index_visible(val)
print "get_statusbar()", get_statusbar()
save_statusbar(val)
print "get_toolbar()", get_toolbar()
save_toolbar(val)
print "get_toolbar_on()", get_toolbar_on()
save_toolbar_on(val)
print "get_view()", get_view()
save_view(val)
print "get_lastfile()", get_lastfile()
save_last_file(val)
print "get_last_import_dir()", get_last_import_dir()
save_last_import_dir(val)
print "get_last_export_dir()", get_last_export_dir()
save_last_export_dir(val)
print "get_report_dir()", get_report_dir()
save_report_dir(val)
print "get_web_dir()", get_web_dir()
save_web_dir(val)
print "get_startup()", get_startup()
save_startup(val)
print "get_screen_size_checked()", get_screen_size_checked()
save_screen_size_checked(val)
print "get_autoload()", get_autoload()
save_autoload(val)
print "get_betawarn()", get_betawarn()
save_betawarn(val)
print "get_welcome()", get_welcome()
save_welcome(val)
print "get_media_reference()", get_media_reference()
save_media_reference(val)
print "get_media_global()", get_media_global()
save_media_global(val)
print "get_media_local()", get_media_local()
save_media_local(val)
print "get_lastnamegen(val)", get_lastnamegen(val)
save_lastnamegen(val, val)
print "get_uselds()", get_uselds()
save_uselds(val)
print "get_usetips()", get_usetips()
save_usetips(val)
print "get_pop_plugin_status()", get_pop_plugin_status()
save_pop_plugin_status(val)
print "get_person_id_prefix()", get_person_id_prefix()
print "get_event_id_prefix()", get_event_id_prefix()
save_iprefix(val)
print "get_object_id_prefix()", get_object_id_prefix()
save_oprefix(val)
print "get_source_id_prefix()", get_source_id_prefix()
save_sprefix(val)
save_eprefix(val)
print "get_place_id_prefix()", get_place_id_prefix()
save_pprefix(val)
print "get_family_id_prefix()", get_family_id_prefix()
save_fprefix(val)
print "get_paper_preference()", get_paper_preference()
save_paper_preference(val)
print "get_output_preference()", get_output_preference()
save_output_preference(val)
print "get_goutput_preference()", get_goutput_preference()
save_goutput_preference(val)
print "get_use_tips()", get_use_tips()
save_use_tips(val)
print "get_date_format(val)", get_date_format(val)
save_date_format(val, val)
print "get_name_format(val)", get_name_format(val)
save_name_format(val, val)
print "get_researcher_name()", get_researcher_name()
save_researcher_name(val)
print "get_researcher_addr()", get_researcher_addr()
save_researcher_addr(val)
print "get_researcher_city()", get_researcher_city()
save_researcher_city(val)
print "get_researcher_state()", get_researcher_state()
save_researcher_state(val)
print "get_researcher_country()", get_researcher_country()
save_researcher_country(val)
print "get_researcher_postal()", get_researcher_postal()
save_researcher_postal(val)
print "get_researcher_phone()", get_researcher_phone()
save_researcher_phone(val)
print "get_researcher_email()", get_researcher_email()
save_researcher_email(val)
print "Syncing..."
sync()
client = IniKeyClient("test1.ini")
client.filename = "test2.ini"
print "Syncing..."
sync()
client = IniKeyClient()
def callback(client, id, data, args):
print "it called back! new value is %s" % data
print "Testing callbacks..."
client.notify_add("/apps/gramps/section/key", callback)
client.set_int("section", "key", 23)
assert client.get_int("section", "key") == 23
print "client.notify_add(): passed"
client.filename = "test3.ini"
print "Syncing..."
sync()
client = IniKeyClient("test3.ini")
assert client.get_int("section", "key") == 23
print "client.load_ini(): passed"
print "-" * 50
print "test1.ini and test2.ini should be identical, except for"
print "maybe the time. test3.ini should have [section] key=23"