Configuration File! v0.1.1a

integgroll · Jan 26, 2012 · 8ee39f7 · 8ee39f7
1 parent ce10662
commit 8ee39f7
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/wharfwhacker.py b/wharfwhacker.py
@@ -12,11 +12,11 @@
 from threading import Thread
 
 class WharfWhacker:
-  def __init__(self):
+  def __init__(self, config_file = "wharfwhacker.conf"):
     attributes = dict()
 
-    if os.path.isfile("wharfwhacker.conf"):
-      conf = file("wharfwhacker.conf","r").readlines()
+    if os.path.isfile(config_file):
+      conf = file(config_file,"r").readlines()
       for line in conf:
         line = line.rstrip()
         if len(line)>0:
@@ -139,10 +139,10 @@ def add_iptable_rule(self,rule):
 
 
 class Whacker():
-  def __init__(self):
+  def __init__(self, config_file = "wharfwhacker.conf"):
     attributes = dict()
-    if os.path.isfile("wharfwhacker.conf"):
-      conf = file("wharfwhacker.conf","r").readlines()
+    if os.path.isfile(config_file):
+      conf = file(config_file,"r").readlines()
       for line in conf:
         line = line.rstrip()
         if len(line)>0: