sudo実行したpythonプログラム内で一部だけ一般ユーザで実行したい

Question

前提・実現したいこと

rootユーザでしか実行できないライブラリを使うためにsudoでpythonのプログラムを実行する際に、
プログラム内のある操作だけを一般ユーザーとして実行することは可能でしょうか。

補足情報（FW/ツールのバージョンなど）

python3

Answer 1

Unix/Linuxなら、実効ユーザー(euid)を一時的に変えることは可能です。

katsukoさんのコードを少し変更・追加して、

import os

file500="myfile-500"
file0  ="myfile-0"

print('0: uid %d.' % os.geteuid())
if os.path.exists(file500):
    os.remove(file500)
if os.path.exists(file0):
    os.remove(file0)

os.seteuid(500)
print('1: uid %d.' % os.geteuid())
with open(file500,"w") as f:
    f.write("OK\n")

os.seteuid(0)
print('2: uid %d.' % os.geteuid())
with open(file0,"w") as f:
    f.write("OK"\n)

myfile-500のオーナーは500になります。myfile-0のオーナーは0。
グループ(egid)も変えるなら、os.setegid()で。

もちろん、上記はrootで実行します。

Comments

[katsuko]

実効ユーザって、そういう事なんですか。
参考になりやす。

[hirohirohiro3]

無事に解決しました。
お二人ともありがとうございました。

Answer 2

os.setuid(Pythonリファレンスマニュアル)で、現在のプロセスのユーザidを設定できます。

なお、(わかっているかもしれませんが)ユーザを変えられるのはスーパーユーザのみです。
sudoで起動しているのでos.setuidで他のユーザに切り替わる事ができますが、切り替わった後にrootに戻る事はできませんので、もしその必要があればプロセスをforkして処理しなければならないかもしれません。(試してないので出来るかどうかはわかりません)

---

チャチャッと確認できるのでやってみましたが、大丈夫そうですね。

# coding: utf-8
# hoge.py
import os, sys

print('0: uid %d.' % os.getuid())
pid = os.fork()
if pid == 0:
	os.setuid(1000)
	print('1: uid %d.' % os.getuid())
	sys.exit(0)
os.wait()
print('2: uid %d.' % os.getuid())

$ sudo python3 hoge.py
0: uid 0.
1: uid 1000.
2: uid 0.
$