get_current_groups: Add revert_command in log #52

UAL-RE · Sep 10, 2020 · 8ec75d0 · 8ec75d0
1 parent ca84e7f
commit 8ec75d0
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/requiam/manual_override.py b/requiam/manual_override.py
@@ -242,6 +242,8 @@ def get_current_groups(uid, ldap_dict, log, verbose=True):
         figshare_dict['active'] = False
         return figshare_dict
 
+    revert_command = f'--netid {uid} '
+
     # Check for active group
     active_stem = figshare_stem('active')
     if active_stem in membership:
@@ -250,6 +252,8 @@ def get_current_groups(uid, ldap_dict, log, verbose=True):
         log.warning(f"{uid} not member of figshare:active group")
         figshare_dict['active'] = False
 
+        revert_command += f'--active_remove '
+
     # Extract portal
     portal_stem = figshare_stem('portal')
     portal = [s for s in membership if ((portal_stem in s) and ('grouper' not in s))]
@@ -265,6 +269,8 @@ def get_current_groups(uid, ldap_dict, log, verbose=True):
             if verbose:
                 log.info(f"Current portal is : {figshare_dict['portal']}")
 
+    revert_command += f"--portal {figshare_dict['portal']} "
+
     # Extract quota
     quota_stem = figshare_stem('quota')
     quota = [s for s in membership if ((quota_stem in s) and ('grouper' not in s))]
@@ -280,4 +286,8 @@ def get_current_groups(uid, ldap_dict, log, verbose=True):
             if verbose:
                 log.info(f"Current quota is : {figshare_dict['quota']} bytes")
 
+    if len(quota) != 0:
+        revert_command += f"--quota {figshare_dict['quota']} "
+
+    log.info(f"To revert, use: {revert_command}")
     return figshare_dict