设置域名

import re
import sys


# 从命令行获取ip地址
if len(sys.argv) < 3:
    raise Exception(‘please input => python3 run.py 127.0.0.1 domain‘)

string_host = sys.argv[1]
pattern = "(?:(?:1[0-9][0-9]\.)|(?:2[0-4][0-9]\.)|(?:25[0-5]\.)|(?:[1-9][0-9]\.)|(?:[0-9]\.)){3}"           "(?:(?:1[0-9][0-9])|(?:2[0-4][0-9])|(?:25[0-5])|(?:[1-9][0-9])|(?:[0-9]))"

ip_pattern = re.compile(pattern)
host = ip_pattern.findall(string_host)[0]
# 判断ip地址是否正确
assert host,‘ip error‘

# 拼接域名
try:
    domain = ‘‘.join(sys.argv[2:])
except IndexError:
    raise Exception(‘please input => python3 run.py 127.0.0.1 domain‘)

try:
    with open(‘C:\Windows\System32\drivers\etc\hosts‘,‘a‘) as f:
        f.write(‘\t{}\t{}\r\n‘.format(host,domain))
except OSError as e:
    err = str(e)
    raise Exception(err)