1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
import pathlib
import io
import os
import re
import sys
import warnings
from importlib.metadata import (
Distribution,
DistributionFinder,
Prepared,
)
from importlib.resources.abc import Traversable
import __res
with warnings.catch_warnings(action="ignore", category=DeprecationWarning):
from importlib.abc import ResourceReader
ResourceReader.register(__res._ResfsResourceReader)
METADATA_NAME = re.compile("^Name: (.*)$", re.MULTILINE)
class ArcadiaResource(Traversable):
def __init__(self, resfs_key):
self.resfs_key = resfs_key
def is_file(self):
return True
def is_dir(self):
return False
def open(self, mode="r", *args, **kwargs):
data = __res.find(self.resfs_key.encode("utf-8"))
if data is None:
raise FileNotFoundError(self.resfs_key)
stream = io.BytesIO(data)
if "b" not in mode:
stream = io.TextIOWrapper(stream, *args, **kwargs)
return stream
def joinpath(self, *name):
raise RuntimeError("Cannot traverse into a resource")
def iterdir(self):
return iter(())
@property
def name(self):
return os.path.basename(self.resfs_key)
def __repr__(self):
return f"ArcadiaResource({self.resfs_key!r})"
class ArcadiaResourceContainer(Traversable):
def __init__(self, prefix):
self.resfs_prefix = prefix
def is_dir(self):
return True
def is_file(self):
return False
def iterdir(self):
seen = set()
for key, path_without_prefix in __res.iter_keys(
self.resfs_prefix.encode("utf-8")
):
if b"/" in path_without_prefix:
subdir = path_without_prefix.split(b"/", maxsplit=1)[0].decode("utf-8")
if subdir not in seen:
seen.add(subdir)
yield ArcadiaResourceContainer(f"{self.resfs_prefix}{subdir}/")
else:
yield ArcadiaResource(key.decode("utf-8"))
def open(self, *args, **kwargs):
raise IsADirectoryError(self.resfs_prefix)
@staticmethod
def _flatten(compound_names):
for name in compound_names:
yield from name.split("/")
def joinpath(self, *descendants):
if not descendants:
return self
names = self._flatten(descendants)
target = next(names)
for traversable in self.iterdir():
if traversable.name == target:
if isinstance(traversable, ArcadiaResource):
return traversable
else:
return traversable.joinpath(*names)
raise FileNotFoundError("/".join(self._flatten(descendants)))
@property
def name(self):
return os.path.basename(self.resfs_prefix[:-1])
def __repr__(self):
return f"ArcadiaResourceContainer({self.resfs_prefix!r})"
class ArcadiaDistribution(Distribution):
def __init__(self, prefix):
self._prefix = prefix
self._path = pathlib.Path(prefix)
def read_text(self, filename):
data = __res.resfs_read(f"{self._prefix}{filename}")
if data is not None:
return data.decode("utf-8")
read_text.__doc__ = Distribution.read_text.__doc__
def locate_file(self, path):
return self._path.parent / path
class MetadataArcadiaFinder(DistributionFinder):
prefixes = {}
@classmethod
def find_distributions(cls, context=DistributionFinder.Context()):
found = cls._search_prefixes(context.name)
return map(ArcadiaDistribution, found)
@classmethod
def _init_prefixes(cls):
cls.prefixes.clear()
for resource in __res.resfs_files():
resource = resource.decode("utf-8")
if not resource.endswith("METADATA"):
continue
data = __res.resfs_read(resource).decode("utf-8")
metadata_name = METADATA_NAME.search(data)
if metadata_name:
cls.prefixes[Prepared(metadata_name.group(1)).normalized] = resource.removesuffix("METADATA")
@classmethod
def _search_prefixes(cls, name):
if not cls.prefixes:
cls._init_prefixes()
if name:
try:
yield cls.prefixes[Prepared(name).normalized]
except KeyError:
pass
else:
yield from sorted(cls.prefixes.values())
|
