1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
#!/usr/bin/env python #-*- encoding:gb2312 -*- # Filename: IP.py import sitecustomize import _winreg import ConfigParser from ctypes import * print '進行中網路介面卡檢測,請稍候…' print netCfgInstanceID = None hkey = _winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE, r'SystemCurrentControlSetControlClass{4d36e972-e325-11ce-bfc1-08002be10318}') keyInfo = _winreg.QueryInfoKey(hkey) # 尋找網卡對應的適配器名稱 netCfgInstanceID for index in range(keyInfo[0]): hSubKeyName = _winreg.EnumKey(hkey, index) hSubKey = _winreg.OpenKey(hkey, hSubKeyName) try: hNdiInfKey = _winreg.OpenKey(hSubKey, r'NdiInterfaces') lowerRange = _winreg.QueryValueEx(hNdiInfKey, 'LowerRange') # 檢查是否是乙太網路 if lowerRange[0] == 'ethernet': driverDesc = _winreg.QueryValueEx(hSubKey, 'DriverDesc')[0] print '檢測到網路介面卡名:', driverDesc netCfgInstanceID = _winreg.QueryValueEx(hSubKey, 'NetCfgInstanceID')[0] print '檢測到網路介面卡ID:', netCfgInstanceID if netCfgInstanceID == None: print '沒有找到網路介面卡,程式退出' exit() break _winreg.CloseKey(hNdiInfKey) except WindowsError: print r'Message: No NdiInterfaces Key' # 迴圈結束,目前只提供修改一個網卡IP的功能 _winreg.CloseKey(hSubKey) _winreg.CloseKey(hkey) # 通過修改註冊表設定IP strKeyName = 'SystemCurrentControlSetServicesTcpipParametersInterfaces' + netCfgInstanceID print '網路介面卡的註冊表地址是:n', strKeyName hkey = _winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE, strKeyName, 0, _winreg.KEY_WRITE) config = ConfigParser.ConfigParser() print print '正在開啟IP.ini設定檔…' config.readfp(open('IP.ini')) IPAddress = config.get("school","IPAddress") SubnetMask = config.get("school","SubnetMask") GateWay = config.get("school","GateWay") DNSServer1 = config.get("school","DNSServer1") DNSServer2 = config.get("school","DNSServer2") DNSServer = [DNSServer1,DNSServer2] print '設定檔內設定的資訊如下,請核對:' print print 'IP地址:', IPAddress print '子關掩碼:', SubnetMask print '預設閘道:', GateWay print '主DNS伺服器:', DNSServer1 print '次DNS伺服器:', DNSServer2 print res = raw_input('現在,請您決定:輸入1,則將設定檔寫入系統;輸入2,則將現有的系統設定還原為全部自動擷取;否則程式退出:') if str(res) == '1': try: _winreg.SetValueEx(hkey, 'EnableDHCP', None, _winreg.REG_DWORD, 0x00000000) _winreg.SetValueEx(hkey, 'IPAddress', None, _winreg.REG_MULTI_SZ, [IPAddress]) _winreg.SetValueEx(hkey, 'SubnetMask', None, _winreg.REG_MULTI_SZ, [SubnetMask]) _winreg.SetValueEx(hkey, 'DefaultGateway', None, _winreg.REG_MULTI_SZ, [GateWay]) _winreg.SetValueEx(hkey, 'NameServer', None, _winreg.REG_SZ, ','.join(DNSServer)) except WindowsError: print 'Set IP Error' exit() _winreg.CloseKey(hkey) print '切換成功!重設網路後即可生效' elif str(res) == '2': try: _winreg.SetValueEx(hkey, 'EnableDHCP', None, _winreg.REG_DWORD, 0x00000001) _winreg.SetValueEx(hkey, 'T1', None, _winreg.REG_DWORD, 0x00000000) _winreg.SetValueEx(hkey, 'T2', None, _winreg.REG_DWORD, 0x00000000) _winreg.SetValueEx(hkey, 'NameServer', None, _winreg.REG_SZ, None) _winreg.SetValueEx(hkey, 'DhcpConnForceBroadcastFlag', None, _winreg.REG_DWORD, 0x00000000) _winreg.SetValueEx(hkey, 'Lease', None, _winreg.REG_DWORD, 0x00000000) _winreg.SetValueEx(hkey, 'LeaseObtainedTime', None, _winreg.REG_DWORD, 0x00000000) _winreg.SetValueEx(hkey, 'LeaseTerminatesTime', None, _winreg.REG_DWORD, 0x00000000) except WindowsError: print 'Set IP Error' exit() _winreg.CloseKey(hkey) print '切換成功!重設網路後即可生效' else: print '使用者手動取消,程式退出' exit('') |