Merge pull request #18 from kurtmckee/include-metadata

Support calls to `importlib.metadata.version()`

Author: kurtmckee

changelog.d/20241028_183138_kurtmckee_include_metadata.rst

@@ -0,0 +1,9 @@
+Added
+-----
+
+*   Support calls to ``importlib.metadata.version()``.
+
+Changed
+-------
+
+*   Bundle ``.dist-info/`` metadata directories into databases.

src/sqliteimport/accessor.py

@@ -63,3 +63,15 @@ def add_file(self, directory: pathlib.Path, file: pathlib.Path) -> None:
                 (directory / file).read_text(),
             ),
         )
+
+    def get_file(self, path_like: str) -> str:
+        source: str = self.connection.execute(
+            """
+            SELECT
+                source
+            FROM code
+            WHERE path LIKE ?;
+            """,
+            (path_like,),
+        ).fetchone()[0]
+        return source

src/sqliteimport/cli.py

@@ -47,7 +47,7 @@ def bundle(directory: pathlib.Path, database: pathlib.Path) -> None:
     files = []
     for path in paths:
         rel_path = path.relative_to(directory)
-        if rel_path.suffix in (".dist-info", ".so"):
+        if rel_path.suffix in {".so"}:
             continue
         if rel_path.name == "__pycache__":
             continue

src/sqliteimport/importer.py

@@ -6,13 +6,16 @@
 
 import importlib.abc
 import importlib.machinery
+import importlib.metadata
 import os.path
 import pathlib
 import sqlite3
 import sys
 import types
 import typing
 
+from sqliteimport.accessor import Accessor
+
 
 class SqliteFinder(importlib.abc.MetaPathFinder):
     def __init__(self, database: pathlib.Path | sqlite3.Connection) -> None:
@@ -23,6 +26,7 @@ def __init__(self, database: pathlib.Path | sqlite3.Connection) -> None:
             _, _, path = database.execute("PRAGMA database_list;").fetchone()
             self.database = pathlib.Path(path)
             self.connection = database
+        self.accessor = Accessor(self.connection)
 
     def find_spec(
         self,
@@ -44,6 +48,15 @@ def find_spec(
         )
         return spec
 
+    def find_distributions(
+        self,
+        context: importlib.metadata.DistributionFinder.Context | None = None,
+    ) -> typing.Generator[SqliteDistribution]:
+        if context is None:
+            context = importlib.metadata.DistributionFinder.Context()
+        if context.name is not None:
+            yield SqliteDistribution(context.name, self.connection)
+
 
 class SqliteLoader(importlib.abc.Loader):
     def __init__(self, source: str) -> None:
@@ -62,3 +75,16 @@ def load(database: pathlib.Path | str | sqlite3.Connection) -> None:
             raise FileNotFoundError(f"{database} must exist.")
         database = pathlib.Path(database)
     sys.meta_path.append(SqliteFinder(database))
+
+
+class SqliteDistribution(importlib.metadata.Distribution):
+    def __init__(self, name: str, connection: sqlite3.Connection):
+        self.__name = name
+        self.__connection = connection
+        self.__accessor = Accessor(connection)
+
+    def locate_file(self, path: typing.Any) -> pathlib.Path:
+        raise NotImplementedError()
+
+    def read_text(self, filename: str) -> str:
+        return self.__accessor.get_file(f"{self.__name}-%/{filename}")