# -*- coding: utf-8 -*- import time import os import glob import tempfile import sys import locale import default_settings sys.path.append(default_settings.LIBS_DIR) try: from privilege import drop_privileges_permanently, coerce_user, coerce_group except (AttributeError): pass # privilege does't work on several plateform def drop_privileges(user=None, group=None, wait=5): """ Try to set the process user and group to another one. If no group is provided, it's set to the same as the user. You can wait for a certain time before doing so. """ if wait: time.sleep(wait) if user: group = group or user try: user = coerce_user(user) group = coerce_group(group) lock_files = glob.glob(os.path.join(tempfile.gettempdir(), 'bottle.*.lock')) for lock_file in lock_files: os.chown(lock_file, user, group) drop_privileges_permanently(user, group, ()) except Exception: print "Failed to drop privileges. Running with current user." def dmerge(*args): """ Return new directionay being the sum of all merged dictionaries passed as arguments """ dictionary = {} for arg in args: dictionary.update(arg) return dictionary def get_pastes_count(): """ Return the number of pastes created (must have option DISPLAY_COUNTER enabled) """ locale.setlocale(locale.LC_ALL, 'en_US') counter_path = settings.PASTE_FILES_ROOT counter_file = os.path.join(counter_path, 'counter') try: with open(counter_file, "r") as f: count = f.read(50) f.close except IOError: count = 0 return locale.format("%d", long(float(count)), grouping=True) class SettingsContainer(object): """ Singleton containing the settings for the whole app """ _instance = None def __new__(cls, *args, **kwargs): if not cls._instance: cls._instance = super(SettingsContainer, cls).__new__(cls, *args, **kwargs) cls._instance.update_with_module(default_settings) return cls._instance def update_with_module(self, module): """ Update settings with values from the given module. (Taking only variable with uppercased name) """ for name, value in module.__dict__.iteritems(): if name.isupper(): setattr(self, name, value) return self @classmethod def from_module(cls, module): """ Create an instance of SettingsContainer with values based on the one in the passed module. """ settings = cls() settings.update_with_module(module) return settings def update_with_file(self, filepath): """ Update settings with values from the given module file. User update_with_module() behind the scene """ sys.path.insert(0, os.path.dirname(filepath)) module_name = os.path.splitext(os.path.basename(filepath))[0] return self.update_with_module(__import__(module_name)) settings = SettingsContainer()