mirror of
https://github.com/tuna/tunasync-scripts.git
synced 2025-04-20 12:24:12 +00:00
82 lines
1.8 KiB
Python
Executable File
82 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import json
|
|
import re
|
|
|
|
|
|
def version(v: str):
|
|
return tuple(map(int, v.split('.')))
|
|
|
|
|
|
def generate_fileindex(filelist):
|
|
fname_re = re.compile(
|
|
r'''
|
|
(?P<tensorflow>[a-z_]+?)- # match tensorflow and tensorflow_gpu
|
|
(?P<tfver>[\d.]+?)- # match version
|
|
(?P<python>[cpy\d]+?)- # python version, 'rc's are ignored
|
|
(?P<rest>.+?\.whl) # everything else
|
|
''',
|
|
re.VERBOSE,
|
|
)
|
|
|
|
versions = set([])
|
|
pythons = {
|
|
'linux': set([]),
|
|
'mac': set([]),
|
|
'windows': set([])
|
|
}
|
|
pkglist = []
|
|
|
|
min_ver = version('0.10.0')
|
|
|
|
for fpath in filelist:
|
|
tokens = fpath.split('/')
|
|
if len(tokens) != 3:
|
|
continue
|
|
os, xpu, fname = tokens
|
|
m = fname_re.match(fname)
|
|
if m is None:
|
|
continue
|
|
|
|
if not m.group('tensorflow').startswith('tensorflow'):
|
|
continue
|
|
|
|
tfver = m.group('tfver')
|
|
if version(tfver) < min_ver:
|
|
continue
|
|
versions.add(tfver)
|
|
|
|
pyver = m.group('python')
|
|
pythons[os].add(pyver)
|
|
|
|
pkglist.append({
|
|
'os': os,
|
|
'xpu': xpu,
|
|
'python': pyver,
|
|
'version': tfver,
|
|
'filename': fname,
|
|
})
|
|
|
|
pythons = {
|
|
k: sorted(list(v), reverse=True)
|
|
for k, v in pythons.items()
|
|
}
|
|
versions = sorted(list(versions), key=lambda x: version(x), reverse=True)
|
|
|
|
index = {
|
|
'versions': versions,
|
|
'pythons': pythons,
|
|
'pkglist': pkglist,
|
|
}
|
|
|
|
return index
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import fileinput
|
|
filelist = [line.strip() for line in fileinput.input()]
|
|
index = generate_fileindex(filelist)
|
|
print(json.dumps(index, sort_keys=True, indent=2))
|
|
|
|
|
|
# vim: ts=4 sw=4 sts=4 expandtab
|