plugins/backup.py

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