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
|
#!/usr/bin/python3
import argparse
import json
import os
import subprocess
def create_debianization():
cmd = (
'dh_make -n -i -y;'
'cd debian;'
'rm *ex *EX README.*;'
'cd ..'
)
subprocess.call(cmd, shell=True)
def create_install(work_dirpath):
with open(os.path.join(work_dirpath, 'meta.json'), 'r') as f:
meta = json.load(f)
print('meta: {}'.format(meta))
with open('debian/install', 'w') as f:
package_name = meta['name']
package_dirpath = os.path.join('/usr/share/dlmodels', package_name)
# add meta
f.write('{meta} {pkgdir}\n'.format(
meta='meta.json',
pkgdir=package_dirpath))
# add model
f.write('{model} {pkgdir}\n'.format(
model=meta['model'],
pkgdir=package_dirpath))
# add label
f.write('{label} {pkgdir}\n'.format(
label=meta['label'],
pkgdir=package_dirpath))
# add config(s)
for k, v in meta['config'].items():
f.write('{config} {pkgdir}\n'.format(
config=v,
pkgdir=package_dirpath))
def build_debian_binary_package():
cmd = 'debuild -us -uc'
subprocess.call(cmd, shell=True)
def main():
parser = argparse.ArgumentParser()
parser.add_argument('modeldir_path', help='DL model directory path')
args = parser.parse_args()
work_dirpath = os.path.abspath(args.modeldir_path)
dirname_components = os.path.basename(args.modeldir_path).split('-')
model_name = '-'.join(dirname_components[:-1])
model_version = dirname_components[-1]
os.chdir(work_dirpath)
create_debianization()
create_install(work_dirpath)
build_debian_binary_package()
if __name__ == '__main__':
main()
|
