2d8d3e9a3f
Make the UI consistent by using python standarized logging everywhere. Also log to the systemd-journald service so the logs are kept around when a user deploys a newer version of a branch. Story: 2010867 Task: 48556 Change-Id: I9f203dd6d3e0e17c0563f59efd5f4fa003fa030e Signed-off-by: Charles Short <charles.short@windriver.com>
81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
"""
|
|
Copyright (c) 2023 Wind River Systems, Inc.
|
|
|
|
SPDX-License-Identifier: Apache-2.0
|
|
|
|
"""
|
|
|
|
import logging
|
|
|
|
from rich.console import Console
|
|
from rich.table import Table
|
|
|
|
from apt_ostree.ostree import Ostree
|
|
|
|
|
|
class Remotes:
|
|
def __init__(self, state):
|
|
self.console = Console()
|
|
self.logging = logging.getLogger(__name__)
|
|
self.state = state
|
|
self.ostree = Ostree(self.state)
|
|
|
|
def remotes(self, refs):
|
|
"""Dispaly remotes or refs at remotes."""
|
|
if refs:
|
|
self.remote_refs()
|
|
else:
|
|
self.remote_list()
|
|
|
|
def remote_refs(self):
|
|
"""List all branches associated with a host."""
|
|
refs = self.ostree.remote_refs(self.state.remote)
|
|
if len(refs) == 0:
|
|
return
|
|
|
|
table = Table(box=None)
|
|
table.add_column("Remote", justify="center")
|
|
table.add_column("Checksum", justify="center")
|
|
|
|
url = self.ostree.get_remote_url(self.state.remote)
|
|
self.logging.info(
|
|
f"\nRemote: {self.state.remote} ({url})\n")
|
|
for remote, csum in refs.items():
|
|
table.add_row(remote, csum)
|
|
|
|
if table.columns:
|
|
self.console.print(table)
|
|
self.console.print("\n")
|
|
|
|
def remote_list(self):
|
|
"""Display the registered hosts."""
|
|
remotes = self._remote_list()
|
|
if len(remotes) == 0:
|
|
return
|
|
|
|
table = Table(box=None)
|
|
table.add_column("Remote")
|
|
table.add_column("URL")
|
|
|
|
for r in remotes:
|
|
url = self.ostree.get_remote_url(r)
|
|
table.add_row(r, url)
|
|
|
|
if table.columns:
|
|
self.console.print(table)
|
|
|
|
def remote_add(self):
|
|
"""Add remote with a given URL."""
|
|
self.ostree.remote_add()
|
|
|
|
def remote_remove(self):
|
|
"""Remove remote with a given URL."""
|
|
self.ostree.remote_remove()
|
|
|
|
def _remote_list(self):
|
|
"""List all remotes but exclude the origin."""
|
|
try:
|
|
return [r for r in self.ostree.remotes_list() if r != "origin"]
|
|
except Exception:
|
|
return []
|