Merge pull request #62 from kamyarrasta/fix/openfile

Open file using with block

routersploit/utils.py

@@ -126,11 +126,7 @@ def multi(fn):
 
             _, _, feed_path = self.target.partition("file://")
             try:
-                file_handler = open(feed_path, 'r')
-            except IOError:
-                print_error("Could not read file: {}".format(self.target))
-                return
-
+                with open(feed_path) as file_handler:
                     for target in file_handler:
                         target = target.strip()
                         if not target:
@@ -140,12 +136,16 @@ def multi(fn):
                             self.port = port
                         else:
                             self.port = original_port
-                print_status("Attack against: {}:{}".format(self.target, self.port))
+                        print_status("Attack against: {}:{}".format(self.target,
+                                                                    self.port))
                         fn(self, *args, **kwargs)
                     self.target = original_target
                     self.port = original_port
-            file_handler.close()
                     return  # Nothing to return, ran multiple times.
+            except IOError:
+                print_error("Could not read file: {}".format(self.target))
+                return
+
         else:
             return fn(self, *args, **kwargs)
     return wrapper