Files
entropy/server/eit/commands/reset.py
2011-10-17 11:26:42 +02:00

97 lines
2.8 KiB
Python

# -*- 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
from entropy.i18n import _
from entropy.output import darkgreen, teal
from eit.commands.descriptor import EitCommandDescriptor
from eit.commands.push import EitPush
class EitReset(EitPush):
"""
Main Eit reset command.
"""
NAME = "reset"
ALIASES = []
def __init__(self, args):
EitPush.__init__(self, args)
self._reset_repository_id = None
self._local = False
def parse(self):
""" Overridden from EitPush """
descriptor = EitCommandDescriptor.obtain_descriptor(
EitReset.NAME)
parser = argparse.ArgumentParser(
description=descriptor.get_description(),
formatter_class=argparse.RawDescriptionHelpFormatter,
prog="%s %s" % (sys.argv[0], EitReset.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("--local", action="store_true",
default=False,
help=_("do not pull the remote repository"))
try:
nsargs = parser.parse_args(self._args)
except IOError as err:
return parser.print_help, []
self._local = nsargs.local
self._ask = not nsargs.quick
if nsargs.repo is not None:
self._repositories.append(nsargs.repo)
self._reset_repository_id = nsargs.repo
return self._call_locked, [self._reset, nsargs.repo]
def _reset(self, entropy_server):
repository_id = self._reset_repository_id
if repository_id is None:
repository_id = entropy_server.repository()
rev_path = entropy_server._get_local_repository_revision_file(
repository_id)
try:
with open(rev_path, "w") as rev_f:
rev_f.write("0\n")
except (IOError, OSError) as err:
entropy_server.output(
"%s: %s" % (_("reset error"), err),
importance=1, level="error")
return 1
entropy_server.output(
darkgreen(_("local repository revision reset complete")),
importance=1)
if self._local:
return 0
return self._push(entropy_server)
EitCommandDescriptor.register(
EitCommandDescriptor(
EitReset,
EitReset.NAME,
_('reset repository to remote status'))
)