package this up as a script

2 files changed, 5 insertions, 2 deletions

diff --git a/termcast_client/__init__.py b/termcast_client/__init__.py
index 64e6c2a..c3c8a55 100644
--- a/termcast_client/__init__.py
+++ b/termcast_client/__init__.py
@@ -1,12 +1,13 @@
 import argparse
 import json
 import os
-import pity
 import shutil
 import signal
 import socket
 import sys
 
+from . import pity
+
 class Client(object):
     def __init__(self, host, port, username, password):
         self.host = host
@@ -71,7 +72,7 @@ class Client(object):
     def _build_metadata_string(self, data):
         return b'\033]499;' + json.dumps(data).encode('utf-8') + b'\007'
 
-if __name__ == '__main__':
+def main():
     parser = argparse.ArgumentParser()
     parser.add_argument('--host', default="noway.ratry.ru")
     parser.add_argument('--port', type=int, default=31337)
diff --git a/termcast_client/__main__.py b/termcast_client/__main__.py
new file mode 100644
index 0000000..f30a76b
--- /dev/null
+++ b/termcast_client/__main__.py
@@ -0,0 +1,2 @@
+import termcast_client
+termcast_client.main()