pkg_resources -> importlib.resources

Author: asottile

git_code_debt/generate.py

@@ -3,6 +3,7 @@
 import argparse
 import collections
 import contextlib
+import importlib.resources
 import itertools
 import multiprocessing.pool
 import os.path
@@ -15,8 +16,6 @@
 from typing import Callable
 from typing import TypeVar
 
-import pkg_resources
-
 from git_code_debt import options
 from git_code_debt.discovery import get_metric_parsers_from_args
 from git_code_debt.file_diff_stat import FileDiffStat
@@ -163,13 +162,10 @@ def load_data(
 
 def create_schema(db: sqlite3.Connection) -> None:
     """Creates the database schema."""
-    schema_dir = pkg_resources.resource_filename('git_code_debt', 'schema')
-    schema_files = os.listdir(schema_dir)
+    schema_dir = importlib.resources.files('git_code_debt.schema')
 
-    for sql_file in schema_files:
-        resource_filename = os.path.join(schema_dir, sql_file)
-        with open(resource_filename) as resource:
-            db.executescript(resource.read())
+    for sql_file in schema_dir.iterdir():
+        db.executescript(sql_file.read_text())
 
 
 def get_metrics_info(

git_code_debt/server/app.py

@@ -1,14 +1,13 @@
 from __future__ import annotations
 
 import argparse
+import importlib.resources
 import os.path
-import shutil
 import sqlite3
 from collections.abc import Sequence
 from typing import NoReturn
 
 import flask
-import pkg_resources
 
 from git_code_debt.server.metric_config import Config
 from git_code_debt.server.servlets.changes import changes
@@ -53,12 +52,12 @@ def create_metric_config_if_not_exists() -> None:
         return
 
     print('WARNING: no metric_config.yaml detected.  Creating sample config!')
-    shutil.copyfile(
-        pkg_resources.resource_filename(
-            'git_code_debt.server', 'metric_config.sample.yaml',
-        ),
-        'metric_config.yaml',
+    bts = importlib.resources.read_binary(
+        'git_code_debt.server',
+        'metric_config.sample.yaml',
     )
+    with open('metric_config.yaml', 'wb') as f:
+        f.write(bts)
 
 
 def main(argv: Sequence[str] | None = None) -> NoReturn:

git_code_debt/server/render_mako.py

@@ -1,18 +1,39 @@
 from __future__ import annotations
 
+import importlib.resources
 from typing import Any
 
 import mako.lookup
-import pkg_resources
+from mako.template import Template
 
 
-template_lookup = mako.lookup.TemplateLookup(
-    directories=[
-        pkg_resources.resource_filename('git_code_debt.server', 'templates'),
-    ],
-    default_filters=['html_escape'],
-    imports=['from mako.filters import html_escape'],
-)
+class ImportlibResourcesLookup(mako.lookup.TemplateCollection):
+    def __init__(self, mod: str) -> None:
+        self.mod = mod
+        self._cache: dict[str, Template] = {}
+
+    def get_template(
+            self,
+            uri: str,
+            relativeto: str | None = None,
+    ) -> Template:
+        if relativeto is not None:
+            raise NotImplementedError(f'{relativeto=}')
+
+        try:
+            return self._cache[uri]
+        except KeyError:
+            pth = importlib.resources.files(self.mod).joinpath(uri)
+            with importlib.resources.as_file(pth) as pth:
+                return Template(
+                    filename=str(pth),
+                    lookup=self,
+                    default_filters=['html_escape'],
+                    imports=['from mako.filters import html_escape'],
+                )
+
+
+template_lookup = ImportlibResourcesLookup('git_code_debt.server.templates')
 
 
 def render_template(template_name: str, **env: Any) -> str: