make imagneto executable when installing magneto as a pip package

MagnetoTesting · Mar 11, 2015 · eef358d · eef358d
1 parent ca4a78b
commit eef358d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/imagneto → magneto/imagneto.py b/imagneto → magneto/imagneto.py
@@ -1,15 +1,24 @@
 import sys
 from IPython import embed
-from magneto.magneto import Magneto
+
+from .magneto import Magneto
+
+
+m = magneto = None
 
 
 def stop():
-    m.server.stop()
+    if m:
+        m.server.stop()
 
 
-if __name__ == '__main__':
+def main():
     args = sys.argv[1:]
     Magneto.configure(*args)
     m = magneto = Magneto.instance()
 
-    embed()
+    embed()
+
+
+if __name__ == '__main__':
+    main()
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='magneto',
-    version='0.1.3',
+    version='0.1.4',
     description='Magneto - Command your droids.',
     author='EverythingMe',
     author_email='automation@everything.me',
@@ -19,7 +19,8 @@
 
     entry_points={
         'console_scripts': [
-            'magneto = magneto.main:main'
+            'magneto = magneto.main:main',
+            'imagneto = magneto.imagneto:main'
         ]
     },