1 __author__="*****"  2 __date__ ="*****"  3   4 import os  5 import wmi  6 import shutil  7 import win32wnet  8 import ConfigParser  9  10 REMOTE_PATH = ‘c:\\‘ 11  12 def main(): 13     pIniFileName = "AppConfig.ini"; 14  15     config = ConfigParser.ConfigParser() 16     config.readfp(open(pIniFileName,"rb")) 17  18     LocalDir = config.get("DepolyFiles","LocalDir").strip() 19     RemoteDir = config.get("DepolyFiles","RemoteDir").strip() 20     Servers = config.get("DepolyFiles","Servers").strip() 21     UserId = config.get("DepolyFiles","UserId").strip() 22     Password = config.get("DepolyFiles","Password").strip() 23     UseIISRestart = config.get("DepolyFiles","UseIISRestart").strip() 24  25     print "LocalDir : "+LocalDir 26     print "RemoteDir : "+RemoteDir 27     print "Servers : "+Servers 28     print "UserId : "+UserId 29     print "Password : "+Password 30     print "UseIISRestart : "+UseIISRestart 31  32     pServerSplit = Servers.split(‘|‘); 33     pServerList = list(); 34     for itemServer in pServerSplit: 35         sServerString = itemServer.strip(); 36         if(sServerString.find(‘-‘)>0): 37             tempList = sServerString.split(‘-‘); 38             iStartValue = http://www.mamicode.com/int(tempList[0].split(‘.‘)[3]); 39             iEndValue = http://www.mamicode.com/int(tempList[1]); 40             sFrontString = tempList[0].split(‘.‘)[0]+"."+tempList[0].split(‘.‘)[1]+"."+tempList[0].split(‘.‘)[2]+"."; 41             while iStartValue<=iEndValue: 42                 pServerList.append(sFrontString+str(iStartValue)); 43                 iStartValue=http://www.mamicode.com/iStartValue+1; 44         else: 45             pServerList.append(sServerString); 46  47     for webServer in pServerList: 48         ip = webServer 49         username = UserId 50         password = Password 51         print ‘‘; 52          53         server = WindowsMachine(ip, username, password) 54         resultlist = server.run_remote(‘iisreset‘,async=True, output=True); 55         for linestring in resultlist: 56             print linestring 57  58 def create_file(filename, file_text): 59     f = open(filename, "w") 60     f.write(file_text) 61     f.close()  62  63 class WindowsMachine: 64     def __init__(self, ip, username, password, remote_path=REMOTE_PATH): 65         self.ip = ip 66         self.username = username 67         self.password = password 68         self.remote_path = remote_path 69         try: 70             print "Establishing connection to %s" %self.ip 71             self.connection = wmi.WMI(self.ip, user=self.username, password=self.password) 72             print "Connection established" 73         except wmi.x_wmi: 74             print "Could not connect to machine" 75             raise 76  77     def run_remote(self, cmd, async=False, minimized=True, output=False): 78         """ 79         this function runs cmd on remote machine, using wmi. the function create a .bat file, 80         copies it to the remote machine, and runs the .bat file 81         inputs: cmd - command to run 82                 async - False, waits for command to finish, True, return immidiatly 83                 mimimized - window state 84                 output - True, returns the command‘s output 85         output: return value of the command 86                 output of the command if true 87         """ 88         output_data =http://www.mamicode.com/ None 89         pwd = os.getcwd() 90         bat_local_path = os.path.join(pwd, ‘output_{0}.bat‘.format(self.ip)) 91         bat_remote_path = os.path.join(self.remote_path, ‘output_{0}.bat‘.format(self.ip)) 92         output_remote_path = os.path.join(self.remote_path, ‘output_{0}.out‘.format(self.ip)) 93         output_local_path = os.path.join(pwd, ‘output_{0}.out‘.format(self.ip)) 94         text = cmd + " > " + output_remote_path 95         create_file(bat_local_path, text) 96         self.net_copy(bat_local_path, self.remote_path) 97         batcmd = bat_remote_path 98  99         SW_SHOWMINIMIZED = 0100         if not minimized:101             SW_SHOWMINIMIZED = 1102         print "Executing %s" %cmd103         startup = self.connection.Win32_ProcessStartup.new (ShowWindow=SW_SHOWMINIMIZED)104         process_id, return_value = http://www.mamicode.com/self.connection.Win32_Process.Create (CommandLine=batcmd, ProcessStartupInformation=startup)105         if async:106             watcher = self.connection.watch_for (107                                   notification_type="Deletion",108                                   wmi_class="Win32_Process",109                                   delay_secs=1,110                                   )111             watcher ()112 113         if output:114             print ‘copying back ‘ + output_remote_path115             self.net_copy_back(output_remote_path, output_local_path)116             output_data = http://www.mamicode.com/open(output_local_path, ‘r‘)117             output_data = http://www.mamicode.com/"".join(output_data.readlines())118             self.net_delete(output_remote_path)119         self.net_delete(bat_remote_path)120         return return_value, output_data121 122     def net_copy(self, source, dest_dir, move=False):123         """ Copies files or directories to a remote computer. """124         print "Start copying files to " + self.ip125         if self.username == ‘‘:126             if not os.path.exists(dest_dir):127                 os.makedirs(dest_dir)128             else:129                 # Create a directory anyway if file exists so as to raise an error.130                  if not os.path.isdir(dest_dir):131                      os.makedirs(dest_dir)132             shutil.copy(source, dest_dir)133 134         else:135             self._wnet_connect()136 137             dest_dir = self._covert_unc(dest_dir)138 139             # Pad a backslash to the destination directory if not provided.140             if not dest_dir[len(dest_dir) - 1] == ‘\\‘:141                 dest_dir = ‘‘.join([dest_dir, ‘\\‘])142 143             # Create the destination dir if its not there.144             if not os.path.exists(dest_dir):145                 os.makedirs(dest_dir)146             else:147                 # Create a directory anyway if file exists so as to raise an error.148                  if not os.path.isdir(dest_dir):149                      os.makedirs(dest_dir)150 151             if move:152                 shutil.move(source, dest_dir)153             else:154                 shutil.copy(source, dest_dir)155 156     def net_copy_back(self, source_file, dest_file):157         """ Copies files or directories to a remote computer. """158         print "Start copying files " + source_file +  " back from " + self.ip159         if self.username == ‘‘:160             shutil.copy(source_file, dest_file)161         else:162             self._wnet_connect()163             source_unc =  self._covert_unc(source_file)164             shutil.copyfile(source_unc, dest_file)165 166     def _wnet_connect(self):167         unc = ‘‘.join([‘\\\\‘, self.ip])168         try:169             win32wnet.WNetAddConnection2(0, None, unc, None, self.username, self.password)170         except Exception, err:171             if isinstance(err, win32wnet.error):172                 # Disconnect previous connections if detected, and reconnect.173                 if err[0] == 1219:174                     win32wnet.WNetCancelConnection2(unc, 0, 0)175                     return self._wnet_connect(self)176             raise err177 178     def _covert_unc(self, path):179         """ Convert a file path on a host to a UNC path."""180         return ‘‘.join([‘\\\\‘, self.ip, ‘\\‘, path.replace(‘:‘, ‘$‘)])181 182     def copy_folder(self, local_source_folder, remote_dest_folder):183         files_to_copy = os.listdir(local_source_folder)184         for file in files_to_copy:185             file_path = os.path.join(local_source_folder, file)186             print "Copying " + file187             if(os.path.isdir(file_path)):188                 self.copy_folder(file_path,remote_dest_folder+"\\"+file);189             else:190                 try:191                     self.net_copy(file_path, remote_dest_folder)192                 except WindowsError:193                     print ‘could not connect to ‘, self.ip194                 except IOError:195                     print ‘One of the files is being used on ‘, self.ip, ‘, skipping the copy procedure‘196 197     def net_delete(self, path):198         """ Deletes files or directories on a remote computer. """199         try:200             if self.username == ‘‘:201                 os.remove(path)202 203             else:204                 self._wnet_connect()205 206                 path = self._covert_unc(path)207                 if os.path.exists(path):208                     # Delete directory tree if object is a directory.209                     if os.path.isfile(path):210                         os.remove(path)211                     else:212                         shutil.rmtree(path)213                 else:214                     # Remove anyway if non-existent so as to raise an error.215                     os.remove(path)216         except:217             print ‘could not delete ‘, path218 219 if __name__ == "__main__":220     main()