def addColumnByFileType(self, fileType, fieldName, required, field_type):
"""
Adds a new column to the schema
Args:
fileType -- One of the set of valid types of files (e.g. Award, AwardFinancial)
fieldName -- The name of the scheam column
required -- marks the column if data is allways required
field_type -- sets the type of data allowed in the column
Returns:
ID of new column
"""
fileId = self.getFileId(fileType)
if (fileId is None):
raise ValueError("Filetype does not exist")
newColumn = FileColumn()
newColumn.required = False
newColumn.name = fieldName
newColumn.file_id = fileId
field_type = field_type.upper()
types = self.getDataTypes()
#Allow for other names
if (field_type == "STR"):
field_type = "STRING"
elif (field_type == "FLOAT"):
field_type = "DECIMAL"
elif (field_type == "BOOL"):
field_type = "BOOLEAN"
#Check types
if field_type in types:
newColumn.field_types_id = types[field_type]
else:
raise ValueError("".join(
["Type ", field_type, " is not vaild for ",
str(fieldName)]))
#Check Required
required = required.upper()
if (required in ["TRUE", "FALSE"]):
if (required == "TRUE"):
newColumn.required = True
else:
raise ValueError("".join(
["Required is not boolean for ",
str(fieldName)]))
# Save
self.session.add(newColumn)
self.session.commit()
return newColumn.file_column_id
def addColumnByFileType(self,fileType,fieldName,required,field_type):
"""
Adds a new column to the schema
Args:
fileType -- One of the set of valid types of files (e.g. Award, AwardFinancial)
fieldName -- The name of the scheam column
required -- marks the column if data is allways required
field_type -- sets the type of data allowed in the column
Returns:
ID of new column
"""
fileId = self.getFileId(fileType)
if(fileId is None) :
raise ValueError("Filetype does not exist")
newColumn = FileColumn()
newColumn.required = False
newColumn.name = fieldName
newColumn.file_id = fileId
field_type = field_type.upper()
types = self.getDataTypes()
#Allow for other names
if(field_type == "STR") :
field_type = "STRING"
elif(field_type == "FLOAT") :
field_type = "DECIMAL"
elif(field_type == "BOOL"):
field_type = "BOOLEAN"
#Check types
if field_type in types :
newColumn.field_types_id = types[field_type]
else :
raise ValueError("".join(["Type ",field_type," is not vaild for ",str(fieldName)]))
#Check Required
required = required.upper()
if( required in ["TRUE","FALSE"]) :
if( required == "TRUE") :
newColumn.required = True
else :
raise ValueError("".join(["Required is not boolean for ",str(fieldName)]))
# Save
self.session.add(newColumn)
self.session.commit()
return newColumn.file_column_id
