[eit.commands.push] new command "eit push"
This commit is contained in:
Fabio Erculiani
@@ -0,0 +1,261 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
|
||||
@author: Fabio Erculiani <lxnay@sabayon.org>
|
||||
@contact: lxnay@sabayon.org
|
||||
@copyright: Fabio Erculiani
|
||||
@license: GPL-2
|
||||
|
||||
B{Entropy Infrastructure Toolkit}.
|
||||
|
||||
"""
|
||||
import sys
|
||||
import os
|
||||
import argparse
|
||||
import tempfile
|
||||
|
||||
from entropy.const import etpConst
|
||||
from entropy.i18n import _
|
||||
from entropy.output import darkgreen, teal, red, darkred, brown, blue, bold
|
||||
from entropy.transceivers import EntropyTransceiver
|
||||
from entropy.server.interfaces import ServerSystemSettingsPlugin
|
||||
from entropy.server.interfaces.rss import ServerRssMetadata
|
||||
|
||||
from eit.commands.descriptor import EitCommandDescriptor
|
||||
from eit.commands.command import EitCommand
|
||||
|
||||
|
||||
class EitPush(EitCommand):
|
||||
"""
|
||||
Main Eit reset command.
|
||||
"""
|
||||
|
||||
NAME = "push"
|
||||
ALIASES = ["pull","sync"]
|
||||
DEFAULT_REPO_COMMIT_MSG = """
|
||||
# This is Entropy Server repository commit message handler.
|
||||
# Please friggin' enter the commit message for your changes. Lines starting
|
||||
# with '#' will be ignored. To avoid encoding issue, write stuff in plain ASCII.
|
||||
"""
|
||||
|
||||
def __init__(self, args):
|
||||
EitCommand.__init__(self, args)
|
||||
self._ask = True
|
||||
self._pretend = False
|
||||
self._all = False
|
||||
self._repositories = []
|
||||
self._cleanup_only = False
|
||||
self._as_repository_id = None
|
||||
|
||||
def parse(self):
|
||||
descriptor = EitCommandDescriptor.obtain_descriptor(
|
||||
EitPush.NAME)
|
||||
parser = argparse.ArgumentParser(
|
||||
description=descriptor.get_description(),
|
||||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||||
prog="%s %s" % (sys.argv[0], EitPush.NAME))
|
||||
|
||||
parser.add_argument("repo", nargs='*', default=None,
|
||||
metavar="<repo>", help=_("repository"))
|
||||
parser.add_argument("--quick", action="store_true",
|
||||
default=False,
|
||||
help=_("no stupid questions"))
|
||||
parser.add_argument("--all", action="store_true",
|
||||
default=False,
|
||||
help=_("push all the repositories"))
|
||||
parser.add_argument("--as", metavar="<repo>", default=None,
|
||||
help=_("push as fake repository"), dest="asrepo")
|
||||
|
||||
try:
|
||||
nsargs = parser.parse_args(self._args)
|
||||
except IOError as err:
|
||||
return parser.print_help, []
|
||||
|
||||
self._ask = not nsargs.quick
|
||||
self._all = nsargs.all
|
||||
self._repositories += nsargs.repo
|
||||
self._as_repository_id = nsargs.asrepo
|
||||
|
||||
return self._call_locked, [self._push, None]
|
||||
|
||||
def _push(self, entropy_server):
|
||||
"""
|
||||
Main Eit push code.
|
||||
"""
|
||||
if not self._repositories and (not self._all):
|
||||
# pick default if none specified
|
||||
self._repositories.append(entropy_server.repository())
|
||||
if not self._repositories and self._all:
|
||||
self._repositories.extend(entropy_server.repositories())
|
||||
|
||||
if self._as_repository_id is not None and \
|
||||
len(self._repositories) > 1:
|
||||
# cannot push as fake repository when
|
||||
# more than one repository is specified
|
||||
entropy_server.output(
|
||||
_("multiple repositories and --as ain't friends"),
|
||||
importance=1, level="error")
|
||||
return 1
|
||||
|
||||
for repository_id in self._repositories:
|
||||
# avoid __default__
|
||||
if repository_id == etpConst['clientserverrepoid']:
|
||||
continue
|
||||
rc = self._push_repo(entropy_server, repository_id)
|
||||
if rc != 0:
|
||||
return rc
|
||||
|
||||
return 0
|
||||
|
||||
def _push_repo(self, entropy_server, repository_id):
|
||||
"""
|
||||
Push (actually sync) the repository.
|
||||
"""
|
||||
done = True
|
||||
if not self._cleanup_only:
|
||||
rc = self.__push_repo(entropy_server, repository_id)
|
||||
if rc != 0:
|
||||
done = False
|
||||
|
||||
if done:
|
||||
done = entropy_server.Mirrors.tidy_mirrors(repository_id,
|
||||
ask = self._ask, pretend = self._pretend)
|
||||
|
||||
if done:
|
||||
return 0
|
||||
return 1
|
||||
|
||||
def _commit_message(self, entropy_server, successfull_mirrors):
|
||||
"""
|
||||
Ask user to enter the commit message for data being pushed.
|
||||
Store inside rss metadata object.
|
||||
"""
|
||||
tmp_fd, tmp_commit_path = tempfile.mkstemp(
|
||||
prefix="eit._push", suffix=".COMMIT_MSG")
|
||||
with os.fdopen(tmp_fd, "w") as tmp_f:
|
||||
tmp_f.write(EitPush.DEFAULT_REPO_COMMIT_MSG)
|
||||
if successfull_mirrors:
|
||||
tmp_f.write("# Changes to be committed:\n")
|
||||
for sf_mirror in sorted(successfull_mirrors):
|
||||
tmp_f.write("#\t updated: %s\n" % (sf_mirror,))
|
||||
|
||||
# spawn editor
|
||||
cm_msg_rc = entropy_server.edit_file(tmp_commit_path)
|
||||
commit_msg = None
|
||||
if not cm_msg_rc:
|
||||
# wtf?, fallback to old way
|
||||
def fake_callback(*args, **kwargs):
|
||||
return True
|
||||
|
||||
input_params = [
|
||||
('message', _("Commit message"), fake_callback, False)]
|
||||
commit_data = entropy_server.input_box(
|
||||
_("Enter the commit message"),
|
||||
input_params, cancel_button = True)
|
||||
if commit_data:
|
||||
commit_msg = commit_data['message']
|
||||
else:
|
||||
commit_msg = ''
|
||||
with open(tmp_commit_path, "r") as tmp_f:
|
||||
for line in tmp_f.readlines():
|
||||
if line.strip().startswith("#"):
|
||||
continue
|
||||
commit_msg += line
|
||||
entropy_server.output(commit_msg)
|
||||
|
||||
os.remove(tmp_commit_path)
|
||||
return commit_msg
|
||||
|
||||
def __print_repository_status(self, entropy_server, repository_id):
|
||||
remote_db_status = entropy_server.Mirrors.remote_repository_status(
|
||||
repository_id)
|
||||
|
||||
entropy_server.output(
|
||||
"%s:" % (brown(_("Entropy Repository Status")),),
|
||||
importance=1,
|
||||
header=darkgreen(" * ")
|
||||
)
|
||||
for url, revision in remote_db_status.items():
|
||||
host = EntropyTransceiver.get_uri_name(url)
|
||||
entropy_server.output(
|
||||
"%s: %s" % (darkgreen(_("Host")), bold(host)),
|
||||
header=" ")
|
||||
entropy_server.output(
|
||||
"%s: %s" % (purple(_("Remote")), blue(str(revision))),
|
||||
header=" ")
|
||||
|
||||
local_revision = entropy_server.local_repository_revision(
|
||||
repository_id)
|
||||
entropy_server.output(
|
||||
"%s: %s" % (brown(_("Local")), teal(str(local_revision))),
|
||||
header=" ")
|
||||
|
||||
def __sync_remote_database(self, entropy_server, repository_id):
|
||||
self.__print_repository_status(entropy_server, repository_id)
|
||||
# do the actual sync
|
||||
sts = entropy_server.Mirrors.sync_repository(repository_id)
|
||||
self.__print_repository_status(entropy_server, repository_id)
|
||||
return sts
|
||||
|
||||
def __push_repo(self, entropy_server, repository_id):
|
||||
sys_settings_plugin_id = \
|
||||
etpConst['system_settings_plugins_ids']['server_plugin']
|
||||
srv_data = self._settings()[sys_settings_plugin_id]['server']
|
||||
rss_enabled = srv_data['rss']['enabled']
|
||||
|
||||
mirrors_tainted, mirrors_errors, successfull_mirrors, \
|
||||
broken_mirrors, check_data = \
|
||||
entropy_server.Mirrors.sync_packages(
|
||||
repository_id, ask = self._ask,
|
||||
pretend = self._pretend)
|
||||
|
||||
if mirrors_errors and not successfull_mirrors:
|
||||
entropy_server.output(red(_("Aborting !")),
|
||||
importance=1, level="error", header=darkred(" !!! "))
|
||||
return 1
|
||||
if not successfull_mirrors:
|
||||
return 0
|
||||
|
||||
if mirrors_tainted and (self._as_repository_id is None):
|
||||
|
||||
commit_msg = None
|
||||
if self._ask and rss_enabled:
|
||||
commit_msg = self._commit_message(entropy_server,
|
||||
successfull_mirrors)
|
||||
elif rss_enabled:
|
||||
commit_msg = "Automatic update"
|
||||
|
||||
if commit_msg is None:
|
||||
commit_msg = "no commit message"
|
||||
ServerRssMetadata()['commitmessage'] = commit_msg
|
||||
|
||||
if self._as_repository_id is not None:
|
||||
# change repository push location
|
||||
ServerSystemSettingsPlugin.set_override_remote_repository(
|
||||
self._settings(), repository_id, self._as_repository_id)
|
||||
|
||||
sts = self.__sync_remote_database(entropy_server, repository_id)
|
||||
if sts == 0:
|
||||
# do not touch locking
|
||||
entropy_server.Mirrors.lock_mirrors(repository_id, False,
|
||||
unlock_locally = (self._as_repository_id is None))
|
||||
|
||||
if (sts == 0) and self._ask:
|
||||
q_rc = entropy_server.ask_question(
|
||||
_("Should I cleanup old packages on mirrors ?"))
|
||||
if q_rc == _("No"):
|
||||
return 0
|
||||
elif sts != 0:
|
||||
entropy_server.output(red(_("Aborting !")),
|
||||
importance=1, level="error", header=darkred(" !!! "))
|
||||
return sts
|
||||
|
||||
return 0
|
||||
|
||||
|
||||
EitCommandDescriptor.register(
|
||||
EitCommandDescriptor(
|
||||
EitPush,
|
||||
EitPush.NAME,
|
||||
_('reset repository to remote status'))
|
||||
)
|
||||
Reference in New Issue
Block a user