mirror of
https://github.com/Tygs/0bin.git
synced 2023-08-10 21:13:00 +03:00
105 lines
2.8 KiB
Python
105 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import time
|
|
import os
|
|
import glob
|
|
import tempfile
|
|
import sys
|
|
|
|
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
|
|
|
|
|
|
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()
|