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
|
import pathlib
import io
import os
import re
import sys
import __res
from importlib.abc import ResourceReader
from importlib.metadata import Distribution, DistributionFinder, PackageNotFoundError, Prepared
from importlib.resources.abc import Traversable
ResourceReader.register(__res._ResfsResourceReader)
METADATA_NAME = re.compile('^Name: (.*)$', re.MULTILINE)
class ArcadiaResourceHandle(Traversable):
def __init__(self, key):
self.resfs_key = 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)
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):
for key, path_without_prefix in __res.iter_keys(self.resfs_prefix.encode("utf-8")):
if b"/" in path_without_prefix:
name = path_without_prefix.decode("utf-8").split("/", maxsplit=1)[0]
yield ArcadiaResourceContainer(f"{self.resfs_prefix}{name}/")
else:
yield ArcadiaResourceHandle(key.decode("utf-8"))
def open(self, *args, **kwargs):
raise IsADirectoryError(self.resfs_prefix)
def joinpath(self, *descendants):
if not descendants:
return self
return ArcadiaResourceHandle(os.path.join(self.resfs_prefix, *descendants))
@property
def name(self):
return os.path.basename(self.resfs_prefix[:-1])
class ArcadiaDistribution(Distribution):
def __init__(self, prefix):
self.prefix = prefix
@property
def _path(self):
return pathlib.Path(self.prefix)
def read_text(self, filename):
data = __res.resfs_read(f'{self.prefix}{filename}')
if data:
return data.decode('utf-8')
read_text.__doc__ = Distribution.read_text.__doc__
def locate_file(self, path):
return f'{self.prefix}{path}'
class ArcadiaMetadataFinder(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:
metadata_name = Prepared(metadata_name.group(1))
cls.prefixes[metadata_name.normalized] = resource[:-len('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:
raise PackageNotFoundError(name)
else:
for prefix in sorted(cls.prefixes.values()):
yield prefix
# monkeypatch standart library
import importlib.metadata
importlib.metadata.MetadataPathFinder = ArcadiaMetadataFinder
|
