167 lines
3.8 KiB
Python
Executable File
167 lines
3.8 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
import argparse
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import time
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='iconograph build_image')
|
|
parser.add_argument(
|
|
'--base-url',
|
|
dest='base_url',
|
|
action='store',
|
|
required=True)
|
|
parser.add_argument(
|
|
'--ca-cert',
|
|
dest='ca_cert',
|
|
action='store',
|
|
required=True)
|
|
parser.add_argument(
|
|
'--device',
|
|
dest='device',
|
|
action='store',
|
|
required=True)
|
|
parser.add_argument(
|
|
'--image-type',
|
|
dest='image_type',
|
|
action='store',
|
|
required=True)
|
|
parser.add_argument(
|
|
'--persistent-percent',
|
|
dest='persistent_percent',
|
|
action='store',
|
|
type=int,
|
|
default=0)
|
|
FLAGS = parser.parse_args()
|
|
|
|
|
|
class Imager(object):
|
|
|
|
def __init__(self, device, persistent_percent, base_url, ca_cert, image_type):
|
|
self._device = device
|
|
self._persistent_percent = persistent_percent
|
|
self._base_url = base_url
|
|
self._ca_cert = ca_cert
|
|
self._image_type = image_type
|
|
|
|
self._icon_server_path = os.path.dirname(sys.argv[0])
|
|
|
|
def _Exec(self, *args, **kwargs):
|
|
print('+', args)
|
|
subprocess.check_call(args, **kwargs)
|
|
|
|
def _PartDev(self, part_num):
|
|
# TODO: better way to figure out what the partition device is
|
|
return '%s%d' % (self._device, part_num)
|
|
|
|
def _PartitionAndMkFS(self):
|
|
self._Exec(
|
|
'parted',
|
|
'--script',
|
|
self._device,
|
|
'mklabel', 'msdos')
|
|
|
|
boot_stop = '%d%%' % (100 - self._persistent_percent)
|
|
self._Exec(
|
|
'parted',
|
|
'--script',
|
|
'--align', 'optimal',
|
|
self._device,
|
|
'mkpart', 'primary', 'ext4', '0%', boot_stop)
|
|
time.sleep(1) # yuck
|
|
self._Exec(
|
|
'mkfs.ext4',
|
|
'-L', 'boot',
|
|
'-F',
|
|
self._PartDev(1))
|
|
|
|
if self._persistent_percent:
|
|
self._Exec(
|
|
'parted',
|
|
'--script',
|
|
'--align', 'optimal',
|
|
self._device,
|
|
'mkpart', 'primary', 'ext4', boot_stop, '100%')
|
|
time.sleep(1) # yuck
|
|
self._Exec(
|
|
'mkfs.ext4',
|
|
'-L', 'persistent',
|
|
'-F',
|
|
self._PartDev(2))
|
|
|
|
def _MountBoot(self):
|
|
root = tempfile.mkdtemp()
|
|
self._rmtree.append(root)
|
|
|
|
self._Exec(
|
|
'mount',
|
|
self._PartDev(1),
|
|
root)
|
|
self._umount.append(root)
|
|
|
|
return root
|
|
|
|
def _InstallGrub(self, root):
|
|
self._Exec(
|
|
'grub-install',
|
|
'--boot-directory', root,
|
|
self._device)
|
|
|
|
def _FetchImages(self, root):
|
|
image_path = os.path.join(root, 'iconograph')
|
|
os.mkdir(image_path)
|
|
|
|
fetcher = os.path.join(self._icon_server_path, '..', 'client', 'fetcher.py')
|
|
|
|
self._Exec(
|
|
fetcher,
|
|
'--image-dir', image_path,
|
|
'--base-url', self._base_url,
|
|
'--ca-cert', self._ca_cert,
|
|
'--image-type', self._image_type)
|
|
|
|
return image_path
|
|
|
|
def _CreateGrubCfg(self, root, image_path):
|
|
grub_cfg_path = os.path.join(root, 'grub', 'grub.cfg')
|
|
|
|
update_grub = os.path.join(self._icon_server_path, '..', 'client', 'update_grub.py')
|
|
|
|
with open(grub_cfg_path, 'w') as fh:
|
|
self._Exec(
|
|
update_grub,
|
|
'--image-dir', image_path,
|
|
'--boot-dir', root,
|
|
stdout=fh)
|
|
|
|
def _Image(self):
|
|
self._PartitionAndMkFS()
|
|
root = self._MountBoot()
|
|
self._InstallGrub(root)
|
|
image_path = self._FetchImages(root)
|
|
self._CreateGrubCfg(root, image_path)
|
|
|
|
def Image(self):
|
|
self._umount = []
|
|
self._rmtree = []
|
|
try:
|
|
self._Image()
|
|
finally:
|
|
for path in reversed(self._umount):
|
|
self._Exec('umount', path)
|
|
for path in self._rmtree:
|
|
shutil.rmtree(path)
|
|
|
|
|
|
def main():
|
|
imager = Imager(FLAGS.device, FLAGS.persistent_percent, FLAGS.base_url, FLAGS.ca_cert, FLAGS.image_type)
|
|
imager.Image()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|