[eit.commands.reset] new command "eit reset"

2011-10-13 20:02:20 +02:00
parent 2fdc36e65a
commit e3acca3102
1 changed files with 98 additions and 0 deletions
@@ -0,0 +1,98 @@
+# -*- 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.
+    """
+
+    # TODO: incomplete, needs eit push implemented first
+
+    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'))
+    )