Use __main__.py instead of SwiftStory.py

Signed-off-by: Olivier Gayot <olivier.gayot@sigexec.com>

3 files changed, 4 insertions, 2 deletions

diff --git a/setup.py b/setup.py
index 8d1f2fd..5c4f5ce 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,7 @@ setup(
         data_files = data_files,
         entry_points = {
             'console_scripts': [
-                'swiftstoryd = swiftstory.SwiftStory:main',
+                'swiftstoryd = swiftstory.__main__:main',
             ],
         }
 )
diff --git a/swiftstory/SwiftStory.py b/swiftstory/__main__.py
index 17d88fe..0b5d7c7 100644
--- a/swiftstory/SwiftStory.py
+++ b/swiftstory/__main__.py
@@ -1,5 +1,7 @@
 #!/usr/bin/env python3
 
+""" Entry point to play the game. """
+
 import argparse
 import asyncio
 import contextlib
diff --git a/tests/test_swiftstory.py b/tests/test_swiftstory.py
index ff6d71d..5c1c2bc 100644
--- a/tests/test_swiftstory.py
+++ b/tests/test_swiftstory.py
@@ -1,6 +1,6 @@
 import unittest
 
-import swiftstory.SwiftStory as SwiftStory
+import swiftstory.__main__ as SwiftStory
 from swiftstory.Status import error