robertvokac/bitrot
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Activity

0.5.0: test mode introduced

Browse Source
This commit is contained in:
Łukasz Langa 2013-03-15 17:12:04 +01:00
parent a46acd36c4
commit 26a6d62b02
2 changed files with 27 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README.rst
View File

@ -36,9 +36,12 @@ under 10 minutes. Both tests on HFS+.
Change Log Change Log
---------- ----------
0.4.1 0.5.0
~~~~~ ~~~~~
* ``--test`` command-line argument for testing the state without updating the
database on disk (works for testing databases you don't have write access to)
* size of the data read is reported upon finish * size of the data read is reported upon finish
* minor performance updates * minor performance updates

28
src/bitrot.py
View File

@ -31,14 +31,16 @@ import atexit
import datetime import datetime
import hashlib import hashlib
import os import os
import shutil
import sqlite3 import sqlite3
import stat import stat
import sys import sys
import tempfile
CHUNK_SIZE = 16384 CHUNK_SIZE = 16384
DOT_THRESHOLD = 200 DOT_THRESHOLD = 200
VERSION = (0, 4, 0) VERSION = (0, 5, 0)
def sha1(path): def sha1(path):
@ -51,7 +53,20 @@ def sha1(path):
return digest.hexdigest() return digest.hexdigest()
def get_sqlite3_cursor(path): def get_sqlite3_cursor(path, copy=False):
if copy:
if not os.path.exists(path):
raise ValueError("error: bitrot database at {} does not exist."
"".format(path))
db_copy = tempfile.NamedTemporaryFile(prefix='bitrot_', suffix='.db',
delete=False)
with open(path, 'rb') as db_orig:
try:
shutil.copyfileobj(db_orig, db_copy)
finally:
db_copy.close()
path = db_copy.name
atexit.register(os.unlink, path)
conn = sqlite3.connect(path) conn = sqlite3.connect(path)
atexit.register(conn.close) atexit.register(conn.close)
cur = conn.cursor() cur = conn.cursor()
@ -64,10 +79,10 @@ def get_sqlite3_cursor(path):
return conn return conn
def run(verbosity=1): def run(verbosity=1, test=False):
current_dir = b'.' # sic, relative path current_dir = b'.' # sic, relative path
bitrot_db = os.path.join(current_dir, b'.bitrot.db') bitrot_db = os.path.join(current_dir, b'.bitrot.db')
conn = get_sqlite3_cursor(bitrot_db) conn = get_sqlite3_cursor(bitrot_db, copy=test)
cur = conn.cursor() cur = conn.cursor()
new_paths = [] new_paths = []
updated_paths = [] updated_paths = []
@ -183,6 +198,7 @@ def run(verbosity=1):
print(' ', path) print(' ', path)
if not any((new_paths, updated_paths, missing_paths)): if not any((new_paths, updated_paths, missing_paths)):
print() print()
print('warning: database file not updated on disk (test mode).')
if error_count: if error_count:
sys.exit(1) sys.exit(1)
@ -211,6 +227,8 @@ def run_from_command_line():
'are used in calculation.') 'are used in calculation.')
parser.add_argument('-v', '--verbose', action='store_true', parser.add_argument('-v', '--verbose', action='store_true',
help='list new, updated and missing entries') help='list new, updated and missing entries')
parser.add_argument('-t', '--test', action='store_true',
help='just test against an existing database, don\'t update anything')
parser.add_argument('--version', action='version', parser.add_argument('--version', action='version',
version='%(prog)s {}.{}.{}'.format(*VERSION)) version='%(prog)s {}.{}.{}'.format(*VERSION))
args = parser.parse_args() args = parser.parse_args()
@ -225,7 +243,7 @@ def run_from_command_line():
verbosity = 0 verbosity = 0
elif args.verbose: elif args.verbose:
verbosity = 2 verbosity = 2
run(verbosity=verbosity) run(verbosity=verbosity, test=args.test)
if __name__ == '__main__': if __name__ == '__main__':