darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-02-23 03:44:38 -05:00
Code Issues Packages Projects Releases Wiki Activity

Make all options descend from 'all templates'

Browse Source
This commit is contained in:
Tom Byers
2018-12-31 11:11:16 +00:00
committed by Katie Smith
parent 3c4a186a25
commit 501ec2a534
1 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
app/main/forms.py
View File

@@ -747,18 +747,25 @@ class NestedRadioField(RadioFieldWithNoneOption):
super().__init__(*args, **kwargs)
def children(self):
child_map = {}
child_ids = [
folder['id'] for folder in self.all_template_folders
if folder['parent_id'] is None]
child_map[None] = [option for idx, option in enumerate(self) if option.data in child_ids or idx == 0]
# start map with root option as a single child entry
child_map = {None: [option for option in self
if option.data == self.NONE_OPTION_VALUE]}
# add entries for all other children
for option in self:
child_ids = [
folder['id'] for folder in self.all_template_folders
if folder['parent_id'] == option.data]
child_map[option.data] = [option for option in self if option.data in child_ids]
if option.data == self.NONE_OPTION_VALUE:
child_ids = [
folder['id'] for folder in self.all_template_folders
if folder['parent_id'] is None]
key = self.NONE_OPTION_VALUE
else:
child_ids = [
folder['id'] for folder in self.all_template_folders
if folder['parent_id'] == option.data]
key = option.data
child_map[key] = [option for option in self if option.data in child_ids]
return child_map