def import_row_to_table(c, table, row):
fields_to_import = [col.csv_name for col in table.columns.values()]
row_data = {}
for csv_field, value in sorted(row.items()):
if csv_field in fields_to_import:
col = csv_util.get_destination_col(table, csv_field)
if is_foreign_key(col, table):
value = lookup_foreign_key(c, table, col, csv_field, value)
else:
value = sqlgen.pre_process_value(col, value)
row_data[col.name] = value
if table.unique_rows == True and row_exists(c, table, row_data):
return
else:
sql = sqlgen.insert_sql(table, row_data)
c.execute(sql)
def lookup_foreign_key(c, table, col, csv_field, value):
foreign_key = table.foreign_keys[col.name]
matching_foreign_col = csv_util.get_destination_col(foreign_key.to_table,
csv_field)
value = sqlgen.pre_process_value(matching_foreign_col, value)
foreign_col_name = matching_foreign_col.name
foreign_keys = query(c, foreign_key.to_table, {foreign_col_name: value})
if len(foreign_keys) == 0:
msg = "{0} table has no row with '{1}' in the {2} column"
raise ForeignKeyException(msg.format(
foreign_key.to_table.name, value, foreign_col_name))
elif len(foreign_keys) > 1:
msg = "{0} table has more than one row with {1} = '{2}'"
raise ForeignKeyException(msg.format(
foreign_key.to_table.name, foreign_col_name, value))
elif foreign_keys[0] == None:
msg = "'{0}' column is empty for row where {1} == '{3}' in {3}"
raise ForeignKeyException(msg.format(
foreign_key.to_col, foreign_col_name, value,
foreign_key.to_table.name))
else:
return foreign_keys[0][foreign_key.to_col]