How to Set Environment Variable in Windows using Python?

In the previous article, we have seen how to set the environment variables with GUI, PowerShell, and CMD. In this section, let’s continue to set the environment variable programmatically using Python

Set Environment Variable using Python

a. Process Level

Setting up environment variables at the process level is easy with Python. We need to import the OS module and use the environment function to set the value.

For Example,

import os os.environ["azSub"] = "TestSub1" print(os.environ["azSub"])

Setting up environment variables at the machine or user levels isn’t straightforward in Python like PowerShell. We need to set the registry keys for them. Let’s have a look.

b. Machine-Level

To set environment variables at the machine level with Python, the registry value should be set. The path for the registry value is, SYSTEM\CurrentControlSet\Control\Session Manager\Environment.

import os import winreg def set_machine_env_variable(variable_name, variable_value): key_path = r"SYSTEM\CurrentControlSet\Control\Session Manager\Environment" try: # Open the registry key for environment variables with write access key = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, key_path, 0, winreg.KEY_SET_VALUE) # Set the environment variable in the registry winreg.SetValueEx(key, variable_name, 0, winreg.REG_EXPAND_SZ, variable_value) # Close the registry key winreg.CloseKey(key) print(f"Machine-level environment variable {variable_name} set to {variable_value}.") except Exception as e: print(f"Error: {e}") if __name__ == "__main__": variable_name = 'azSub' variable_value = 'TestSub4' set_machine_env_variable(variable_name, variable_value)

c. User-Level

Similarly, for the user level, you just need to change the registry key path as shown below.

import os import winreg def set_user_env_variable(variable_name, variable_value): key_path = r"Environment" try: # Open the registry key for environment variables with write access key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, key_path, 0, winreg.KEY_SET_VALUE) # Set the environment variable in the registry winreg.SetValueEx(key, variable_name, 0, winreg.REG_EXPAND_SZ, variable_value) # Close the registry key winreg.CloseKey(key) print(f"User-level environment variable {variable_name} set to {variable_value}.") except Exception as e: print(f"Error: {e}") if __name__ == "__main__": variable_name = 'azsub' variable_value = 'TestSub4' set_user_env_variable(variable_name, variable_value)

You can get the code at the GitHub repo below.

https://github.com/chiragce17/MyPublicRepo/tree/main/Python/Windows/EnvironmentVariable

The post How to Set Environment Variable in Windows using Python? appeared first on The Automation Code.