1
0
mirror of https://github.com/Tygs/0bin.git synced 2023-08-10 21:13:00 +03:00
0bin/zerobin/utils.py

107 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()