python切换hosts文件代码示例

219 阅读 0 评论 145 点赞

我是靠谱客的博主文艺夕阳，最近开发中收集的这篇文章主要介绍python切换hosts文件代码示例，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

win7以上需要使用管理员权限操作。

复制代码代码如下:

# -*- coding: utf-8 -*-
import os
import glob
import shutil

def format_file_list(files):
        all_files_str = ""
        for i in range(len(files)):
                all_files_str += str(i)+":"+files[i]+"\n"
        return all_files_str

hosts_path = "C:\\Windows\\System32\\drivers\\etc"
files = os.listdir(hosts_path)
os.chdir(hosts_path)

if os.getcwd() != hosts_path:
print("Switch Dir to System32 Error,check permission!\npwd:"+os.getcwd())
exit()

hosts_files = glob.glob("host*")
choosed_file_idx = int(input("Choose Hosts File Index:\n"+format_file_list(hosts_files)))
files_num = len(hosts_files)

if (choosed_file_idx < 0 or choosed_file_idx >= files_num) :
print("Please choose a file in the lists!")
exit()

print("Choosed idx:{0},file:{1}.".format(choosed_file_idx,hosts_files[choosed_file_idx]))
shutil.copy("hosts","hosts.bak")
shutil.copy(hosts_files[choosed_file_idx],"hosts")
print("Copy ok,then flush dns...")
os.system("ipconfig /flushdns")