"""Windows user profiles collector."""
from winregrc import interface
[docs]
class UserProfile:
"""User profile.
Attributes:
profile_path (str): path of the users profile.
security_identifier (str): security identifier of the user.
"""
[docs]
def __init__(self, security_identifier, profile_path):
"""Initializes an user profile.
Args:
security_identifier (str): security identifier of the user.
profile_path (str): path of the users profile.
"""
super().__init__()
self.profile_path = profile_path
self.security_identifier = security_identifier
[docs]
class UserProfilesCollector(interface.WindowsRegistryKeyCollector):
"""Windows user profiles collector."""
_PROFILE_LIST_KEY_PATH = (
"HKEY_LOCAL_MACHINE\\Software\\Microsoft\\Windows NT\\CurrentVersion\\"
"ProfileList"
)
def _CollectUserProfiles(self, profile_list_key):
"""Collects user profiles.
Args:
profile_list_key (dfwinreg.WinRegistryKey): profile list Windows Registry.
Yields:
UserProfile: an user profile.
"""
for subkey in profile_list_key.GetSubkeys():
profile_image_path = self._GetValueFromKey(subkey, "ProfileImagePath")
yield UserProfile(subkey.name, profile_image_path)
[docs]
def Collect(self, registry):
"""Collects user profiles.
Args:
registry (dfwinreg.WinRegistry): Windows Registry.
Yields:
UserProfile: an user profile.
"""
profile_list_key = registry.GetKeyByPath(self._PROFILE_LIST_KEY_PATH)
if profile_list_key:
yield from self._CollectUserProfiles(profile_list_key)