Added check that record with specified primary key exists

Fixes bug 1213205

Change-Id: I1c8bc0557f8219a52e3e28ef0f78a5a8f26e3042
This commit is contained in:
Ilya Shakhat
2013-08-17 15:02:08 +04:00
parent b721d95d87
commit d30f283971

View File

@@ -112,11 +112,11 @@ class CachedMemoryStorage(MemoryStorage):
yield self.records[i]
def get_record_by_primary_key(self, primary_key):
record_id = list(self.primary_key_index[primary_key])
if record_id:
return self.records[record_id[0]]
else:
return None
if primary_key in self.primary_key_index:
record_id = list(self.primary_key_index[primary_key])
if record_id:
return self.records[record_id[0]]
return None
def get_original_company_name(self, company_name):
normalized = company_name.lower()