config.py
上传用户:king477883
上传日期:2021-03-01
资源大小:9553k
文件大小:8k
源码类别:

游戏引擎

开发平台:

C++ Builder

  1. """
  2. @file config.py
  3. @brief Utility module for parsing and accessing the indra.xml config file.
  4. $LicenseInfo:firstyear=2006&license=mit$
  5. Copyright (c) 2006-2010, Linden Research, Inc.
  6. Permission is hereby granted, free of charge, to any person obtaining a copy
  7. of this software and associated documentation files (the "Software"), to deal
  8. in the Software without restriction, including without limitation the rights
  9. to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10. copies of the Software, and to permit persons to whom the Software is
  11. furnished to do so, subject to the following conditions:
  12. The above copyright notice and this permission notice shall be included in
  13. all copies or substantial portions of the Software.
  14. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19. OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  20. THE SOFTWARE.
  21. $/LicenseInfo$
  22. """
  23. import copy
  24. import errno
  25. import os
  26. import traceback
  27. import time
  28. import types
  29. from os.path import dirname, getmtime, join, realpath
  30. from indra.base import llsd
  31. _g_config = None
  32. class IndraConfig(object):
  33.     """
  34.     IndraConfig loads a 'indra' xml configuration file
  35.     and loads into memory.  This representation in memory
  36.     can get updated to overwrite values or add new values.
  37.     The xml configuration file is considered a live file and changes
  38.     to the file are checked and reloaded periodically.  If a value had
  39.     been overwritten via the update or set method, the loaded values
  40.     from the file are ignored (the values from the update/set methods
  41.     override)
  42.     """
  43.     def __init__(self, indra_config_file):
  44.         self._indra_config_file = indra_config_file
  45.         self._reload_check_interval = 30 # seconds
  46.         self._last_check_time = 0
  47.         self._last_mod_time = 0
  48.         self._config_overrides = {}
  49.         self._config_file_dict = {}
  50.         self._combined_dict = {}
  51.         self._load()
  52.     def _load(self):
  53.         # if you initialize the IndraConfig with None, no attempt
  54.         # is made to load any files
  55.         if self._indra_config_file is None:
  56.             return
  57.         config_file = open(self._indra_config_file)
  58.         self._config_file_dict = llsd.parse(config_file.read())
  59.         self._combine_dictionaries()
  60.         config_file.close()
  61.         self._last_mod_time = self._get_last_modified_time()
  62.         self._last_check_time = time.time() # now
  63.     def _get_last_modified_time(self):
  64.         """
  65.         Returns the mtime (last modified time) of the config file,
  66.         if such exists.
  67.         """
  68.         if self._indra_config_file is not None:
  69.             return os.path.getmtime(self._indra_config_file)
  70.         return 0
  71.     def _combine_dictionaries(self):
  72.         self._combined_dict = {}
  73.         self._combined_dict.update(self._config_file_dict)
  74.         self._combined_dict.update(self._config_overrides)
  75.     def _reload_if_necessary(self):
  76.         now = time.time()
  77.         if (now - self._last_check_time) > self._reload_check_interval:
  78.             self._last_check_time = now
  79.             try:
  80.                 modtime = self._get_last_modified_time()
  81.                 if modtime > self._last_mod_time:
  82.                     self._load()
  83.             except OSError, e:
  84.                 if e.errno == errno.ENOENT: # file not found
  85.                     # someone messed with our internal state
  86.                     # or removed the file
  87.                     print 'WARNING: Configuration file has been removed ' + (self._indra_config_file)
  88.                     print 'Disabling reloading of configuration file.'
  89.                     traceback.print_exc()
  90.                     self._indra_config_file = None
  91.                     self._last_check_time = 0
  92.                     self._last_mod_time = 0
  93.                 else:
  94.                     raise  # pass the exception along to the caller
  95.     def __getitem__(self, key):
  96.         self._reload_if_necessary()
  97.         return self._combined_dict[key]
  98.     def get(self, key, default = None):
  99.         try:
  100.             return self.__getitem__(key)
  101.         except KeyError:
  102.             return default
  103.     def __setitem__(self, key, value):
  104.         """
  105.         Sets the value of the config setting of key to be newval
  106.         Once any key/value pair is changed via the set method,
  107.         that key/value pair will remain set with that value until
  108.         change via the update or set method
  109.         """
  110.         self._config_overrides[key] = value
  111.         self._combine_dictionaries()
  112.     def set(self, key, newval):
  113.         return self.__setitem__(key, newval)
  114.     def update(self, new_conf):
  115.         """
  116.         Load an XML file and apply its map as overrides or additions
  117.         to the existing config.  Update can be a file or a dict.
  118.         Once any key/value pair is changed via the update method,
  119.         that key/value pair will remain set with that value until
  120.         change via the update or set method
  121.         """
  122.         if isinstance(new_conf, dict):
  123.             overrides = new_conf
  124.         else:
  125.             # assuming that it is a filename
  126.             config_file = open(new_conf)
  127.             overrides = llsd.parse(config_file.read())
  128.             config_file.close()
  129.   
  130.         self._config_overrides.update(overrides)
  131.         self._combine_dictionaries()
  132.     def as_dict(self):
  133.         """
  134.         Returns immutable copy of the IndraConfig as a dictionary
  135.         """
  136.         return copy.deepcopy(self._combined_dict)
  137. def load(config_xml_file = None):
  138.     global _g_config
  139.     load_default_files = config_xml_file is None
  140.     if load_default_files:
  141.         ## going from:
  142.         ## "/opt/linden/indra/lib/python/indra/base/config.py"
  143.         ## to:
  144.         ## "/opt/linden/etc/indra.xml"
  145.         config_xml_file = realpath(
  146.             dirname(realpath(__file__)) + "../../../../../../etc/indra.xml")
  147.     try:
  148.         _g_config = IndraConfig(config_xml_file)
  149.     except IOError:
  150.         # Failure to load passed in file
  151.         # or indra.xml default file
  152.         if load_default_files:
  153.             try:
  154.                 config_xml_file = realpath(
  155.                     dirname(realpath(__file__)) + "../../../../../../etc/globals.xml")
  156.                 _g_config = IndraConfig(config_xml_file)
  157.                 return
  158.             except IOError:
  159.                 # Failure to load globals.xml
  160.                 # fall to code below
  161.                 pass
  162.         # Either failed to load passed in file
  163.         # or failed to load all default files
  164.         _g_config = IndraConfig(None)
  165. def dump(indra_xml_file, indra_cfg = None, update_in_mem=False):
  166.     '''
  167.     Dump config contents into a file
  168.     Kindof reverse of load.
  169.     Optionally takes a new config to dump.
  170.     Does NOT update global config unless requested.
  171.     '''
  172.     global _g_config
  173.     if not indra_cfg:
  174.         if _g_config is None:
  175.             return
  176.         indra_cfg = _g_config.as_dict()
  177.     if not indra_cfg:
  178.         return
  179.     config_file = open(indra_xml_file, 'w')
  180.     _config_xml = llsd.format_xml(indra_cfg)
  181.     config_file.write(_config_xml)
  182.     config_file.close()
  183.     if update_in_mem:
  184.         update(indra_cfg)
  185. def update(new_conf):
  186.     global _g_config
  187.     if _g_config is None:
  188.         # To keep with how this function behaved
  189.         # previously, a call to update
  190.         # before the global is defined
  191.         # make a new global config which does not
  192.         # load data from a file.
  193.         _g_config = IndraConfig(None)
  194.     return _g_config.update(new_conf)
  195. def get(key, default = None):
  196.     global _g_config
  197.     if _g_config is None:
  198.         load()
  199.     return _g_config.get(key, default)
  200. def set(key, newval):
  201.     """
  202.     Sets the value of the config setting of key to be newval
  203.     Once any key/value pair is changed via the set method,
  204.     that key/value pair will remain set with that value until
  205.     change via the update or set method or program termination
  206.     """
  207.     global _g_config
  208.     if _g_config is None:
  209.         _g_config = IndraConfig(None)
  210.     _g_config.set(key, newval)
  211. def get_config():
  212.     global _g_config
  213.     return _g_config