Fix dbapi feature packaging

Author: auxten

MANIFEST.in

@@ -5,12 +5,12 @@
     DatabaseError, OperationalError, IntegrityError, InternalError,
     NotSupportedError, ProgrammingError)
 from . import connections as _orig_conn
+from .. import chdb_version
 
-VERSION = (0, 1, 0, None)
-if VERSION[3] is not None:
-    VERSION_STRING = "%d.%d.%d_%s" % VERSION
+if chdb_version[3] is not None:
+    VERSION_STRING = "%d.%d.%d_%s" % chdb_version
 else:
-    VERSION_STRING = "%d.%d.%d" % VERSION[:3]
+    VERSION_STRING = "%d.%d.%d" % chdb_version[:3]
 
 threadsafety = 1
 apilevel = "2.0"
@@ -71,9 +71,9 @@ def Connect(*args, **kwargs):
 
 
 def get_client_info():  # for MySQLdb compatibility
-    version = VERSION
-    if VERSION[3] is None:
-        version = VERSION[:3]
+    version = chdb_version
+    if chdb_version[3] is None:
+        version = chdb_version[:3]
     return '.'.join(map(str, version))
 
 

chdb/dbapi/__init__.py

@@ -6,6 +6,15 @@ DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 
 cd ${DIR}
 
-echo "include chdb/*.py" > MANIFEST.in
+rm -f MANIFEST.in
+
+echo "include README.md" >> MANIFEST.in
+echo "include LICENSE.txt" >> MANIFEST.in
+echo "graft chdb" >> MANIFEST.in
+echo "global-exclude *.py[cod]" >> MANIFEST.in
+echo "global-exclude __pycache__" >> MANIFEST.in
+echo "global-exclude .DS_Store" >> MANIFEST.in
+echo "global-exclude .git*" >> MANIFEST.in
+echo "global-exclude ~*" >> MANIFEST.in
 export SO_SUFFIX=$(python3 -c "import sysconfig; print(sysconfig.get_config_var('EXT_SUFFIX'))")
 echo "include chdb/_chdb${SO_SUFFIX}" >> MANIFEST.in

gen_manifest.sh

@@ -170,10 +170,19 @@ def build_extensions(self):
         # fix the version in chdb/__init__.py
         versionStr = get_latest_git_tag()
         fix_version_init(versionStr)
+        
+        # scan the chdb directory and add all the .py files to the package
+        pkg_files = []
+        for root, dirs, files in os.walk(libdir):
+            for file in files:
+                if file.endswith(".py"):
+                    pkg_files.append(os.path.join(root, file))
+        pkg_files.append(chdb_so)
+        
         setup(
             packages=['chdb'],
             version=versionStr,
-            package_data={'chdb': [chdb_so]},
+            package_data={'chdb': pkg_files},
             exclude_package_data={'': ['*.pyc', 'src/**']},
             ext_modules=ext_modules,
             python_requires='>=3.7',