mirror of
https://github.com/xcat2/confluent.git
synced 2025-01-15 12:17:47 +00:00
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
|
import optparse
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
import tempfile
|
||
|
|
try:
|
||
|
|
import configparser
|
||
|
|
except ImportError:
|
||
|
|
import ConfigParser as configparser
|
||
|
|
|
||
|
|
|
||
|
|
def create_yumconf(sourcedir):
|
||
|
|
hdl, name = tempfile.mkstemp(prefix='genimage-yumconfig.')
|
||
|
|
yumconf = os.fdopen(hdl, mode='w+')
|
||
|
|
if os.path.exists(sourcedir + '/repodata'):
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
c = configparser.ConfigParser()
|
||
|
|
c.read(sourcedir + '/.treeinfo')
|
||
|
|
for sec in c.sections():
|
||
|
|
if sec.startswith('variant-'):
|
||
|
|
try:
|
||
|
|
repopath = c.get(sec, 'repository')
|
||
|
|
except Exception:
|
||
|
|
continue
|
||
|
|
_, varname = sec.split('-', 1)
|
||
|
|
yumconf.write('[genimage-{0}]\n'.format(varname.lower()))
|
||
|
|
yumconf.write('name=Local install repository for {0}\n'.format(varname))
|
||
|
|
currdir = os.path.join(sourcedir, repopath)
|
||
|
|
yumconf.write('baseurl={0}\n'.format(currdir))
|
||
|
|
yumconf.write('enabled=1\ngpgcheck=0\n\n')
|
||
|
|
return name
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = optparse.OptionParser()
|
||
|
|
parser.add_option('-s', '--source', help='Directory to pull installation from (e.g. /var/lib/confluent/distributions/rocky-8.3-x86_64')
|
||
|
|
(opts, args) = parser.parse_args()
|
||
|
|
yumargs = ['yum', '--installroot={0}'.format(args[0])]
|
||
|
|
if opts.source:
|
||
|
|
yumconfig = create_yumconf(opts.source)
|
||
|
|
yumargs.extend(['-c', yumconfig, '--disablerepo=*', '--enablerepo=genimage-*'])
|
||
|
|
yumargs.append('install')
|
||
|
|
with open(os.path.join(os.path.dirname(__file__), 'pkglist'), 'r') as pkglist:
|
||
|
|
pkgs = pkglist.read()
|
||
|
|
pkgs = pkgs.split()
|
||
|
|
yumargs.extend(pkgs)
|
||
|
|
subprocess.check_call(yumargs)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|