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
|
#!/usr/bin/env python3
# Copyright (c) 2009 Giampaolo Rodola'. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""List and pretty print tarball & wheel files in the dist/ directory."""
import argparse
import collections
import os
from psutil._common import bytes2human
from psutil._common import print_color
class Wheel:
def __init__(self, path):
self._path = path
self._name = os.path.basename(path)
def __repr__(self):
return "<%s(name=%s, plat=%s, arch=%s, pyver=%s)>" % (
self.__class__.__name__, self.name, self.platform(), self.arch(),
self.pyver())
__str__ = __repr__
@property
def name(self):
return self._name
def platform(self):
plat = self.name.split('-')[-1]
pyimpl = self.name.split('-')[3]
ispypy = 'pypy' in pyimpl
if 'linux' in plat:
if ispypy:
return 'pypy_on_linux'
else:
return 'linux'
elif 'win' in plat:
if ispypy:
return 'pypy_on_windows'
else:
return 'windows'
elif 'macosx' in plat:
if ispypy:
return 'pypy_on_macos'
else:
return 'macos'
else:
raise ValueError("unknown platform %r" % self.name)
def arch(self):
if self.name.endswith(('x86_64.whl', 'amd64.whl')):
return '64'
if self.name.endswith(("i686.whl", "win32.whl")):
return '32'
if self.name.endswith("arm64.whl"):
return 'arm64'
return '?'
def pyver(self):
pyver = 'pypy' if self.name.split('-')[3].startswith('pypy') else 'py'
pyver += self.name.split('-')[2][2:]
return pyver
def size(self):
return os.path.getsize(self._path)
class Tarball(Wheel):
def platform(self):
return "source"
def arch(self):
return "-"
def pyver(self):
return "-"
def main():
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('dir', nargs="?", default="dist",
help='directory containing tar.gz or wheel files')
args = parser.parse_args()
groups = collections.defaultdict(list)
ls = sorted(os.listdir(args.dir), key=lambda x: x.endswith("tar.gz"))
for name in ls:
path = os.path.join(args.dir, name)
if path.endswith(".whl"):
pkg = Wheel(path)
elif path.endswith(".tar.gz"):
pkg = Tarball(path)
else:
raise ValueError("invalid package %r", path)
groups[pkg.platform()].append(pkg)
tot_files = 0
tot_size = 0
templ = "%-100s %7s %7s %7s"
for platf, pkgs in groups.items():
ppn = "%s (%s)" % (platf, len(pkgs))
s = templ % (ppn, "size", "arch", "pyver")
print_color('\n' + s, color=None, bold=True)
for pkg in sorted(pkgs, key=lambda x: x.name):
tot_files += 1
tot_size += pkg.size()
s = templ % (" " + pkg.name, bytes2human(pkg.size()), pkg.arch(),
pkg.pyver())
if 'pypy' in pkg.pyver():
print_color(s, color='violet')
else:
print_color(s, color='brown')
print_color("\n\ntotals: files=%s, size=%s" % (
tot_files, bytes2human(tot_size)), bold=True)
if __name__ == '__main__':
main()
|