def get_message_bank(domain, risk_profile=None, for_comparing=False):
if risk_profile is not None:
messages = FRIMessageBankMessage.view("fri/message_bank", key=[domain, risk_profile], include_docs=True).all()
else:
messages = FRIMessageBankMessage.view(
"fri/message_bank", startkey=[domain], endkey=[domain, {}], include_docs=True
).all()
if for_comparing:
result = []
for message in messages:
result.append({"message": message, "compare_string": letters_only(message.message)})
return result
else:
return messages
def get_message_bank(domain, risk_profile=None, for_comparing=False):
if risk_profile is not None:
messages = FRIMessageBankMessage.view("fri/message_bank", key=[domain, risk_profile], include_docs=True).all()
else:
messages = FRIMessageBankMessage.view("fri/message_bank", startkey=[domain], endkey=[domain, {}], include_docs=True).all()
if for_comparing:
result = []
for message in messages:
result.append({
"message": message,
"compare_string": letters_only(message.message),
})
return result
else:
return messages
def handle(self, **options):
self.delete_docs(
'FRIMessageBankMessage',
FRIMessageBankMessage.view('fri/message_bank',
include_docs=True).all())
self.delete_docs(
'FRIRandomizedMessage',
FRIRandomizedMessage.view('fri/randomized_message',
include_docs=True).all())
self.delete_docs(
'FRIExtraMessage',
FRIExtraMessage.view('fri/extra_message', include_docs=True).all())
