Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-06-23 15:30:45 -04:00
Code Issues Packages Projects Releases Wiki Activity

Make moving the library location via the Preferences dialog faster

Browse Source
This commit is contained in:
Kovid Goyal 2009-08-19 14:45:22 -06:00
parent 642506bba0
commit d6124b6e36
1 changed files with 21 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/calibre/library/database2.py
View File

@ -1445,33 +1445,37 @@ class LibraryDatabase2(LibraryDatabase):
self.notify('add', [id])
def move_library_to(self, newloc, progress=lambda x: x):
books = self.conn.get('SELECT id, path, title FROM books')
if not os.path.exists(newloc):
os.makedirs(newloc)
items = os.listdir(self.library_path)
old_dirs = set([])
for i, book in enumerate(books):
path = book[1]
if not path:
continue
dir = path.split('/')[0]
srcdir = os.path.join(self.library_path, dir)
tdir = os.path.join(newloc, dir)
if os.path.exists(tdir):
shutil.rmtree(tdir)
if os.path.exists(srcdir):
shutil.copytree(srcdir, tdir)
old_dirs.add(srcdir)
progress(book[2])
for i, x in enumerate(items):
src = os.path.join(self.library_path, x)
dest = os.path.join(newloc, x)
if os.path.isdir(src):
if os.path.exists(dest):
shutil.rmtree(dest)
shutil.copytree(src, dest)
old_dirs.add(src)
else:
if os.path.exists(dest):
os.remove(dest)
shutil.copyfile(src, dest)
if not isinstance(x, unicode):
x = x.decode(filesystem_encoding, 'replace')
progress(x)
dbpath = os.path.join(newloc, os.path.basename(self.dbpath))
shutil.copyfile(self.dbpath, dbpath)
opath = self.dbpath
self.conn.close()
self.library_path, self.dbpath = newloc, dbpath
self.connect()
try:
os.unlink(opath)
except:
pass
for dir in old_dirs:
try:
shutil.rmtree(dir)
except:
pass