blob: 4aebd513d824d6e2acb1c92aa3f9acb0e369d00b (
plain)
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
|
# This is part of Kaylee
# -- this code is licensed GPLv3
# Copyright 2015-2016 Clayton G. Hobbs
# Portions Copyright 2013 Jezra
import json
import hashlib
class Hasher:
"""Keep track of hashes for Kaylee"""
def __init__(self, config):
self.config = config
try:
with open(self.config.hash_file, 'r') as f:
self.hashes = json.load(f)
except IOError:
# No stored hash
self.hashes = {}
def __getitem__(self, hashname):
try:
return self.hashes[hashname]
except (KeyError, TypeError):
return None
def __setitem__(self, hashname, value):
self.hashes[hashname] = value
def get_hash_object(self):
"""Returns an object to compute a new hash"""
return hashlib.sha256()
def store(self):
"""Store the current hashes into a the hash file"""
with open(self.config.hash_file, 'w') as f:
json.dump(self.hashes, f)
|