[python] PyInstaller를 사용하여 작성한 프로그램에서 사용자의 권한을 관리하는 방법은?

29 Nov 2023

python

여러가지 방법으로 사용자 권한을 관리할 수 있지만, 일반적으로 다음과 같은 방법들이 사용됩니다.

사용자 계정 제어: 프로그램을 실행하는 사용자 계정의 권한을 확인하는 방법입니다. Windows 운영체제에서는 win32api 모듈을 사용하여 현재 실행 중인 사용자 계정의 권한을 확인할 수 있습니다. 이를 통해 관리자 권한이 필요한 작업을 수행하기 전에 사용자에게 경고를 표시하거나 작업을 중지할 수 있습니다.

import win32api
import win32security

# 현재 실행 중인 사용자 계정의 이름을 가져옵니다.
username = win32api.GetUserName()

# 사용자 계정의 속성을 확인하여 권한을 확인합니다.
sid = win32security.GetNamedSecurityInfo(username, win32security.SE_FILE_OBJECT,
                                         win32security.OWNER_SECURITY_INFORMATION).GetSecurityDescriptorOwner()
user_account = win32security.LookupAccountSid(None, sid)

# 관리자 권한인지 확인합니다.
is_admin = (
    win32security.GetSecurityInfo(
        win32security.HKEY_LOCAL_MACHINE,
        win32security.SE_REGISTRY_KEY,
        win32security.DACL_SECURITY_INFORMATION
    ).GetSecurityDescriptorDacl() is not None
)

if is_admin:
    print(f'{user_account[0]}님은 관리자 권한을 가지고 있습니다.')
else:
    print(f'{user_account[0]}님은 일반 사용자 권한을 가지고 있습니다.')

프로그램 권한 설정: 프로그램 실행 파일의 권한을 설정함으로써 사용자가 프로그램을 관리자 권한으로 실행할 수 있도록 할 수 있습니다. 이를 위해서는 프로그램의 실행 파일을 관리자 권한으로 실행하여 실행 파일의 속성에서 “관리자 권한으로 실행” 옵션을 체크해야 합니다. 이렇게 설정된 실행 파일은 사용자가 일반적인 방법으로 실행할 때도 관리자 권한으로 실행됩니다.
작업 단계별 권한 체크: 프로그램의 특정 작업을 수행하기 전에 해당 작업에 필요한 권한을 확인하는 방법입니다. 사용자에게 작업을 수행할 권한이 있는지 묻고, 권한이 없는 경우 작업을 중지하거나 경고를 표시할 수 있습니다. 이를 위해 프로그램 내에 사용자 권한을 확인하는 함수를 작성하고, 필요한 작업을 수행하기 전에 이 함수를 호출하여 권한을 확인할 수 있습니다.

def check_permission():
    # 사용자 권한 확인 로직 작성
    if user_permission:
        return True
    else:
        return False

# 작업을 수행하기 전에 사용자 권한을 확인합니다.
if check_permission():
    # 작업 수행
else:
    print("작업을 수행하기 위한 권한이 없습니다.")

위의 방법들 중 필요에 맞게 선택하여 사용자 권한을 관리할 수 있습니다. 프로그램의 보안을 강화하고 사용자의 권한을 효과적으로 관리함으로써 불필요한 보안 문제를 예방할 수 있습니다.

참고 자료: PyInstaller 공식 문서