zhangyiss/the-littlest-jupyterhub
1
0
Fork 0
mirror of https://github.com/jupyterhub/the-littlest-jupyterhub.git synced 2025-12-18 21:54:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support user resource limits

Browse Source 
Set default limit to 1G. Possibly controversial!
This commit is contained in:
yuvipanda
2018-06-27 01:30:34 -07:00
parent 459b985a19
commit 6fcc5cd357
1 changed files with 19 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tljh/configurer.py
View File

@@ -22,19 +22,24 @@ default = {
'allowed': [], 'allowed': [],
'banned': [], 'banned': [],
'admin': [] 'admin': []
},
'limits': {
'memory': '1G',
'cpu': None
} }
} }
def apply_yaml_config(path, c): def apply_yaml_config(path, c):
if not os.path.exists(path): if not os.path.exists(path):
user_config = copy.deepcopy(default) tljh_config = copy.deepcopy(default)
with open(path) as f: with open(path) as f:
user_config = _merge_dictionaries(yaml.safe_load(f), default) tljh_config = _merge_dictionaries(yaml.safe_load(f), default)
update_auth(c, user_config) update_auth(c, tljh_config)
update_userlists(c, user_config) update_userlists(c, tljh_config)
update_limits(c, tljh_config)
def update_auth(c, config): def update_auth(c, config):
@@ -62,6 +67,16 @@ def update_userlists(c, config):
c.Authenticator.admin_users = set(users['admin']) c.Authenticator.admin_users = set(users['admin'])
def update_limits(c, config):
"""
Set user server limits
"""
limits = config['limits']
c.SystemdSpawner.mem_limit = limits['memory']
c.SystemdSpawner.cpu_limit = limits['cpu']
def _merge_dictionaries(a, b, path=None, update=True): def _merge_dictionaries(a, b, path=None, update=True):
""" """
Merge two dictionaries recursively. Merge two dictionaries recursively.