Moved trash logic to plugin structure

plugins/trasher/xdgtrash.py

@@ -0,0 +1,161 @@
+from .trash import Trash
+import shutil
+import os
+import os.path
+import datetime
+import sys
+import logging
+
+try:
+    import configparser
+except ImportError:
+    import ConfigParser as configparser
+
+
+
+
+class XDGTrash(Trash):
+    """XDG trash backend."""
+    def __init__(self):
+        self.trashdir = None
+        self.filedir  = None
+        self.infodir  = None
+
+        if os.getenv('XDG_DATA_HOME') is None:
+            self.trashdir = os.path.expanduser('~/.local/share/Trash')
+        else:
+            self.trashdir = os.getenv('XDG_DATA_HOME') + '/Trash'
+
+        try:
+            if not os.path.exists(self.trashdir):
+                os.mkdir(self.trashdir)
+        except OSError:
+            self.trashdir = os.path.join('tmp' 'TRASH')
+            raise('Couldn’t access the proper directory, temporary trash  is in in /tmp/TRASH')
+
+        self.filedir = self.trashdir + '/files/'
+        self.infodir = self.trashdir + '/info/'
+
+    def regenerate(self):
+        """Regenerate the trash and recreate metadata."""
+        print('Regenerating the trash and recreating metadata...')
+        zerosize = False
+
+        if not os.path.exists(self.trashdir):
+            os.mkdir(self.trashdir)
+            zerosize = True
+
+        if ((not os.path.exists(self.filedir)) or
+                (not os.path.exists(self.infodir))):
+            os.mkdir(self.filedir)
+            os.mkdir(self.infodir)
+            zerosize = True
+        if not zerosize:
+            trashsize = (self.size_dir(self.filedir) + self.size_dir(self.infodir))
+        else:
+            trashsize = 0
+
+        infofile = '[Cached]\nSize=' + str(trashsize) + '\n'
+        fh = open(os.path.join(self.trashdir, 'metadata'), 'w')
+        fh.write(infofile)
+        fh.close()
+
+    def empty(self, verbose):
+        """Empty the trash."""
+        print('emptying (verbose={})'.format(verbose))
+        shutil.rmtree(self.filedir)
+        shutil.rmtree(self.infodir)
+        self.regenerate()
+        if verbose:
+            sys.stderr.write(_('emptied the trash\n'))
+
+    def list(self, human=True):
+        """List the trash contents."""
+        if human:
+            print('listing contents (on stdout; human=True)')
+        else:
+            print('listing contents (return; human=False)')
+        dirs  = []
+        files = []
+        for f in os.listdir(self.filedir):
+            if os.path.isdir(self.filedir + f):
+                dirs.append(f)
+            else:
+                files.append(f)
+
+        dirs.sort()
+        files.sort()
+
+        allfiles = []
+        for i in dirs:
+            allfiles.append(i + '/')
+        for i in files:
+            allfiles.append(i)
+        if human:
+            if allfiles != []:
+                print('\n'.join(allfiles))
+        else:
+            return allfiles
+
+    def trash(self, filepath, verbose):
+        """Move specified file to trash."""
+        print('trashing file {} (verbose={})'.format(filepath, verbose))
+        # Filename alteration, a big mess.
+        filename = os.path.basename(filepath)
+        fileext  = os.path.splitext(filename)
+
+        tomove = filename
+        collision = True
+        i = 1
+
+        while collision:
+            if os.path.lexists(self.filedir + tomove):
+                tomove = fileext[0] + ' ' + str(i) + fileext[1]
+                i = i + 1
+            else:
+                collision = False
+
+        infofile = """[Trash Info]
+Path={}
+DeletionDate={}
+""".format(os.path.realpath(filepath),
+           datetime.datetime.now().strftime('%Y-%m-%dT%H:%m:%S'))
+
+        os.rename(filepath, self.filedir + tomove)
+
+        f = open(os.path.join(self.infodir, tomove + '.trashinfo'), 'w')
+        f.write(infofile)
+        f.close()
+
+        self.regenerate()
+
+        if verbose:
+            sys.stderr.write(_('trashed \'{}\'\n').format(filename))
+
+    def restore(self, filename, verbose, tocwd=False):
+        """Restore a file from trash."""
+        print('restoring file {} (verbose={}, tocwd={})'.format(filename, verbose, tocwd))
+        info = configparser.ConfigParser()
+        if os.path.exists(os.path.join(self.filedir, filename)):
+            info.read(os.path.join(self.infodir, filename + '.trashinfo'))
+            restname = os.path.basename(info.get('Trash Info', 'Path'))
+
+            if tocwd:
+                restdir = os.path.abspath('.')
+            else:
+                restdir = os.path.dirname(info.get('Trash Info', 'Path'))
+
+            restfile = os.path.join(restdir, restname)
+            if not os.path.exists(restdir):
+                raise TMError('restore', 'nodir', _('no such directory: {}'
+                              ' -- cannot restore').format(restdir))
+            os.rename(os.path.join(self.filedir, filename), restfile)
+            os.remove(os.path.join(self.infodir, filename + '.trashinfo'))
+            self.regenerate()
+            print('restored {} to {}'.format(filename, restfile))
+            if verbose:
+                sys.stderr.write(_('restored {} to {}\n').format(filename, restfile))
+
+        else:
+            print('couldn\'t find {} in trash'.format(filename))
+            raise TMError('restore', 'nofile', _('no such file in trash'))