mirror of https://github.com/morpheus65535/bazarr
92 lines
3.0 KiB
Python
92 lines
3.0 KiB
Python
# Copyright 2013 Donald Stufft
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# https://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
import io
|
|
import os
|
|
import re
|
|
import zipfile
|
|
from typing import List, Optional
|
|
|
|
from pkginfo import distribution
|
|
|
|
from twine import exceptions
|
|
|
|
# Monkeypatch Metadata 2.0 support
|
|
distribution.HEADER_ATTRS_2_0 = distribution.HEADER_ATTRS_1_2
|
|
distribution.HEADER_ATTRS.update({"2.0": distribution.HEADER_ATTRS_2_0})
|
|
|
|
|
|
wheel_file_re = re.compile(
|
|
r"""^(?P<namever>(?P<name>.+?)(-(?P<ver>\d.+?))?)
|
|
((-(?P<build>\d.*?))?-(?P<pyver>.+?)-(?P<abi>.+?)-(?P<plat>.+?)
|
|
\.whl|\.dist-info)$""",
|
|
re.VERBOSE,
|
|
)
|
|
|
|
|
|
class Wheel(distribution.Distribution):
|
|
def __init__(self, filename: str, metadata_version: Optional[str] = None) -> None:
|
|
self.filename = filename
|
|
self.basefilename = os.path.basename(self.filename)
|
|
self.metadata_version = metadata_version
|
|
self.extractMetadata()
|
|
|
|
@property
|
|
def py_version(self) -> str:
|
|
wheel_info = wheel_file_re.match(self.basefilename)
|
|
if wheel_info is None:
|
|
return "any"
|
|
else:
|
|
return wheel_info.group("pyver")
|
|
|
|
@staticmethod
|
|
def find_candidate_metadata_files(names: List[str]) -> List[List[str]]:
|
|
"""Filter files that may be METADATA files."""
|
|
tuples = [x.split("/") for x in names if "METADATA" in x]
|
|
return [x[1] for x in sorted([(len(x), x) for x in tuples])]
|
|
|
|
def read(self) -> bytes:
|
|
fqn = os.path.abspath(os.path.normpath(self.filename))
|
|
if not os.path.exists(fqn):
|
|
raise exceptions.InvalidDistribution("No such file: %s" % fqn)
|
|
|
|
if fqn.endswith(".whl"):
|
|
archive = zipfile.ZipFile(fqn)
|
|
names = archive.namelist()
|
|
|
|
def read_file(name: str) -> bytes:
|
|
return archive.read(name)
|
|
|
|
else:
|
|
raise exceptions.InvalidDistribution(
|
|
"Not a known archive format for file: %s" % fqn
|
|
)
|
|
|
|
try:
|
|
for path in self.find_candidate_metadata_files(names):
|
|
candidate = "/".join(path)
|
|
data = read_file(candidate)
|
|
if b"Metadata-Version" in data:
|
|
return data
|
|
finally:
|
|
archive.close()
|
|
|
|
raise exceptions.InvalidDistribution("No METADATA in archive: %s" % fqn)
|
|
|
|
def parse(self, data: bytes) -> None:
|
|
super().parse(data)
|
|
|
|
fp = io.StringIO(distribution.must_decode(data))
|
|
msg = distribution.parse(fp)
|
|
self.description = msg.get_payload()
|