python 自动化测试前先切换hosts

自动化测试代码测试过程中，如果多个环境的项目放在一套代码，每次手动切换不是很方便，如下代码则可实现

# hosts.PY(实现代码) 
# coding=utf-8

# '''a测试环境（根据自己的环境设置）'''
zg_sides = ['14.215.177.39 www.baidu.cn',
            '14.215.177.39 test.baidu.cn',
            '14.215.177.39 test1.baidu.cn']

# '''b测试环境（根据自己的环境设置）'''
ab_sides = ['14.215.177.39 www.baidu.cn',
            '14.215.177.39 test.baidu.cn',
 '14.215.177.39 test1.baidu.cn']

class Switch:
    @staticmethod
    def zg_sides_test():
        output = open(r'C:\WINDOWS\system32\drivers\etc\HOSTS', 'w')
        for zg_sid in zg_sides:
            # print(zg_sid)
            output.write(zg_sid)
            output.write("\n")
        output.close()

    @staticmethod
    def ab_sides_test():
        output = open(r'C:\WINDOWS\system32\drivers\etc\HOSTS', 'w')
        for ab_sid in ab_sides:
            # print(ab_sid)
            output.write(ab_sid)
            output.write("\n")
        output.close()

# A.PY(执行代码前先切换hosts)
import hosts
hosts.Switch().ab_sides_test()