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
|
#!/usr/bin/env python
# Copyright 2009 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
"""Script to generate SQL to add MD5 sums for existing librarian files."""
__metaclass__ = type
import commands
import os
import sys
SQL = "UPDATE LibraryFileContent SET md5 = '%s' WHERE id = %d;"
def main(path, minimumID=0):
if not path.endswith('/'):
path += '/'
for dirpath, dirname, filenames in os.walk(path):
dirname.sort()
databaseID = dirpath[len(path):]
if not len(databaseID) == 8: # "xx/xx/xx"
continue
for filename in filenames:
databaseID = int(databaseID.replace('/', '') + filename, 16)
if databaseID < minimumID:
continue
filename = os.path.join(dirpath, filename)
md5sum = commands.getoutput('md5sum ' + filename).split(' ', 1)[0]
yield databaseID, md5sum
if __name__ == '__main__':
if len(sys.argv) > 2:
minimumID = int(sys.argv[2])
else:
minimumID = 0
for databaseID, md5sum in main(sys.argv[1], minimumID):
print SQL % (md5sum, databaseID)
|