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
|
# Copyright (C) 2001 Artifex Software Inc.
# All Rights Reserved.
#
# This software is provided AS-IS with no warranty, either express or
# implied.
#
# This software is distributed under license and may not be copied, modified
# or distributed except as expressly authorized under the terms of that
# license. Refer to licensing information at http://www.artifex.com/
# or contact Artifex Software, Inc., 7 Mt. Lassen Drive - Suite A-134,
# San Rafael, CA 94903, U.S.A., +1(415)492-9861, for further information.
# $Id: gssum.py 8409 2007-11-27 20:43:09Z giles $
# gssum.py
#
# this module contains routines for calculating sums and managing
# the sum database
import anydbm
import gsconf
import os, string, md5
from stat import *
myself="gssum.py"
def exists(imagefile, dbname):
db = anydbm.open(dbname)
imagefilebase=os.path.basename(imagefile)
exists = db.has_key(imagefilebase)
db.close()
return exists
def add_file(imagefile, dbname, sum=None):
db = anydbm.open(dbname, 'w')
if sum == None:
sum = make_sum(imagefile)
if sum != None:
imagefilebase=os.path.basename(imagefile)
db[imagefilebase] = sum
else:
print "gssum.add_file failed to create a sum for",imagefile
db.close()
return sum
def get_sum(imagefile, dbname):
try:
db = anydbm.open(dbname)
except:
print "cannot open", dbname, "for", imagefile
imagefilebase=os.path.basename(imagefile)
sum = db[imagefilebase]
db.close()
return sum
def make_sum(imagefile):
try:
mode = os.stat(imagefile)[ST_MODE]
except OSError:
print "gssum.add_file failed to stat",imagefile
return None
if S_ISREG(mode):
sum = md5.new()
f = open(imagefile, "r")
data = f.read(1024)
while data:
sum.update(data)
data = f.read(1024)
f.close()
return sum.hexdigest()
print "gssum.add_file failed ISREG",imagefile
return None
|
