python读取环境变量方法示例

发布时间:2019-10-05编辑:脚本学堂
有关python读取环境变量的方法,自定义python环境变量类的实现代码,教你用python代码读取系统环境变量,感兴趣的朋友参考下。

python怎么读取环境变量分享二种方法。

1、使用os.environ来读取和修改环境变量:
 

复制代码 代码示例:
import os
print (os.environ["TEMP"])
mydir = "c:mydir"
os.environ["MYDIR"] = mydir
print (os.environ["MYDIR"])
pathV = os.environ["PATH"]
print (pathV)
os.environ["PATH"]= mydir + ";" + os.environ["PATH"]
print (os.environ["PATH"])

2、自定义python读取环境变量类:
 

复制代码 代码示例:
import os
class MyEnv:
  def __init__(self):
    self.envFile = "c:myenv.txt"
    self.envs = {}
 
  def SetEnvFile(self, filename) :
    self.envFile = filename
       
  def Save(self) :
    outf = open(self.envFile, "w")
    if not outf:
      print ("env file cannot be opened for write!")
    for k, v in self.envs.items() :
      outf.write(k + "=" + v + "n")
    outf.close()
   
  def Load(self) :
    inf = open(self.envFile, "r")
    if not inf:
      print ("env file cannot be opened for open!")
    for line in inf.readlines() :
      k, v = line.split("=")
      self.envs[k] = v
    inf.close()
   
  def ClearAll(self) :
    self.envs.clear()
   
  def AddEnv(self, k, v) :
    self.envs[k] = v
   
  def RemoveEnv(self, k) :
    del self.envs[k]
   
  def PrintAll(self) :
    for k, v in self.envs.items():
      print ( k + "=" + v )
  
if __name__ == "__main__" :
  myEnv = MyEnv()
  myEnv.SetEnvFile("c:myenv.txt")
  myEnv.Load()
  myEnv.AddEnv("MYDIR", "c:mydir")
  myEnv.AddEnv("MYDIR2", "c:mydir2")
  myEnv.AddEnv("MYDIR3", "c:mydir3")
  myEnv.Save()
  myEnv.PrintAll()