Update setup.py to correctly pick up Python files

And also re-update package name in utils/build-dists.py so that it is uninstalled successfully Signed-off-by: Rushi Agrawal <[email protected]>
2021-09-19 18:18:09 -07:00
parent 2dc1992032
commit eaa0cd8765
2 changed files with 10 additions and 2 deletions
@@ -41,10 +41,11 @@ with open(join(base_dir, package_name.replace("-", ""), "_version.py")) as f:
 with open(join(base_dir, "README.md")) as f:
    long_description = f.read().strip()

+module_dir = package_name.replace("-", "")
 packages = [
    package
    for package in find_packages(where=".", exclude=("test_opensearchpy*",))
-    if package == package_name or package.startswith(package_name + ".")
+    if package == module_dir or package.startswith(module_dir + ".")
 ]

 install_requires = [
@@ -161,7 +161,14 @@ def test_dist(dist):
            )

        # Uninstall the dist, see that we can't import things anymore
-        run(venv_python, "-m", "pip", "uninstall", "--yes", dist_name)
+        run(
+            venv_python,
+            "-m",
+            "pip",
+            "uninstall",
+            "--yes",
+            dist_name.replace("opensearchpy", "opensearch-py"),
+        )
        run(
            venv_python,
            "-c",