GeezerVison is showing github.com — mongo-python-driver/hatch_build.py at master · Rogdham/mongo-python-driver

Expand file tree

hatch_build.py

Copy path

More file actions

36 lines (29 loc) · 1.12 KB

Copy path

36 lines (29 loc) · 1.12 KB

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

"""A custom hatch build hook for pymongo."""

from __future__ import annotations

import os

import subprocess

import sys

from pathlib import Path

from hatchling.builders.hooks.plugin.interface import BuildHookInterface

class CustomHook(BuildHookInterface):

"""The pymongo build hook."""

def initialize(self, version, build_data):

"""Initialize the hook."""

if self.target_name == "sdist":

return

here = Path(__file__).parent.resolve()

sys.path.insert(0, str(here))

subprocess.run([sys.executable, "_setup.py", "build_ext", "-i"], check=True)

# Ensure wheel is marked as binary and contains the binary files.

build_data["infer_tag"] = True

build_data["pure_python"] = False

if os.name == "nt":

patt = ".pyd"

else:

patt = ".so"

for pkg in ["bson", "pymongo"]:

dpath = here / pkg

for fpath in dpath.glob(f"*{patt}"):

relpath = os.path.relpath(fpath, here)

build_data["artifacts"].append(relpath)

build_data["force_include"][relpath] = relpath