| |
| |
| |
| |
| |
| |
|
|
|
|
| from .. import * |
|
|
| CMD_HELP = {} |
| |
|
|
|
|
| class _SudoManager: |
| def __init__(self): |
| self.db = None |
| self.owner = None |
| self._owner_sudos = [] |
|
|
| def _init_db(self): |
| if not self.db: |
| from .. import udB |
|
|
| self.db = udB |
| return self.db |
|
|
| def get_sudos(self): |
| db = self._init_db() |
| SUDOS = db.get_key("SUDOS") |
| return SUDOS or [] |
|
|
| @property |
| def should_allow_sudo(self): |
| db = self._init_db() |
| return db.get_key("SUDO") |
|
|
| def owner_and_sudos(self): |
| if not self.owner: |
| db = self._init_db() |
| self.owner = db.get_key("OWNER_ID") |
| return [self.owner, *self.get_sudos()] |
|
|
| @property |
| def fullsudos(self): |
| db = self._init_db() |
| fsudos = db.get("FULLSUDO") |
| if not self.owner: |
| self.owner = db.get_key("OWNER_ID") |
| if not fsudos: |
| return [self.owner] |
| fsudos = fsudos.split() |
| fsudos.append(self.owner) |
| return [int(_) for _ in fsudos] |
|
|
| def is_sudo(self, id_): |
| return bool(id_ in self.get_sudos()) |
|
|
|
|
| SUDO_M = _SudoManager() |
| owner_and_sudos = SUDO_M.owner_and_sudos |
| sudoers = SUDO_M.get_sudos |
| is_sudo = SUDO_M.is_sudo |
|
|
| |
|
|
|
|
| def append_or_update(load, func, name, arggs): |
| if isinstance(load, list): |
| return load.append(func) |
| if isinstance(load, dict): |
| if load.get(name): |
| return load[name].append((func, arggs)) |
| return load.update({name: [(func, arggs)]}) |
|
|
