Jasata/with_identity.py

## with_identity.py
#! /usr/bin/env python3
#
#   with_identity.py - Jani Tammi <jasata@utu.fi>
#
#   Convenient way to assume another user identity for limited work.
#   Created for installation scripts that run as root.
#
import os
import pwd
import grp

class Identity():
    def __init__(self, user: str, group: str = None):
        self.uid = pwd.getpwnam(user).pw_uid
        if not group:
            self.gid = pwd.getpwnam(user).pw_gid
        else:
            self.gid = grp.getgrnam(group).gr_gid
    def __enter__(self):
        self.original_uid = os.getuid()
        self.original_gid = os.getgid()
        os.setegid(self.uid)
        os.seteuid(self.gid)
    def __exit__(self, type, value, traceback):
        os.seteuid(self.original_uid)
        os.setegid(self.original_gid)

if __name__ == '__main__':

    with Identity('pi', 'www-data'):
        with open("/var/www/hello.html", "w") as f:
            f.write("<html><body>Hello World!</body></html>")

    # 'with' context ends, effecive user and group are restored

# EOF
	#! /usr/bin/env python3
	#
	# with_identity.py - Jani Tammi <jasata@utu.fi>
	#
	# Convenient way to assume another user identity for limited work.
	# Created for installation scripts that run as root.
	#
	import os
	import pwd
	import grp

	class Identity():
	def __init__(self, user: str, group: str = None):
	self.uid = pwd.getpwnam(user).pw_uid
	if not group:
	self.gid = pwd.getpwnam(user).pw_gid
	else:
	self.gid = grp.getgrnam(group).gr_gid
	def __enter__(self):
	self.original_uid = os.getuid()
	self.original_gid = os.getgid()
	os.setegid(self.uid)
	os.seteuid(self.gid)
	def __exit__(self, type, value, traceback):
	os.seteuid(self.original_uid)
	os.setegid(self.original_gid)

	if __name__ == '__main__':

	with Identity('pi', 'www-data'):
	with open("/var/www/hello.html", "w") as f:
	f.write("<html><body>Hello World!</body></html>")

	# 'with' context ends, effecive user and group are restored

	# EOF