-
-
Notifications
You must be signed in to change notification settings - Fork 264
/
FileUtilities.py
797 lines (705 loc) · 26.3 KB
/
FileUtilities.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
# vim: ts=4:sw=4:expandtab
# -*- coding: UTF-8 -*-
# BleachBit
# Copyright (C) 2014 Andrew Ziem
# http://bleachbit.sourceforge.net
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
File-related utilities
"""
import atexit
import codecs
import errno
import glob
import locale
import os
import random
import re
import stat
import string
import sys
import tempfile
import time
import ConfigParser
if 'nt' == os.name:
import win32file
if 'posix' == os.name:
from General import WindowsError
class OpenFiles:
"""Cached way to determine whether a file is open by active process"""
def __init__(self):
self.last_scan_time = None
self.files = []
def file_qualifies(self, filename):
"""Return boolean whether filename qualifies to enter cache (check \
against blacklist)"""
return not filename.startswith("/dev") and \
not filename.startswith("/proc")
def scan(self):
"""Update cache"""
self.last_scan_time = time.time()
self.files = []
for filename in glob.iglob("/proc/*/fd/*"):
try:
target = os.path.realpath(filename)
except TypeError:
# happens, for example, when link points to
# '/etc/password\x00 (deleted)'
continue
if self.file_qualifies(target):
self.files.append(target)
def is_open(self, filename):
"""Return boolean whether filename is open by running process"""
if None == self.last_scan_time or (time.time() -
self.last_scan_time) > 10:
self.scan()
return filename in self.files
def __random_string(length):
"""Return random alphanumeric characters of given length"""
return ''.join(random.choice(string.ascii_letters + '0123456789_.-')
for i in xrange(length))
def bytes_to_human(bytes_i):
"""Display a file size in human terms (megabytes, etc.) using SI standard"""
storage_multipliers = {1000 ** 5: 'PB', 1000 ** 4: 'TB',
1000 ** 3: 'GB', 1000 ** 2: 'MB', 1000: 'kB', 1: 'B'}
assert(isinstance(bytes_i, (int, long)))
if 0 == bytes_i:
return "0"
if bytes_i >= 1000 ** 3:
decimals = 2
elif bytes_i >= 1000:
decimals = 1
else:
decimals = 0
for key in sorted(storage_multipliers.keys(), reverse=True):
if bytes_i >= key:
abbrev = round((1.0 * bytes_i) / key, decimals)
suf = storage_multipliers[key]
return locale.str(abbrev) + suf
if bytes_i < 0:
return "-" + bytes_to_human(abs(bytes_i))
def children_in_directory(top, list_directories=False):
"""Iterate files and, optionally, subdirectories in directory"""
if type(top) is tuple:
for top_ in top:
for pathname in children_in_directory(top_, list_directories):
yield pathname
return
for (dirpath, dirnames, filenames) in os.walk(top, topdown=False):
if list_directories:
for dirname in dirnames:
yield os.path.join(dirpath, dirname)
for filename in filenames:
yield os.path.join(dirpath, filename)
def clean_ini(path, section, parameter):
"""Delete sections and parameters (aka option) in the file"""
# read file to parser
config = ConfigParser.RawConfigParser()
fp = codecs.open(path, 'r', encoding='utf_8_sig')
config.readfp(fp)
# change file
changed = False
if config.has_section(section):
if None == parameter:
changed = True
config.remove_section(section)
elif config.has_option(section, parameter):
changed = True
config.remove_option(section, parameter)
# write file
if changed:
from Options import options
fp.close()
if options.get('shred'):
delete(path, True)
fp = codecs.open(path, 'wb', encoding='utf_8')
config.write(fp)
def clean_json(path, target):
"""Delete key in the JSON file"""
try:
import json
except:
import simplejson as json
changed = False
targets = target.split('/')
# read file to parser
js = json.load(open(path, 'r'))
# change file
pos = js
while True:
new_target = targets.pop(0)
if not isinstance(pos, dict):
break
if new_target in pos and len(targets) > 0:
# descend
pos = pos[new_target]
elif new_target in pos:
# delete terminal target
changed = True
del(pos[new_target])
else:
# target not found
break
if 0 == len(targets):
# target not found
break
if changed:
from Options import options
if options.get('shred'):
delete(path, True)
# write file
json.dump(js, open(path, 'w'))
def delete(path, shred=False, ignore_missing=False, allow_shred=True):
"""Delete path that is either file, directory, link or FIFO.
If shred is enabled as a function parameter or the BleachBit global
parameter, the path will be shredded unless allow_shred = False.
"""
from Options import options
is_special = False
if not os.path.lexists(path):
if ignore_missing:
return
raise OSError(2, 'No such file or directory', path)
if 'posix' == os.name:
# With certain (relatively rare) files on Windows os.lstat()
# may return Access Denied
mode = os.lstat(path)[stat.ST_MODE]
is_special = stat.S_ISFIFO(mode) or stat.S_ISLNK(mode)
if is_special:
os.remove(path)
elif os.path.isdir(path):
delpath = path
if allow_shred and (shred or options.get('shred')):
delpath = wipe_name(path)
try:
os.rmdir(delpath)
except OSError, e:
# [Errno 39] Directory not empty
# https://bugs.launchpad.net/bleachbit/+bug/1012930
if errno.ENOTEMPTY == e.errno:
print "info: directory '%s' is not empty" % (path)
else:
raise
except WindowsError, e:
# WindowsError: [Error 145] The directory is not empty:
# 'C:\\Documents and Settings\\username\\Local Settings\\Temp\\NAILogs'
# Error 145 may happen if the files are scheduled for deletion
# during reboot.
if 145 == e.winerror:
print "info: directory '%s' is not empty" % (path)
else:
raise
elif os.path.isfile(path):
# wipe contents
if allow_shred and (shred or options.get('shred')):
try:
wipe_contents(path)
except IOError, e:
# permission denied (13) happens shredding MSIE 8 on Windows 7
print "debug: IOError #%s shredding '%s'" % (e.errno, path)
# wipe name
os.remove(wipe_name(path))
else:
# unlink
os.remove(path)
else:
print "info: special file type cannot be deleted: '%s'" % (path)
def ego_owner(filename):
"""Return whether current user owns the file"""
return os.lstat(filename).st_uid == os.getuid()
def exists_in_path(filename):
"""Returns boolean whether the filename exists in the path"""
delimiter = ':'
if 'nt' == os.name:
delimiter = ';'
for dirname in os.getenv('PATH').split(delimiter):
if os.path.exists(os.path.join(dirname, filename)):
return True
return False
def exe_exists(pathname):
"""Returns boolean whether executable exists"""
if os.path.isabs(pathname):
return os.path.exists(pathname)
else:
return exists_in_path(pathname)
def execute_sqlite3(path, cmds):
"""Execute 'cmds' on SQLite database 'path'"""
import sqlite3
conn = sqlite3.connect(path)
cursor = conn.cursor()
for cmd in cmds.split(';'):
try:
cursor.execute(cmd)
except sqlite3.DatabaseError, exc:
raise sqlite3.DatabaseError('%s: %s' % (exc, path))
except sqlite3.OperationalError, exc:
if exc.message.find('no such function: ') >= 0:
# fixme: determine why randomblob and zeroblob are not
# available
print 'warning: %s' % exc.message
else:
raise sqlite3.OperationalError('%s: %s' % (exc, path))
cursor.close()
conn.commit()
conn.close()
def expand_glob_join(pathname1, pathname2):
"""Join pathname1 and pathname1, expand pathname, glob, and return as list"""
ret = []
pathname3 = os.path.expanduser(
os.path.expandvars(os.path.join(pathname1, pathname2)))
for pathname4 in glob.iglob(pathname3):
ret.append(pathname4)
return ret
def free_space(pathname):
"""Return free space in bytes"""
if 'nt' == os.name:
import ctypes
free_bytes = ctypes.c_int64()
rc = ctypes.windll.kernel32.GetDiskFreeSpaceExW(unicode(pathname),
ctypes.byref(free_bytes), None, None)
if 0 == rc:
raise ctypes.WinError()
return free_bytes.value
mystat = os.statvfs(pathname)
return mystat.f_bfree * mystat.f_bsize
def getsize(path):
"""Return the actual file size considering spare files
and symlinks"""
if 'posix' == os.name:
try:
__stat = os.lstat(path)
except OSError, e:
# OSError: [Errno 13] Permission denied
# can happen when a regular user is trying to find the size of /var/log/hp/tmp
# where /var/log/hp is 0774 and /var/log/hp/tmp is 1774
if errno.EACCES == e.errno:
return 0
raise
return __stat.st_blocks * 512
if 'nt' == os.name:
# On rare files os.path.getsize() returns access denied, so try
# FindFilesW instead
finddata = win32file.FindFilesW(path)
if finddata == []:
# FindFilesW doesn't work for directories, so fall back to getsize()
return os.path.getsize(path)
else:
size = (finddata[0][4] * (0xffffffff + 1)) + finddata[0][5]
return size
return os.path.getsize(path)
def getsizedir(path):
"""Return the size of the contents of a directory"""
total_bytes = 0
for node in children_in_directory(path, list_directories=False):
total_bytes += getsize(node)
return total_bytes
def globex(pathname, regex):
"""Yield a list of files with pathname and filter by regex"""
if type(pathname) is tuple:
for singleglob in pathname:
for path in globex(singleglob, regex):
yield path
else:
for path in glob.iglob(pathname):
if re.search(regex, path):
yield path
def guess_overwrite_paths():
"""Guess which partitions to overwrite (to hide deleted files)"""
# In case overwriting leaves large files, placing them in
# ~/.config makes it easy to find them and clean them.
ret = []
if 'posix' == os.name:
home = os.path.expanduser('~/.cache')
if not os.path.exists(home):
home = os.path.expanduser("~")
ret.append(home)
if not same_partition(home, '/tmp/'):
ret.append('/tmp')
elif 'nt' == os.name:
localtmp = os.path.expandvars('$TMP')
if not os.path.exists(localtmp):
print 'ERROR: %TMP% does not exist:', localtmp
localtmp = None
else:
from win32file import GetLongPathName
localtmp = GetLongPathName(localtmp)
from Windows import get_fixed_drives
for drive in get_fixed_drives():
try:
if localtmp and same_partition(localtmp, drive):
ret.append(localtmp)
else:
ret.append(drive)
except Exception, e:
# see https://github.com/az0/bleachbit/issues/27
# https://bugs.launchpad.net/bleachbit/+bug/1372179
print 'ERROR in same_partition(%s, %s): %s' % (localtmp, drive, e)
else:
NotImplementedError('Unsupported OS in guess_overwrite_paths')
return ret
def human_to_bytes(human, hformat='si'):
"""Convert a string like 10.2GB into bytes. By
default use SI standard (base 10). The format of the
GNU command 'du' (base 2) also supported."""
if 'si' == hformat:
multiplier = {'B': 1, 'kB': 1000, 'MB': 1000 ** 2,
'GB': 1000 ** 3, 'TB': 1000 ** 4}
matches = re.findall("^([0-9]*)(\.[0-9]{1,2})?([kMGT]{0,1}B)$", human)
elif 'du' == hformat:
multiplier = {'B': 1, 'KB': 1024, 'MB': 1024 ** 2,
'GB': 1024 ** 3, 'TB': 1024 ** 4}
matches = re.findall("^([0-9]*)(\.[0-9]{1,2})?([KMGT]{0,1}B)$", human)
else:
raise ValueError("Invalid format: '%s'" % hformat)
if [] == matches or 2 > len(matches[0]):
raise ValueError(
"Invalid input for '%s' (hformat='%s')" % (human, hformat))
return int(float(matches[0][0] + matches[0][1]) * multiplier[matches[0][2]])
def listdir(directory):
"""Return full path of files in directory.
Path may be a tuple of directories."""
if type(directory) is tuple:
for dirname in directory:
for pathname in listdir(dirname):
yield pathname
return
dirname = os.path.expanduser(directory)
if not os.path.lexists(dirname):
return
for filename in os.listdir(dirname):
yield os.path.join(dirname, filename)
def same_partition(dir1, dir2):
"""Are both directories on the same partition?"""
if 'nt' == os.name:
return free_space(dir1) == free_space(dir2)
stat1 = os.statvfs(dir1)
stat2 = os.statvfs(dir2)
return (stat1.f_blocks == stat2.f_blocks and
stat1.f_bfree == stat2.f_bfree)
def sync():
"""Flush file system buffers. sync() is different than fsync()"""
if 'posix' == os.name:
import ctypes
rc = ctypes.cdll.LoadLibrary('libc.so.6').sync()
if 0 != rc:
print 'ERROR: sync() returned code %d' % rc
if 'nt' == os.name:
import ctypes
ctypes.cdll.LoadLibrary('msvcrt.dll')._flushall()
def whitelisted(path):
"""Check whether this path is whitelisted"""
from Options import options
for pathname in options.get_whitelist_paths():
if pathname[0] == 'file' and path == pathname[1]:
return True
if pathname[0] == 'folder':
if path == pathname[1]:
return True
if path.startswith(pathname[1] + os.sep):
return True
if 'nt' == os.name:
# Windows is case insensitive
if pathname[0] == 'file' and path.lower() == pathname[1].lower():
return True
if pathname[0] == 'folder':
if path.lower() == pathname[1].lower():
return True
if path.lower().startswith(pathname[1].lower() + os.sep):
return True
# Simple drive letter like C:\ matches everything below
if len(pathname[1]) == 3 and path.lower().startswith(pathname[1].lower()):
return True
return False
def wipe_contents(path, truncate=True):
"""Wipe files contents
http://en.wikipedia.org/wiki/Data_remanence
2006 NIST Special Publication 800-88 (p. 7): "Studies have
shown that most of today's media can be effectively cleared
by one overwrite"
"""
size = getsize(path)
try:
f = open(path, 'wb')
except IOError, e:
if e.errno == errno.EACCES: # permission denied
os.chmod(path, 0200) # user write only
f = open(path, 'wb')
else:
raise
blanks = chr(0) * 4096
while size > 0:
f.write(blanks)
size -= 4096
f.flush() # flush to OS buffer
os.fsync(f.fileno()) # force write to disk
if truncate:
f.truncate(0)
f.flush()
os.fsync(f.fileno())
f.close()
def wipe_inodes(pathname, idle=False):
"""Create empty files with long names to clear remanence in metadata"""
# When the file system is full, Linux 3.5 will sometimes
# give an error and sometimes succeed.
errors = 0
files = []
if 'nt' == os.name:
# 255 according to http://en.wikipedia.org/wiki/Long_filename
f_namemax = 250
if not pathname.endswith('\\'):
pathname += '\\'
elif 'posix' == os.name:
f_namemax = os.statvfs(pathname).f_namemax
stats = os.statvfs(pathname)
inodes_avail_original = stats.f_favail
suffix = ' ' * (f_namemax - len(pathname) - 10)
print 'info: creating empty files'
print 'debug: suffix length = ', len(suffix)
last_idle = time.time()
# Creating the files in a sub-directory improves recovery
# of disk space:
# http://bleachbit.sourceforge.net/news/deleting-files-doesnt-free-disk-space
tmpdir = os.path.join(pathname, __random_string(10))
os.mkdir(tmpdir)
# create empty files until file system is exhausted
while True:
try:
(fd, fn) = tempfile.mkstemp(dir=tmpdir, suffix=suffix, prefix='')
except KeyboardInterrupt:
print 'info: wipe_inodes: keyboard interrupt'
break
except:
print 'error', sys.exc_info()[1]
errors += 1
if os.path.exists(fn):
# on Windows with FAT32, files sometimes exist despite error
files.append(fn)
try:
os.close(fd)
except:
pass
else:
errors = 0
files.append(fn)
os.close(fd)
if idle and (time.time() - last_idle) > 2:
stats = os.statvfs(pathname)
percent_done = None # not implemented on Windows
if 'posix' == os.name:
if inodes_avail_original > 0:
percent_done = 1.0 * \
(inodes_avail_original - stats.f_favail) / \
inodes_avail_original
else:
percent_done = 1.0
yield (2, percent_done, None)
last_idle = time.time()
# In case the application closes prematurely, make sure this file
# is deleted
atexit.register(delete, fn, allow_shred=False, ignore_missing=True)
if errors > 10:
break
if 'posix' == os.name and 0 == os.statvfs(pathname).f_ffree:
break
# flush once because it is slow
sync()
# display effectiveness
if 'posix' == os.name:
stats = os.statvfs(pathname)
print 'info: %d inodes available, %d to super user' % (stats.f_ffree, stats.f_favail)
# clean up
files_len_original = len(files)
files_remaining = len(files)
print 'info: deleting %d empty files' % files_len_original
for f in files:
try:
os.remove(f)
except:
if os.path.exists(f):
print 'warning: could not delete empty file %s' % f
else:
print 'warning: exception deleting empty file %s' % f
files_remaining = - 1
if idle and (time.time() - last_idle) > 2:
percent_done = 1.0 * \
(files_len_original - len(files)) / files_len_original
yield (3, percent_done, None)
last_idle = time.time()
os.rmdir(tmpdir)
print 'debug: done deleting empty files'
def wipe_name(pathname1):
"""Wipe the original filename and return the new pathname"""
(head, _) = os.path.split(pathname1)
# reference http://en.wikipedia.org/wiki/Comparison_of_file_systems#Limits
maxlen = 226
# first, rename to a long name
i = 0
while True:
try:
pathname2 = os.path.join(head, __random_string(maxlen))
os.rename(pathname1, pathname2)
break
except OSError:
if maxlen > 10:
maxlen -= 10
i += 1
if i > 100:
print 'warning: exhausted long rename: ', pathname1
pathname2 = pathname1
break
# finally, rename to a short name
i = 0
while True:
try:
pathname3 = os.path.join(head, __random_string(i + 1))
os.rename(pathname2, pathname3)
break
except:
i = i + 1
if i > 100:
print 'warning: exhausted short rename: ', pathname2
pathname3 = pathname2
break
return pathname3
def wipe_path(pathname, idle=False):
"""Wipe the free space in the path"""
def temporaryfile():
# reference
# http://en.wikipedia.org/wiki/Comparison_of_file_systems#Limits
maxlen = 245
f = None
i = 0
while True:
try:
kwargs = {
'dir': pathname, 'suffix': __random_string(maxlen)}
if sys.hexversion >= 0x02060000:
kwargs['delete'] = False
f = tempfile.NamedTemporaryFile(**kwargs)
# In case the application closes prematurely, make sure this
# file is deleted
atexit.register(
delete, f.name, allow_shred=False, ignore_missing=True)
break
except OSError, e:
if e.errno in (errno.ENAMETOOLONG, errno.ENOSPC):
# ext3 on Linux 3.5 returns ENOSPC if the full path is greater than 264.
# Shrinking the size helps.
if maxlen > 5:
maxlen -= 5
continue
raise
return f
def estimate_completion():
"""Return (percent, seconds) to complete"""
remaining_bytes = free_space(pathname)
done_bytes = start_free_bytes - remaining_bytes
if done_bytes < 0:
# maybe user deleted large file after starting wipe
done_bytes = 0
if 0 == start_free_bytes:
done_percent = 0
else:
done_percent = 1.0 * done_bytes / (start_free_bytes + 1)
done_time = time.time() - start_time
rate = done_bytes / (done_time + 0.0001) # bytes per second
remaining_seconds = int(remaining_bytes / (rate + 0.0001))
return (1, done_percent, remaining_seconds)
print "debug: wipe_path('%s')" % pathname
files = []
total_bytes = 0
start_free_bytes = free_space(pathname)
start_time = time.time()
# Because FAT32 has a maximum file size of 4,294,967,295 bytes,
# this loop is sometimes necessary to create multiple files.
while True:
try:
print 'debug: wipe_path: creating new temporary file'
f = temporaryfile()
except OSError, e:
# Linux gives errno 24
# Windows gives errno 28 No space left on device
if e.errno in (errno.EMFILE, errno.ENOSPC):
break
else:
raise
last_idle = time.time()
# Write large blocks to quickly fill the disk.
blanks = chr(0) * 65535
while True:
try:
f.write(blanks)
except IOError, e:
if e.errno == errno.ENOSPC:
if len(blanks) > 1:
# Try writing smaller blocks
blanks = blanks[0: (len(blanks) / 2)]
else:
break
elif e.errno != errno.EFBIG:
raise
if idle and (time.time() - last_idle) > 2:
# Keep the GUI responding, and allow the user to abort.
# Also display the ETA.
yield estimate_completion()
last_idle = time.time()
# Write to OS buffer
try:
f.flush()
except:
# IOError: [Errno 28] No space left on device
# seen on Microsoft Windows XP SP3 with ~30GB free space but
# not on another XP SP3 with 64MB free space
print "info: exception on f.flush()"
os.fsync(f.fileno()) # write to disk
# Remember to delete
files.append(f)
# For statistics
total_bytes += f.tell()
# If no bytes were written, then quit
if f.tell() < 1:
break
# sync to disk
sync()
# statistics
elapsed_sec = time.time() - start_time
rate_mbs = (total_bytes / (1000 * 1000)) / elapsed_sec
print 'note: wrote %d files and %d bytes in %d seconds at %.2f MB/s' % \
(len(files), total_bytes, elapsed_sec, rate_mbs)
# how much free space is left (should be near zero)
if 'posix' == os.name:
stats = os.statvfs(pathname)
print 'note: %d bytes and %d inodes available to non-super-user' % (stats.f_bsize * stats.f_bavail, stats.f_favail)
print 'note: %d bytes and %d inodes available to super-user' % (stats.f_bsize * stats.f_bfree, stats.f_ffree)
# truncate and close files
for f in files:
f.truncate(0)
while True:
try:
# Nikita: I noticed a bug that prevented file handles from
# being closed on FAT32. It sometimes takes two .close() calls
# to do actually close (and therefore delete) a temporary file
f.close()
break
except IOError, e:
if e.errno == 0:
print 'debug: handled unknown error 0'
time.sleep(0.1)
# explicitly delete
delete(f.name)
def vacuum_sqlite3(path):
"""Vacuum SQLite database"""
execute_sqlite3(path, 'vacuum')
openfiles = OpenFiles()