Получение идентификатора пользователя в Python: примеры и варианты

Расширенные примеры помогут глубже понять возможности получения user id в различных сценариях.

1. Получение UID всех пользователей системы (Unix/Linux)

Модуль pwd позволяет перечислить всех пользователей, зарегистрированных в системе. Это может быть полезно для аудита или массовых операций.

import pwd

all_users = pwd.getpwall()
for user in all_users[:5]:  # вывести первых 5
    print(f"Имя: {user.pw_name}, UID: {user.pw_uid}, GID: {user.pw_gid}, домашняя: {user.pw_dir}")

Имя: root, UID: 0, GID: 0, домашняя: /root
Имя: daemon, UID: 1, GID: 1, домашняя: /usr/sbin
Имя: bin, UID: 2, GID: 2, домашняя: /bin
Имя: sys, UID: 3, GID: 3, домашняя: /dev
Имя: sync, UID: 4, GID: 65534, домашняя: /bin

2. Получение UID владельца файла

Функция os.stat() возвращает структуру со свойствами файла, включая поле st_uid. Затем можно преобразовать UID в имя пользователя.

import os
import pwd

filepath = "/etc/passwd"
stat_info = os.stat(filepath)
owner_uid = stat_info.st_uid
owner_name = pwd.getpwuid(owner_uid).pw_name
print(f"Файл {filepath} принадлежит {owner_name} (UID {owner_uid})")

Файл /etc/passwd принадлежит root (UID 0)

3. Использование Flask-Login для получения get_id()

Flask-Login требует реализации метода get_id() в модели пользователя. Этот метод возвращает строковое представление идентификатора.

from flask_login import UserMixin

class User(UserMixin):
    def __init__(self, id, name):
        self.id = id
        self.name = name
    
    def get_id(self):
        # Возвращает строку, Flask-Login использует её для сессии
        return str(self.id)

# Во view:
from flask_login import current_user
@app.route('/me')
def me():
    if current_user.is_authenticated:
        user_id = current_user.get_id()
        return f"Ваш ID: {user_id}"

Ваш ID: 42

4. Извлечение user id из Authorization header с Bearer токеном

В REST API клиенты передают JWT в заголовке. Пример Flask-представления, извлекающего ID из токена.

import jwt
from flask import request, jsonify

@app.route('/api/user')
def api_user():
    auth_header = request.headers.get('Authorization', '')
    if not auth_header.startswith('Bearer '):
        return jsonify(error='Missing token'), 401
    token = auth_header.split(' ')[1]
    try:
        payload = jwt.decode(token, app.config['SECRET_KEY'], algorithms=['HS256'])
        user_id = payload['sub']
        return jsonify(user_id=user_id)
    except (jwt.ExpiredSignatureError, jwt.InvalidTokenError) as e:
        return jsonify(error=str(e)), 401

HTTP/1.1 200 OK
Content-Type: application/json

{"user_id": 123}

5. Получение user id по имени пользователя в Unix

Если известно только имя, можно получить UID через pwd.getpwnam().

import pwd

try:
    user = pwd.getpwnam('www-data')
    print(f"UID: {user.pw_uid}")
except KeyError:
    print("Пользователь не найден")

UID: 33

6. Получение UID процесса по PID

В Linux статус процесса можно прочитать из /proc. Модуль os даёт доступ к информации о текущем процессе, но для произвольного PID потребуется чтение файла.

import os

def get_uid_for_pid(pid):
    try:
        with open(f'/proc/{pid}/status', 'r') as f:
            for line in f:
                if line.startswith('Uid:'):
                    uid = line.split()[1]
                    return int(uid)
    except (FileNotFoundError, PermissionError):
        return None

print(get_uid_for_pid(1))  # UID процесса init (обычно 0)

0

7. Получение user id из базы данных с использованием Django ORM вне представления

Если нужно получить ID пользователя по его имени, используя Django ORM в shell или скрипте.

import os
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'myproject.settings')
import django
django.setup()

from django.contrib.auth.models import User

try:
    user = User.objects.get(username='john')
    print(f"ID: {user.id}")
except User.DoesNotExist:
    print("Пользователь не найден")

ID: 5