Merge pull request #327 from jodal/importlib-resources

Use `importlib.resources` to load data files

src/biip/gs1/_application_identifiers.py

@@ -3,8 +3,8 @@
 from __future__ import annotations
 
 import json
-import pathlib
 from dataclasses import dataclass, field
+from importlib import resources
 
 from biip import ParseError
 
@@ -90,9 +90,8 @@ def __str__(self) -> str:
 
 
 _GS1_APPLICATION_IDENTIFIERS_FILE = (
-    pathlib.Path(__file__).parent / "_application_identifiers.json"
+    resources.files("biip") / "gs1" / "_application_identifiers.json"
 )
-
 _GS1_APPLICATION_IDENTIFIERS = {
     entry.ai: entry
     for entry in [

src/biip/gs1/_prefixes.py

@@ -12,8 +12,8 @@
 
 import json
 import lzma
-import pathlib
 from dataclasses import dataclass
+from importlib import resources
 from typing import Optional, Union
 
 from biip import ParseError
@@ -142,16 +142,15 @@ class _GS1PrefixRange:
     usage: str
 
 
-_GS1_PREFIX_RANGES_FILE = pathlib.Path(__file__).parent / "_prefix_ranges.json"
-
+_GS1_PREFIX_RANGES_FILE = resources.files("biip") / "gs1" / "_prefix_ranges.json"
 _GS1_PREFIX_RANGES = [
     _GS1PrefixRange(**kwargs)
     for kwargs in json.loads(_GS1_PREFIX_RANGES_FILE.read_text())
 ]
 
 _GS1_COMPANY_PREFIX_TRIE_FILE = (
-    pathlib.Path(__file__).parent / "_company_prefix_trie.json.lzma"
+    resources.files("biip") / "gs1" / "_company_prefix_trie.json.lzma"
+)
+_GS1_COMPANY_PREFIX_TRIE: _TrieNode = json.loads(
+    lzma.decompress(_GS1_COMPANY_PREFIX_TRIE_FILE.read_bytes()).decode()
 )
-
-with lzma.open(_GS1_COMPANY_PREFIX_TRIE_FILE) as fh:
-    _GS1_COMPANY_PREFIX_TRIE: _TrieNode = json.load(fh)