setux_actions/setux/actions/user.py

from setux.core.action import Actions, Action


class User_(Action):
    '''Ensure User'''

    @property
    def label(self):
        return 'user'

    def check(self):
        ret, out, err = self.target.run(f'which {self.shell}')
        shell = out[0]
        usr = self.target.user.fetch(self.user,
            uid   = self.uid,
            gid   = self.gid,
            shell = shell,
            home  = f'/home/{self.user}',
        )
        ok = usr.check() is True
        if not ok: return False
        ok = usr.home.check() is True
        return ok

    def deploy(self):
        kw = dict()
        try:
            kw['uid'] = self.uid
        except Exception: pass
        try:
            kw['gid'] = self.gid
        except Exception: pass
        try:
            ret, out, err = self.target.run(f'which {self.shell}')
            kw['shell'] = out[0]
        except Exception: pass
        try:
            kw['home'] = f'/home/{self.user}'
        except Exception: pass
        usr = self.target.user.fetch(self.user, **kw)
        return usr.deploy() is True


class Groups(Action):
    '''Ensure Group'''

    @property
    def label(self):
        return 'groups'

    def check(self):
        grp = self.target.groups.fetch(self.user, *self.groups.split())
        return grp.check() is True

    def deploy(self):
        grp = self.target.groups.fetch(self.user, *self.groups.split())
        return grp.deploy() is True


class User(Actions):
    '''Manage User and its Groups'''

    @property
    def label(self):
        return f'User {self.user}'

    @property
    def actions(self):
        return [
            User_,
            Groups,
        ]