plugins/backup.py

   1
   2 from yap.yap import YapCore, YapError
   3 import os
   4
   5 class BackupPlugin(YapCore):
   6     "Save a backup of files before reverting them"
   7
   8     def cmd_revert(self, *args, **flags):
   9         files = set(args)
  10         changed = set(self._get_staged_files() + self._get_unstaged_files())
  11
  12         if '-a' in flags:
  13             x = changed
  14         else:
  15             x = files.intersection(changed)
  16
  17         for f in x:
  18             os.system("cp %s %s~" % (f, f))
  19         super(BackupPlugin, self).cmd_revert(*args, **flags)