spacepaste

Paste details

reply | raw

  1.  
  2. import pwd
  3. import time
  5. # Required by the copy/pastes below.
  6. from salt.modules.mac_user import _dscl
  7. import salt.utils
  9. # The function to override
  10. from salt.utils import get_group_list as list_groups
  12. __virtualname__ = 'user'
  15. print("LOADED MY MAC USER MODULE")
  17. def __virtual__():
  18. if (__grains__.get('kernel') != 'Darwin' or
  19. __grains__['osrelease_info'] < (10, 7)):
  20. print("Returning False from __virtual__")
  21. return False
  22. else:
  23. print("Returning from __virtual__ with %s" % __virtualname__)
  24. return __virtualname__
  27. # {{{ Copy-pasted identically from the mac_user module (and prints added)
  28. def _format_info(data):
  29. '''
  30. Return user information in a pretty way
  31. '''
  32. print("Called my mac_user _format_info function")
  33. return {'gid': data.pw_gid,
  34. 'groups': list_groups(data.pw_name),
  35. 'home': data.pw_dir,
  36. 'name': data.pw_name,
  37. 'shell': data.pw_shell,
  38. 'uid': data.pw_uid,
  39. 'fullname': data.pw_gecos}
  42. def chgroups(name, groups, append=False):
  43. '''
  44. Change the groups to which the user belongs. Note that the user's primary
  45. group does not have to be one of the groups passed, membership in the
  46. user's primary group is automatically assumed.
  48. groups
  49. Groups to which the user should belong, can be passed either as a
  50. python list or a comma-separated string
  52. append
  53. Instead of removing user from groups not included in the ``groups``
  54. parameter, just add user to any groups for which they are not members
  56. CLI Example:
  58. .. code-block:: bash
  60. salt '*' user.chgroups foo wheel,root
  61. '''
  63. print("Called my mac_user chgroups function")
  64. ### NOTE: **args isn't used here but needs to be included in this
  65. ### function for compatibility with the user.present state
  66. uinfo = info(name)
  67. if not uinfo:
  68. raise CommandExecutionError('User \'{0}\' does not exist'.format(name))
  69. if isinstance(groups, string_types):
  70. groups = groups.split(',')
  72. bad_groups = [x for x in groups if salt.utils.contains_whitespace(x)]
  73. if bad_groups:
  74. raise SaltInvocationError(
  75. 'Invalid group name(s): {0}'.format(', '.join(bad_groups))
  76. )
  77. ugrps = set(list_groups(name))
  78. desired = set(str(x) for x in groups if bool(str(x)))
  79. primary_group = __salt__['file.gid_to_group'](uinfo['gid'])
  80. if primary_group:
  81. desired.add(primary_group)
  82. if ugrps == desired:
  83. return True
  84. # Add groups from which user is missing
  85. for group in desired - ugrps:
  86. _dscl(
  87. ['/Groups/{0}'.format(group), 'GroupMembership', name],
  88. ctype='append'
  89. )
  90. if not append:
  91. # Remove from extra groups
  92. for group in ugrps - desired:
  93. _dscl(
  94. ['/Groups/{0}'.format(group), 'GroupMembership', name],
  95. ctype='delete'
  96. )
  97. time.sleep(1)
  98. return set(list_groups(name)) == desired
  101. def getent(refresh=False):
  102. '''
  103. Return the list of all info for all users
  105. CLI Example:
  107. .. code-block:: bash
  109. salt '*' user.getent
  110. '''
  111. print("Called my mac_user getent function")
  112. if 'user.getent' in __context__ and not refresh:
  113. return __context__['user.getent']
  115. ret = []
  116. for data in pwd.getpwall():
  117. ret.append(_format_info(data))
  118. __context__['user.getent'] = ret
  119. return ret
  122. def info(name):
  123. '''
  124. Return user information
  126. CLI Example:
  128. .. code-block:: bash
  130. salt '*' user.info root
  131. '''
  132. print("Called my mac_user info function")
  133. try:
  134. data = pwd.getpwnam(name)
  135. except KeyError:
  136. return {}
  137. else:
  138. return _format_info(data)
  139. # }}}
  140.  
Powered by spacepaste. Paste removal.