From 1847557670e3b3ae160fd2ac42047f8e6ccbff16 Mon Sep 17 00:00:00 2001 From: Chris Hines <chris.hines@monash.edu> Date: Wed, 18 May 2016 00:55:49 +0000 Subject: [PATCH] speed up the default slurm provisioing by running groups of users rather than adding each user individually --- .../templates/provision_slurm.py.j2 | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/roles/provision_slurm/templates/provision_slurm.py.j2 b/roles/provision_slurm/templates/provision_slurm.py.j2 index a8a6bca3..986d7c15 100644 --- a/roles/provision_slurm/templates/provision_slurm.py.j2 +++ b/roles/provision_slurm/templates/provision_slurm.py.j2 @@ -68,9 +68,21 @@ s.searchFilter = "{{ search_filter }}" users=get_users(s) mk_slurmaccount("default") +usergrouplist=[] +userlist=[] +i=0 for user in users: + if i==200: + i=0 + usergrouplist.append(",".join(userlist)) + userlist=[] + i=i+1 + userlist.append(users[user].entry['uid'][0]) +usergrouplist.append(",".join(userlist)) +for usergroup in usergrouplist: + try: - mk_slurmuser(users[user].entry['uid'][0],"default") + mk_slurmuser(usergroup,"default") except: print traceback.format_exc() pass -- GitLab