Make setup.py install and register modules, as well as VERSION.

Author: progval

bitbot/core_modules/__init__.py

@@ -15,7 +15,7 @@ from bitbot import (
 
 faulthandler.enable()
 
-directory = os.path.dirname(os.path.realpath(__file__))
+directory = os.getcwd()
 home = os.path.expanduser("~")
 default_data = os.path.join(home, ".bitbot")
 

bitbotd

@@ -1,10 +1,41 @@
+import os
 import setuptools
 
 with open("README.md", "r") as readme_file:
     long_description = readme_file.read()
 with open("bitbot/VERSION", "r") as version_file:
     version = version_file.read().strip()
 
+def list_modules(dirname):
+    for (module_name, ext) in map(os.path.splitext, os.listdir(dirname)):
+        if ext in ('', '.py') and not module_name.startswith('_'):
+            yield module_name
+
+package_dir = {
+    # Install bitbot's modules as a separate 'bitbot_modules' package
+    'bitbot_modules': 'modules',
+}
+
+for module_package in setuptools.find_packages('modules'):
+    package_dir[f'bitbot_modules.{module_package}'] = \
+        f'modules/{module_package}'
+
+packages = setuptools.find_packages(exclude=['modules', 'modules.*'])
+
+for package in package_dir:
+    packages.append(package)
+
+entry_points = {
+    'bitbot.core_modules': [
+        f'{module_name} = bitbot.core_modules.{module_name}:Module'
+        for module_name in list_modules('bitbot/core_modules')
+    ],
+    'bitbot.extra_modules': [
+        f'{module_name} = bitbot_modules.{module_name}:Module'
+        for module_name in list_modules('modules')
+    ],
+}
+
 setuptools.setup(
     name="bitbot",
     version=version,
@@ -15,7 +46,21 @@
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/jesopo/bitbot",
-    packages=setuptools.find_packages(),
+
+    packages=packages,
+    include_package_data=True,
+    package_dir=package_dir,
+    package_data={
+        '': ['VERSION'],
+    },
+
+    entry_points=entry_points,
+
+    # We need to read hashflags of modules; and currently this is done by
+    # opening the modules' files. So it doesn't work in a zip, because
+    # module files are then not available.
+    zip_safe=False,
+
     classifiers=[
         "Environment :: Console",
         "Environment :: No Input/Output (Daemon)",