1 files changed, 16 insertions, 11 deletions
diff --git a/butler.py b/butler.py
index 72ccd33..2cfc62f 100644
--- a/butler.py
+++ b/butler.py
@@ -4,17 +4,22 @@ from ausbutler.config import load_config
 from ausbutler.goniec import Goniec
 from ausbutler.interface import Interface
 
-args = sys.argv[1:]
-if len(args) == 0:
-    args = ['calculate', 'generate', 'send']
 
-i = Interface(load_config('butler'))
+def main():
+    args = sys.argv[1:]
+    if len(args) == 0:
+        args = ['calculate', 'generate', 'send']
 
-if 'calculate' in args:
-    i.calculate_all()
+    i = Interface(load_config('butler'))
 
-if 'generate' in args:
-    files = i.generate_all()
-    if 'send' in args:
-        g = Goniec(load_config('goniec'))
-        g.send(files)
+    if 'calculate' in args:
+        i.calculate_all()
+
+    if 'generate' in args:
+        files = i.generate_all()
+        if 'send' in args:
+            client = Goniec(load_config('goniec'))
+            client.send(files)
+
+if __name__ == '__main__':
+    main()