python package

partybot/pnutbot.py

@@ -10,11 +10,10 @@ import random
 import re
 import argparse
 
-# from database import db_session, init_db
 from sqlalchemy import create_engine, and_
 from sqlalchemy.orm import scoped_session, sessionmaker
 from sqlalchemy.ext.declarative import declarative_base
-from models import Base, Karma, Optout, Queue, Preferences, MdnpRequests
+from partybot.models import Base, Karma, Optout, Queue, Preferences, MdnpRequests
 
 _startup = threading.Event()
 _shutdown = threading.Event()
@@ -353,8 +352,7 @@ def on_open(ws):
     t = threading.Thread(target=run)
     t.start()
 
-if __name__ == "__main__":
+def main():
-
     logger = logging.getLogger()
     a_parser = argparse.ArgumentParser()
     a_parser.add_argument(
@@ -394,3 +392,7 @@ if __name__ == "__main__":
     while r:
         _error.clear()
         r = ws.run_forever()
+
+if __name__ == "__main__":
+    main()
+

setup.py

@@ -0,0 +1,18 @@
+from setuptools import setup
+
+setup(
+    name='partybot',
+    py_modules=['models','pnutbot'],
+    install_requires=[
+        'PyYAML',
+        'pnutpy',
+        'requests',
+        'SQLAlchemy',
+        'websocket-client',
+    ],
+    entry_points={
+        'console_scripts': [
+            'partybot = partybot.pnutbot:main'
+        ]
+    },
+)