darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 15:46:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix boolean columns

Browse Source
This commit is contained in:
Rebecca Law
2019-04-05 10:59:43 +01:00
parent 46c2f6a4ac
commit 38b8a00344
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
app/commands.py
View File

@@ -694,14 +694,22 @@ def populate_organisations_from_file(file_name):
# and user_to_organisation will be cleared before running this command. # and user_to_organisation will be cleared before running this command.
# Ignoring duplicates allows us to run the command again with the same file or same file with new rows. # Ignoring duplicates allows us to run the command again with the same file or same file with new rows.
with open(file_name, 'r') as f: with open(file_name, 'r') as f:
def boolean_or_none(field):
if field == '1':
return True
elif field == '0':
return False
elif field == '':
return None
for line in itertools.islice(f, 1, None): for line in itertools.islice(f, 1, None):
columns = line.split('|') columns = line.split('|')
print(columns) print(columns)
data = { data = {
'name': columns[0], 'name': columns[0],
'active': True, 'active': True,
'agreement_signed': True if columns[3] == 'TRUE' else False, 'agreement_signed': boolean_or_none(columns[3]),
'crown': True if columns[2] == 'TRUE' else False, 'crown': boolean_or_none(columns[2]),
'organisation_type': columns[1].lower() 'organisation_type': columns[1].lower()
} }
org = Organisation(**data) org = Organisation(**data)